lib/reuser.rb in reuser-3.1.0 vs lib/reuser.rb in reuser-3.1.1

@@ -1,6 +1,5 @@
-require 'reuser/role'
 require 'reuser/role_definition'
 require 'reuser/errors'
 
 module ReUser
   module ClassMethods
@@ -30,13 +29,16 @@
 
   def cant? permission
     !can?(permission)
   end
 
-  def could? permission, block_args
-    self.class.role(self.role).could? permission, block_args
+  def could? permission, *block_args
+    if block_args.empty?
+      block_args << self
+    end
+    self.class.role(self.role).could? permission, *block_args
   end
 
-  def couldnt? permission, block_args
-    !could?(permission, block_args)
+  def couldnt? permission, *block_args
+    !could?(permission, *block_args)
   end
 end