1 files changed, 97 insertions, 0 deletions
diff --git a/lib/attachment_to_html/adapters/pdf.rb b/lib/attachment_to_html/adapters/pdf.rb
new file mode 100644
index 000000000..1fca2f201
--- /dev/null
+++ b/lib/attachment_to_html/adapters/pdf.rb
@@ -0,0 +1,97 @@
+module AttachmentToHTML
+    module Adapters
+        # Convert application/pdf documents in to HTML
+        class PDF
+
+            attr_reader :attachment, :tmpdir
+
+            # Public: Initialize a PDF converter
+            #
+            # attachment - the FoiAttachment to convert to HTML
+            # opts       - a Hash of options (default: {}):
+            #              :tmpdir  - String name of directory to store the
+            #                         converted document
+            def initialize(attachment, opts = {})
+                @attachment = attachment
+                @tmpdir = opts.fetch(:tmpdir, ::Rails.root.join('tmp'))
+            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
+
+            # Public: Was the document conversion successful?
+            #
+            # Returns a Boolean
+            def success?
+                has_content? || contains_images?
+            end
+
+            private
+
+            def parse_body
+                match = convert.match(/<body[^>]*>(.*?)<\/body>/mi)
+                match ? match[1] : ''
+            end
+
+            def has_content?
+                !body.gsub(/\s+/,"").gsub(/\<[^\>]*\>/, "").empty?
+            end
+
+            def contains_images?
+                body.match(/<img[^>]*>/mi) ? true : false
+            end
+
+            def convert
+                # Get the attachment body outside of the chdir call as getting
+                # the body may require opening files too
+                text = attachment_body
+
+                @converted ||= Dir.chdir(tmpdir) do
+                    tempfile = create_tempfile(text)
+
+                    html = AlaveteliExternalCommand.run("pdftohtml",
+                      "-nodrm", "-zoom", "1.0", "-stdout", "-enc", "UTF-8",
+                      "-noframes", tempfile.path, :timeout => 30
+                    )
+
+                    cleanup_tempfile(tempfile)
+
+                    html
+                end
+            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
+end