Merge pull request jamesgolick#4 from cs/master

Support for passing lambdas to `fires` as :actor, :secondary_subject and :subject
jeremedia · Jun 23, 2011 · 1472e5f · 1472e5f
2 parents 476d60d + 8537ffd
commit 1472e5f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/lib/timeline_fu/fires.rb b/lib/timeline_fu/fires.rb
@@ -19,11 +19,14 @@ def fires(event_type, opts)
         method_name = :"fire_#{event_type}_after_#{opts[:on]}"
         define_method(method_name) do
           create_options = [:actor, :subject, :secondary_subject].inject({}) do |memo, sym|
-            case opts[sym]
-            when :self
-              memo[sym] = self
-            else
-              memo[sym] = send(opts[sym]) if opts[sym]
+            if opts[sym]
+              if opts[sym].respond_to?(:call)
+                memo[sym] = opts[sym].call(self)
+              elsif opts[sym] == :self
+                memo[sym] = self
+              else
+                memo[sym] = send(opts[sym])
+              end
             end
             memo
           end

diff --git a/test/test_helper.rb b/test/test_helper.rb
@@ -33,7 +33,7 @@ class Person < ActiveRecord::Base
   attr_accessor :new_watcher, :fire
 
   fires :follow_created,  :on     => :update, 
-                          :actor  => :new_watcher, 
+                          :actor  => lambda { |person| person.new_watcher }, 
                           :if     => lambda { |person| !person.new_watcher.nil? }
   fires :person_updated,  :on     => :update,
                           :if     => :fire?