Created
May 3, 2012 14:01
-
-
Save NeQuissimus/2585848 to your computer and use it in GitHub Desktop.
pdfkit.rb for working with showoff
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # /Library/Ruby/Gems/1.8/gems/pdfkit-0.5.2/lib/pdfkit/pdfkit.rb | |
| # I kept getting invalid argument and file not found errors when creating a PDF | |
| # from my showoff presentations. Changing a few lines fixed this behaviour. | |
| # Thanks to clyfe (https://gist.github.com/1330326) for his optimized version! | |
| class PDFKit | |
| class NoExecutableError < StandardError | |
| def initialize | |
| msg = "No wkhtmltopdf executable found at #{PDFKit.configuration.wkhtmltopdf}\n" | |
| msg << ">> Please install wkhtmltopdf - https://github.com/jdpace/PDFKit/wiki/Installing-WKHTMLTOPDF" | |
| super(msg) | |
| end | |
| end | |
| class ImproperSourceError < StandardError | |
| def initialize(msg) | |
| super("Improper Source: #{msg}") | |
| end | |
| end | |
| attr_accessor :source, :stylesheets | |
| attr_reader :options | |
| def initialize(url_file_or_html, options = {}) | |
| @source = Source.new(url_file_or_html) | |
| @stylesheets = [] | |
| @options = PDFKit.configuration.default_options.merge(options) | |
| @options.merge! find_options_in_meta(url_file_or_html) unless source.url? | |
| @options = normalize_options(@options) | |
| raise NoExecutableError.new unless File.exists?(PDFKit.configuration.wkhtmltopdf) | |
| end | |
| def command(path = nil) | |
| args = [executable] | |
| args += @options.to_a.flatten.compact | |
| args << '--quiet' | |
| if @source.html? | |
| args << '-' # Get HTML from stdin | |
| else | |
| args << @source.to_s | |
| end | |
| args << (path || '-') # Write to file or stdout | |
| args.map {|arg| %Q{"#{arg.gsub('"', '\"')}"}} | |
| end | |
| def executable | |
| default = PDFKit.configuration.wkhtmltopdf | |
| return default if default !~ /^\// # its not a path, so nothing we can do | |
| if File.exist?(default) | |
| default | |
| else | |
| default.split('/').last | |
| end | |
| end | |
| #def to_pdf(path=nil) | |
| def to_pdf(path) | |
| append_stylesheets | |
| args = command(path) | |
| invoke = args.join(' ') | |
| result = IO.popen(invoke, "wb+") do |pdf| | |
| pdf.puts(@source.to_s) if @source.html? | |
| pdf.close_write | |
| pdf.gets(nil) | |
| end | |
| #result = File.read(path) if path | |
| #raise "command failed: #{invoke}" if result.to_s.strip.empty? | |
| raise "command failed: #{invoke}" unless $?.exitstatus == 0 | |
| #return result | |
| path | |
| end | |
| def to_file(path) | |
| self.to_pdf(path) | |
| File.new(path) | |
| end | |
| protected | |
| def find_options_in_meta(content) | |
| # Read file if content is a File | |
| content = content.read if content.is_a?(File) | |
| found = {} | |
| content.scan(/<meta [^>]*>/) do |meta| | |
| if meta.match(/name=["']#{PDFKit.configuration.meta_tag_prefix}/) | |
| name = meta.scan(/name=["']#{PDFKit.configuration.meta_tag_prefix}([^"']*)/)[0][0] | |
| found[name.to_sym] = meta.scan(/content=["']([^"']*)/)[0][0] | |
| end | |
| end | |
| found | |
| end | |
| def style_tag_for(stylesheet) | |
| "<style>#{File.read(stylesheet)}</style>" | |
| end | |
| def append_stylesheets | |
| raise ImproperSourceError.new('Stylesheets may only be added to an HTML source') if stylesheets.any? && !@source.html? | |
| stylesheets.each do |stylesheet| | |
| if @source.to_s.match(/<\/head>/) | |
| @source = Source.new(@source.to_s.gsub(/(<\/head>)/, style_tag_for(stylesheet)+'\1')) | |
| else | |
| @source.to_s.insert(0, style_tag_for(stylesheet)) | |
| end | |
| end | |
| end | |
| def normalize_options(options) | |
| normalized_options = {} | |
| options.each do |key, value| | |
| next if !value | |
| normalized_key = "--#{normalize_arg key}" | |
| normalized_options[normalized_key] = normalize_value(value) | |
| end | |
| normalized_options | |
| end | |
| def normalize_arg(arg) | |
| arg.to_s.downcase.gsub(/[^a-z0-9]/,'-') | |
| end | |
| def normalize_value(value) | |
| case value | |
| when TrueClass | |
| nil | |
| else | |
| value.to_s | |
| end | |
| end | |
| end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment