require 'wicked_pdf/pdf_helper'
require 'wicked_pdf/renderer'
require 'wicked_pdf/asset_helper'

module WickedPdf
  class Railtie < Rails::Railtie
    initializer 'wicked_pdf.register' do |_app|
      ActiveSupport.on_load(:action_controller) {
        ActionController::Base.send :prepend, PdfHelper
        ActionController::Renderers.add :pdf do |template, options|
          WickedPdf::Renderer.new(self).render(options.merge(pdf: template))
        end
      }
      ActiveSupport.on_load(:action_view) { include WickedPdf::AssetHelper }
    end
  end
end

if Mime::Type.lookup_by_extension(:pdf).nil?
  Mime::Type.register('application/pdf', :pdf)
end