objectbouncer 0.0.3 → 0.0.4

data/README.mdown

@@ -15,6 +15,10 @@ Let's say we have a President who is protected by the SecretService:
       def high_five
         "high five!"
       end
+
+      def give(gift)
+        "thanks"
+      end
     end
 
 And the following people:
@@ -46,10 +50,12 @@ To protect the President we'd define a SecretService class like so:
     class SecretService
       include ObjectBouncer::Doorman
       door_policy do
-        deny :shake_hands, :if => Proc.new{|person| person.dictator? }
+        deny :shake_hands,  :if => Proc.new{|person| person.dictator? }
         allow :shake_hands, :if => Proc.new{|person| person.democrat? }
-        deny :high_five, :unless => Proc.new{|person, president|
+        deny :high_five,    :unless => Proc.new{|person, president|
                                       person.friend?(president) }
+        deny :give,         :unless => Proc.new{|person, president, *args|
+                                      args.first == :donation }
       end
     end
 
@@ -66,6 +72,8 @@ the President through SecretService first:
     SecretService.new(@tommy_chong, @president).shake_hands # Allowed
     SecretService.new(@joe_biden, @president).high_five # Allowed
     SecretService.new(@tommy_chong, @president).high_five # Raises PermissionDenied
+    SecretService.new(@gaddafi, @president).give(:donation) # Allowed
+    SecretService.new(@gaddafi, @president).give(:suspect_package) # Raises PermissionDenied
 
 ## Why would I want to use this?
 

data/lib/objectbouncer/base.rb

@@ -57,15 +57,16 @@ module ObjectBouncer
     def initialize(accessee, object)
       @accessee = accessee
       @object = object
+      super(nil)
       self
     end
 
     def method_missing(meth, *args, &block)
       if respond_to?(meth)
         raise "TODO!!!" if self.class.policies.nil? or self.class.policies.empty?
-        if call_allowed?(meth)
+        if call_allowed?(meth, *args)
           @object.send(meth, *args, &block)
-        elsif call_denied?(meth)
+        elsif call_denied?(meth, *args)
           raise ObjectBouncer::PermissionDenied.new
         end
       else
@@ -78,19 +79,19 @@ module ObjectBouncer
     end
 
     private
-      def call_allowed?(meth)
+      def call_allowed?(meth, *args)
         if policies = self.class.policies[meth]
-          return true if !policies[:allow][:unless].empty? && !policies[:allow][:unless].detect{|policy| policy.call(@accessee, @object) rescue nil}
-          return true if policies[:allow][:if].detect{|policy| policy.call(@accessee, @object) rescue nil}
-          return true if policies[:deny][:unless].detect{|policy| policy.call(@accessee, @object) rescue nil}
+          return true if !policies[:allow][:unless].empty? && !policies[:allow][:unless].detect{|policy| policy.call(@accessee, @object, *args) rescue nil}
+          return true if policies[:allow][:if].detect{|policy| policy.call(@accessee, @object, *args) rescue nil}
+          return true if policies[:deny][:unless].detect{|policy| policy.call(@accessee, @object, *args) rescue nil}
         end
       end
 
-      def call_denied?(meth)
+      def call_denied?(meth, *args)
         return true if self.class.lockdown?
         if policies = self.class.policies[meth]
-          return true if policies[:allow][:unless].detect{|policy| policy.call(@accessee, @object) rescue nil}
-          return true if policies[:deny][:if].detect{|policy| policy.call(@accessee, @object) rescue nil}
+          return true if policies[:allow][:unless].detect{|policy| policy.call(@accessee, @object, *args) rescue nil}
+          return true if policies[:deny][:if].detect{|policy| policy.call(@accessee, @object, *args) rescue nil}
           return true if !policies[:deny][:unless].empty? && !call_allowed?(meth)
         end
       end

data/test/objectbouncer/base_test.rb

@@ -7,6 +7,7 @@ class SecretService
     deny :shake_hands, :if => Proc.new{|person, president| person != president}
     allow :shake_hands, :if => Proc.new{|person, president| person.class == MichelleObama}
     deny :high_five, :unless => Proc.new{|person, president| person.who? == "it's me, Joe!"}
+    deny :give, :unless => Proc.new{|person, president, *args| args.first == :campaign_donation }
   end
 end
 
@@ -30,6 +31,10 @@ class President
   def watch_tv_appearance
     "I'm on your TV!"
   end
+
+  def give(gift)
+    "thanks"
+  end
 end
 
 class MichelleObama
@@ -79,6 +84,20 @@ class ObjectBouncerTest < Test::Unit::TestCase
       end
     end
 
+    should "let the public give a donation" do
+      joe_public = JoePublic.new
+      secret_service = SecretService.new(joe_public, @president)
+      assert_equal "thanks", secret_service.give(:campaign_donation)
+    end
+
+    should "not let the public give a package" do
+      joe_public = JoePublic.new
+      secret_service = SecretService.new(joe_public, @president)
+      assert_raise ObjectBouncer::PermissionDenied do
+        secret_service.give(:suspect_package)
+      end
+    end
+
   end
 
   context "going into complete lockdown" do

metadata

@@ -2,7 +2,7 @@
 name: objectbouncer
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors: 
 - Glenn Gillen
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-03-07 00:00:00 +00:00
+date: 2011-03-25 00:00:00 +00:00
 default_executable: 
 dependencies: []