1 files changed, 70 insertions, 0 deletions
diff --git a/railties/lib/generators/base.rb b/railties/lib/generators/base.rb
new file mode 100644
index 0000000000..5f0b3eb369
--- /dev/null
+++ b/railties/lib/generators/base.rb
@@ -0,0 +1,70 @@
+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__), 'rails', generator_name, 'templates'))
+      end
+
+      # Convenience method to get the namespace from the class name.
+      #
+      def self.namespace(name=nil)
+        if name
+          super
+        else
+          @namespace ||= "rails:generators:#{generator_name}"
+        end
+      end
+
+    protected
+
+      # Use Rails default banner.
+      #
+      def self.banner
+        "#{$0} #{generator_name} #{self.arguments.map(&:usage).join(' ')} [options]"
+      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:generators:metal")
+      # and to set the source root ("generators/metal/templates").
+      #
+      def self.generator_name
+        @generator_name ||= begin
+          klass_name = self.name.gsub(/^Rails::Generators::/, '')
+          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"
+
+        class_eval do
+          protected
+          def shebang
+            "#!#{options[:ruby] || "/usr/bin/env ruby"}"
+          end
+        end
+      end
+
+    end
+  end
+end