1 files changed, 121 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..8f826b910
--- /dev/null
+++ b/lib/attachment_to_html/adapters/pdf.rb
@@ -0,0 +1,121 @@
+module AttachmentToHTML
+    module Adapters
+        # Convert application/pdf documents in to HTML
+        class PDF
+
+            attr_reader :attachment, :wrapper, :tmpdir
+
+            # Public: Initialize a PDF converter
+            #
+            # attachment - the FoiAttachment to convert to HTML
+            # opts       - a Hash of options (default: {}):
+            #              :wrapper - String id of the div that wraps the
+            #                         attachment body
+            #              :tmpdir  - String name of directory to store the
+            #                         converted document
+            def initialize(attachment, opts = {})
+                @attachment = attachment
+                @wrapper = opts.fetch(:wrapper, 'wrapper')
+                @tmpdir = opts.fetch(:tmpdir, ::Rails.root.join('tmp'))
+            end
+
+            # Public: Convert the attachment to HTML
+            #
+            # Returns a String
+            def to_html
+                @html ||= generate_html
+            end
+
+            # Public: Was the document conversion successful?
+            #
+            # Returns a Boolean
+            def success?
+                has_content? || contains_images?
+            end
+
+            private
+
+            def generate_html
+                html =  "<!DOCTYPE html>"
+                html += "<html>"
+                html += "<head>"
+                html += "<title>#{ title }</title>"
+                html += "</head>"
+                html += "<body>"
+                html += "<div id=\"#{ wrapper }\">"
+                html += "<div id=\"view-html-content\">"
+                html += body
+                html += "</div>"
+                html += "</div>"
+                html += "</body>"
+                html += "</html>"
+            end
+
+            def title
+                @title ||= attachment.display_filename
+            end
+
+            def body
+                parsed_body
+            end
+
+            # Parse the output of the converted attachment so that we can pluck
+            # the parts we need and insert in to our own sensible template
+            #
+            # Returns a Nokogiri::HTML::Document
+            def parsed
+                @parsed ||= Nokogiri::HTML.parse(convert)
+            end
+
+            def parsed_body
+                parsed.css('body').inner_html
+            end
+
+            # Does the body element have any content, excluding HTML tags?
+            #
+            # Returns a Boolean
+            def has_content?
+                !parsed.css('body').inner_text.empty?
+            end
+
+            def contains_images?
+                parsed.css('body img').any?
+            end
+
+            def convert
+                @converted ||= Dir.chdir(tmpdir) do
+                    tempfile = create_tempfile
+                    write_attachment_body_to_tempfile(tempfile)
+
+                    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
+                if RUBY_VERSION.to_f >= 1.9
+                    Tempfile.new('foiextract', '.', :encoding => attachment.body.encoding)
+                else
+                    Tempfile.new('foiextract', '.')
+                end
+            end
+
+            def write_attachment_body_to_tempfile(tempfile)
+                tempfile.print(attachment.body)
+                tempfile.flush
+            end
+
+            def cleanup_tempfile(tempfile)
+                tempfile.close
+                tempfile.delete
+            end
+
+        end
+    end
+end