1 files changed, 93 insertions, 0 deletions
diff --git a/railties/lib/generators/base.rb b/railties/lib/generators/base.rb
new file mode 100644
index 0000000000..b530713805
--- /dev/null
+++ b/railties/lib/generators/base.rb
@@ -0,0 +1,93 @@
+require 'generators/actions'
+
+module Rails
+  module Generators
+    class Error < Thor::Error
+    end
+
+    class Base < Thor::Group
+      include Rails::Generators::Actions
+      include Thor::Actions
+
+      # Automatically sets the source root based on the class name.
+      #
+      def self.source_root
+        @source_root ||= File.expand_path(File.join(File.dirname(__FILE__), base_name, generator_name, 'templates'))
+      end
+
+      # Convenience method to get the namespace from the class name.
+      #
+      def self.namespace(name=nil)
+        if name
+          super
+        else
+          @namespace ||= "#{base_name}:#{generator_name}"
+        end
+      end
+
+      protected
+
+        # Use Rails default banner.
+        #
+        def self.banner
+          "#{$0} #{generator_name} #{self.arguments.map(&:usage).join(' ')} [options]"
+        end
+
+        # Sets the base_name. Overwriten by test unit generators.
+        #
+        def self.base_name
+          'rails'
+        end
+
+        # Removes the namespaces and get the generator name. For example,
+        # Rails::Generators::MetalGenerator will return "metal" as generator name.
+        #
+        # The name is used to set the namespace (in this case "rails:metal")
+        # and to set the source root ("rails/metal/templates").
+        #
+        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"
+
+          no_tasks do
+            define_method :shebang do
+              "#!#{options[:ruby] || "/usr/bin/env ruby"}"
+            end
+          end
+        end
+
+        # Small macro to add test_framework option and invoke it.
+        #
+        def self.add_test_framework_option!
+          # TODO Reduce the example name
+          class_option :test_framework, :type => :string, :aliases => "-t", :default => "test_unit",
+                                        :desc => "Test framework to be invoked by this generator"
+
+          define_method :invoke_test_framework do
+            return unless options[:test_framework]
+            name = "#{options[:test_framework]}:#{self.class.generator_name}"
+
+            begin
+              invoke name
+            rescue Thor::UndefinedTaskError # TODO Ensure this message is called.
+              say "Could not find and/or invoke #{name}."
+            end
+          end
+        end
+    end
+  end
+end