PDFKit

# File lib/pdfkit/configuration.rb, line 40
def self.configure
  yield(configuration)
end

# File lib/pdfkit/pdfkit.rb, line 22
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

# File lib/pdfkit/pdfkit.rb, line 34
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| %{"#{arg.shellescape}"}}
end

# File lib/pdfkit/pdfkit.rb, line 50
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

# File lib/pdfkit/pdfkit.rb, line 77
def to_file(path)
  self.to_pdf(path)
  File.new(path)
end

# File lib/pdfkit/pdfkit.rb, line 60
def to_pdf(path=nil)
  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?
  return result
end

# File lib/pdfkit/pdfkit.rb, line 103
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

# File lib/pdfkit/pdfkit.rb, line 84
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

# File lib/pdfkit/pdfkit.rb, line 126
def normalize_arg(arg)
  arg.to_s.downcase.gsub(/[^a-z0-9]/,'-')
end

# File lib/pdfkit/pdfkit.rb, line 115
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

# File lib/pdfkit/pdfkit.rb, line 130
def normalize_value(value)
  case value
  when TrueClass
    nil
  else
    value.to_s
  end
end

# File lib/pdfkit/pdfkit.rb, line 99
def style_tag_for(stylesheet)
  "<style>#{File.read(stylesheet)}</style>"
end