aboutsummaryrefslogtreecommitdiffstats
path: root/lib/attachment_to_html/adapter.rb
blob: 058fb2a01b631b2352ce5fa658a03802c8d43e50 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
module AttachmentToHTML
  class Adapter
    attr_reader :attachment

    # Public: Initialize a converter
    #
    # attachment - the FoiAttachment to convert to HTML
    # opts       - a Hash of options (default: {}):
    #              No options currently accepted
    def initialize(attachment, opts = {})
      @attachment = attachment
    end

    # Public: The title to use in the <title> tag
    #
    # Returns a String
    def title
      @title ||= attachment.display_filename
    end

    # Public: The contents of the extracted html <body> tag
    #
    # Returns a String
    def body
      @body ||= parse_body
    end

    def parse_body
      convert
    end

    # Public: Was the document conversion successful?
    #
    # Returns true
    def success?
      true
    end

    def has_content?
        !body.gsub(/\s+/,"").gsub(/\<[^\>]*\>/, "").empty?
    end

    def contains_images?
      body.match(/<img[^>]*>/mi)
    end

    def create_tempfile(text)
      tempfile = if RUBY_VERSION.to_f >= 1.9
        Tempfile.new('foiextract', '.', :encoding => text.encoding)
      else
        Tempfile.new('foiextract', '.')
      end
      tempfile.print(text)
      tempfile.flush
      tempfile
    end

    def cleanup_tempfile(tempfile)
      tempfile.close
      tempfile.delete
    end

    def attachment_body
      @attachment_body ||= attachment.body
    end
  end
end