Skip to content

Commit 7af3260

Browse files
authored
Merge pull request #813 from mileszs/use-prepend-when-available
Use module#prepend when available.
2 parents fb879fd + da516f1 commit 7af3260

File tree

1 file changed

+7
-11
lines changed

1 file changed

+7
-11
lines changed

lib/wicked_pdf/railtie.rb

+7-11
Original file line numberDiff line numberDiff line change
@@ -5,20 +5,16 @@
55
class WickedPdf
66
if defined?(Rails.env)
77

8-
if Rails::VERSION::MAJOR >= 5
8+
if Rails::VERSION::MAJOR >= 4
99

1010
class WickedRailtie < Rails::Railtie
1111
initializer 'wicked_pdf.register' do |_app|
12-
ActionController::Base.send :prepend, PdfHelper
13-
ActionView::Base.send :include, WickedPdfHelper::Assets
14-
end
15-
end
16-
17-
elsif Rails::VERSION::MAJOR == 4
18-
19-
class WickedRailtie < Rails::Railtie
20-
initializer 'wicked_pdf.register' do |_app|
21-
ActionController::Base.send :include, PdfHelper
12+
if ActionController::Base.respond_to?(:prepend) &&
13+
Object.method(:new).respond_to?(:super_method)
14+
ActionController::Base.send :prepend, PdfHelper
15+
else
16+
ActionController::Base.send :include, PdfHelper
17+
end
2218
ActionView::Base.send :include, WickedPdfHelper::Assets
2319
end
2420
end

0 commit comments

Comments
 (0)