Add invoke_iff and invoke_for to the public interface.

railties/lib/generators/base.rb

@@ -44,80 +44,6 @@ def self.namespace(name=nil)
         @namespace ||= "#{base_name}:generators:#{generator_name}"
       end
 
-      protected
-
-        # Check whether the given class names are already taken by user
-        # application or Ruby on Rails.
-        #
-        def class_collisions(*class_names)
-          return unless behavior == :invoke
-
-          class_names.flatten.each do |class_name|
-            class_name = class_name.to_s
-            next if class_name.strip.empty?
-
-            # Split the class from its module nesting
-            nesting = class_name.split('::')
-            last_name = nesting.pop
-
-            # Hack to limit const_defined? to non-inherited on 1.9
-            extra = []
-            extra << false unless Object.method(:const_defined?).arity == 1
-
-            # Extract the last Module in the nesting
-            last = nesting.inject(Object) do |last, nest|
-              break unless last.const_defined?(nest, *extra)
-              last.const_get(nest)
-            end
-
-            if last && last.const_defined?(last_name.camelize, *extra)
-              raise Error, "The name '#{class_name}' is either already used in your application " <<
-                           "or reserved by Ruby on Rails. Please choose an alternative and run "  <<
-                           "this generator again."
-            end
-          end
-        end
-
-        # Use Rails default banner.
-        #
-        def self.banner
-          "#{$0} #{generator_name} #{self.arguments.map(&:usage).join(' ')} [options]"
-        end
-
-        # Sets the base_name taking into account the current class namespace.
-        #
-        def self.base_name
-          @base_name ||= self.name.split('::').first.underscore
-        end
-
-        # Removes the namespaces and get the generator name. For example,
-        # Rails::Generators::MetalGenerator will return "metal" as generator name.
-        #
-        def self.generator_name
-          @generator_name ||= begin
-            klass_name = self.name.split('::').last
-            klass_name.gsub!(/Generator$/, '')
-            klass_name.underscore
-          end
-        end
-
-        # Small macro to add ruby as an option to the generator with proper
-        # default value plus an instance helper method called shebang.
-        #
-        def self.add_shebang_option!
-          require 'rbconfig'
-          default = File.join(Config::CONFIG['bindir'], Config::CONFIG['ruby_install_name'])
-
-          class_option :ruby, :type => :string, :aliases => "-r", :default => default,
-                              :desc => "Path to the Ruby binary of your choice", :banner => "PATH"
-
-          no_tasks do
-            define_method :shebang do
-              "#!#{options[:ruby] || "/usr/bin/env ruby"}"
-            end
-          end
-        end
-
       # Invoke a generator based on the value supplied by the user to the
       # given option named "name". A class option is created when this method
       # is invoked and you can set a hash to customize it, although type and
@@ -229,6 +155,80 @@ def invoke_#{name}
         end
       end
 
+      protected
+
+        # Check whether the given class names are already taken by user
+        # application or Ruby on Rails.
+        #
+        def class_collisions(*class_names)
+          return unless behavior == :invoke
+
+          class_names.flatten.each do |class_name|
+            class_name = class_name.to_s
+            next if class_name.strip.empty?
+
+            # Split the class from its module nesting
+            nesting = class_name.split('::')
+            last_name = nesting.pop
+
+            # Hack to limit const_defined? to non-inherited on 1.9
+            extra = []
+            extra << false unless Object.method(:const_defined?).arity == 1
+
+            # Extract the last Module in the nesting
+            last = nesting.inject(Object) do |last, nest|
+              break unless last.const_defined?(nest, *extra)
+              last.const_get(nest)
+            end
+
+            if last && last.const_defined?(last_name.camelize, *extra)
+              raise Error, "The name '#{class_name}' is either already used in your application " <<
+                           "or reserved by Ruby on Rails. Please choose an alternative and run "  <<
+                           "this generator again."
+            end
+          end
+        end
+
+        # Use Rails default banner.
+        #
+        def self.banner
+          "#{$0} #{generator_name} #{self.arguments.map(&:usage).join(' ')} [options]"
+        end
+
+        # Sets the base_name taking into account the current class namespace.
+        #
+        def self.base_name
+          @base_name ||= self.name.split('::').first.underscore
+        end
+
+        # Removes the namespaces and get the generator name. For example,
+        # Rails::Generators::MetalGenerator will return "metal" as generator name.
+        #
+        def self.generator_name
+          @generator_name ||= begin
+            klass_name = self.name.split('::').last
+            klass_name.gsub!(/Generator$/, '')
+            klass_name.underscore
+          end
+        end
+
+        # Small macro to add ruby as an option to the generator with proper
+        # default value plus an instance helper method called shebang.
+        #
+        def self.add_shebang_option!
+          require 'rbconfig'
+          default = File.join(Config::CONFIG['bindir'], Config::CONFIG['ruby_install_name'])
+
+          class_option :ruby, :type => :string, :aliases => "-r", :default => default,
+                              :desc => "Path to the Ruby binary of your choice", :banner => "PATH"
+
+          no_tasks do
+            define_method :shebang do
+              "#!#{options[:ruby] || "/usr/bin/env ruby"}"
+            end
+          end
+        end
+
     end
   end
 end