Make activeadmin-mongoid 0.3.0 play nice with regular activerecord ac…

…tiveadmin resources
activeadmin · Jan 5, 2014 · 47cae71 · 47cae71
1 parent c2547d4
commit 47cae71
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/lib/active_admin/mongoid/helpers/collection.rb b/lib/active_admin/mongoid/helpers/collection.rb
@@ -1,8 +1,15 @@
 module ActiveAdmin
   module Helpers
     module Collection
+
+      alias original_collection_size collection_size
+      original_collection_size = instance_method(:collection_size)
       def collection_size(collection=collection)
-        collection.count(true)
+        if(not collection.empty? and collection.first.class.included_modules.include?(Mongoid::Document))
+          collection.count(true)
+        else
+          original_collection_size(collection)
+        end
       end
     end
   end