support for ActionController::API

fabrik42 · May 11, 2018 · a7d627e · a7d627e
1 parent 874ccfb
commit a7d627e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/acts_as_api.rb b/lib/acts_as_api.rb
@@ -53,8 +53,9 @@ module ActsAsApi
   Mongoid::Document.send :include, ActsAsApi::Adapters::Mongoid
 end
 
-# Attach ourselves to the action controller of Rails
-if defined?(ActionController::Base)
-  ActionController::Base.send :include, ActsAsApi::Rendering
+# Attach ourselves to the controller classes (Base and API) of Rails
+if defined?(ActionController::Base) || defined?(ActionController::API)
+  ActionController::Base.send :include, ActsAsApi::Rendering defined?(ActionController::Base)
+  ActionController::API.send :include, ActsAsApi::Rendering defined?(ActionController::API)
   ActsAsApi::RailsRenderer.setup
 end