From efbd3731c986a8555869c184cda69ff9b910ce9f Mon Sep 17 00:00:00 2001
From: Luke Shumaker <lukeshu@lukeshu.com>
Date: Sat, 23 Dec 2017 15:52:22 -0500
Subject: pdf support

---
 lib/page_local.rb |  3 +++
 lib/page_pdf.rb   | 62 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 65 insertions(+)
 create mode 100644 lib/page_pdf.rb

(limited to 'lib')

diff --git a/lib/page_local.rb b/lib/page_local.rb
index e956f6a..6c70ac3 100644
--- a/lib/page_local.rb
+++ b/lib/page_local.rb
@@ -16,6 +16,9 @@ class LocalPage < Page
 			return IndexPage::new(inpath)
 		when [".md", ".org"].include?(File::extname(inpath))
 			return LocalPage::new(inpath)
+		when ".pdf" == File::extname(inpath)
+			require 'page_pdf'
+			return PdfPage::new(inpath)
 		else
 			return nil
 		end
diff --git a/lib/page_pdf.rb b/lib/page_pdf.rb
new file mode 100644
index 0000000..e70c887
--- /dev/null
+++ b/lib/page_pdf.rb
@@ -0,0 +1,62 @@
+# coding: utf-8
+require 'erb'
+require 'open3'
+require 'yaml'
+
+require 'page_local'
+
+class PdfPage < LocalPage
+	def initialize(filename)
+		super(filename)
+	end
+
+	def pdf_metadata
+		if @metadata.nil?
+			stdout, stderr, status = Open3::capture3("pdfinfo", "--", local_infile)
+			unless stderr.empty?
+				raise stderr
+			end
+			unless status.success?
+				raise status
+			end
+			raw_metadata = stdout.split("\n").map{|l|l.split(":", 2).map{|c|c.strip}}.to_h
+
+			# Transform the PDF property names to match our metadata names
+			key_map = {
+				"Title" => "title",
+				"Author" => "author",
+				"CreationDate" => "published",
+				"ModDate" => "updated",
+				# "Keywords" => "categories",
+			}
+			@metadata = raw_metadata.map{|k,v|[key_map[k]||k,v]}.to_h
+
+			yamlfile = local_infile.sub(/\.pdf$/, '.yaml')
+			if File::exist?(yamlfile)
+				@metadata = @metadata.merge(YAML::load(File::read(yamlfile)))
+			end
+		end
+		@metadata
+	end
+	def pdf_js_url
+		@@pdjfs ||= Config::get.url + 'pdfjs/web/viewer.html'
+	end
+	def pdf_viewer_url
+		@viewer_url ||= pdf_js_url + ('?file=' + URI::encode_www_form_component(pdf_js_url.route_to(local_srcurl)))
+	end
+
+	def local_intype
+		return 'markdown'
+	end
+	def local_depends
+		if @depends.nil?
+			yamlfile = local_infile.sub(/\.pdf$/, '.yaml')
+			metafile = File::exist?(yamlfile) ? yamlfile : File::dirname(yamlfile)
+			tmplfile = "tmpl/pdf.md.erb"
+			@depends = super.map{|k,v|[k,v.merge([metafile, tmplfile])]}.to_h
+		end
+		@depends
+	end
+end
+
+ERB::new(File::read("tmpl/pdf.md.erb")).def_method(PdfPage, 'local_input()', "tmpl/pdf.md.erb")
-- 
cgit v1.2.3