1 files changed, 62 insertions, 0 deletions
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")