Since Rails will require Ruby >= 1.8.7, we can rely on instance_exec.

rails · Jan 19, 2010 · e1029be · e1029be
1 parent e6a68a5
commit e1029be
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 9 deletions.
diff --git a/railties/lib/generators/rails/plugin/plugin_generator.rb b/railties/lib/generators/rails/plugin/plugin_generator.rb
@@ -20,15 +20,15 @@ def create_tasks_files
         directory 'tasks', plugin_dir('tasks')
       end
 
-      hook_for :generator do |instance, generator|
-        instance.inside instance.send(:plugin_dir), :verbose => true do
-          instance.invoke generator, [ instance.name ], :namespace => false
+      hook_for :generator do |generator|
+        inside plugin_dir, :verbose => true do
+          invoke generator, [ name ], :namespace => false
         end
       end
 
-      hook_for :test_framework do |instance, test_framework|
-        instance.inside instance.send(:plugin_dir), :verbose => true do
-          instance.invoke test_framework
+      hook_for :test_framework do |test_framework|
+        inside plugin_dir, :verbose => true do
+          invoke test_framework
         end
       end
 

diff --git a/railties/lib/rails/vendor/thor-0.12.3/lib/thor/group.rb b/railties/lib/rails/vendor/thor-0.12.3/lib/thor/group.rb
@@ -252,10 +252,13 @@ def _invoke_for_class_method(klass, task=nil, *args, &block) #:nodoc:
       shell.padding += 1
 
       result = if block_given?
-        if block.arity == 2
-          block.call(self, klass)
-        else
+        case block.arity
+        when 3
           block.call(self, klass, task)
+        when 2
+          block.call(self, klass)
+        when 1
+          instance_exec(klass, &block)
         end
       else
         invoke klass, task, *args