Add Rails 5 API support

CanCanCommunity · Nov 20, 2017 · 5ae12b0 · 5ae12b0
1 parent e88a267
commit 5ae12b0
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/lib/cancan/controller_additions.rb b/lib/cancan/controller_additions.rb
@@ -384,9 +384,12 @@ def cannot?(*args)
   end
 end
 
-if defined? ActionController
-  %w[Base API].each do |klass|
-    next unless ActionController.const_defined?(klass)
-    ActionController.const_get(klass).class_eval { include CanCan::ControllerAdditions }
+if defined? ActionController::Base
+  ActionController::Base.class_eval do
+    include CanCan::ControllerAdditions
+  end
+elsif defined? ActionController::API
+  ActionController::API.class_eval do
+    include CanCan::ControllerAdditions
   end
 end