lib/jb/railtie.rb in jb-0.5.0 vs lib/jb/railtie.rb in jb-0.6.0

@@ -1,12 +1,28 @@
 # frozen_string_literal: true
+
 module Jb
   class Railtie < ::Rails::Railtie
     initializer 'jb' do
       ActiveSupport.on_load :action_view do
         require 'jb/action_view_monkeys'
         require 'jb/handler'
         ::ActionView::Template.register_template_handler :jb, Jb::Handler
+      end
+    end
+
+    if Rails::VERSION::MAJOR >= 5
+      module ::ActionController
+        module ApiRendering
+          include ActionView::Rendering
+        end
+      end
+
+      ActiveSupport.on_load :action_controller do
+        if self == ActionController::API
+          include ActionController::Helpers
+          include ActionController::ImplicitRender
+        end
       end
     end
 
     generators do |app|
       Rails::Generators.configure! app.config.generators