1 files changed, 112 insertions, 0 deletions
diff --git a/railties/lib/generator.rb b/railties/lib/generator.rb
new file mode 100644
index 0000000000..28b41c60f0
--- /dev/null
+++ b/railties/lib/generator.rb
@@ -0,0 +1,112 @@
+require 'fileutils'
+require 'active_record/support/inflector'
+
+module Generator
+  class GeneratorError < StandardError; end
+
+  class Base
+    @@template_root = File.dirname(__FILE__) + '/../generators/templates'
+    cattr_accessor :template_root
+
+    attr_reader :rails_root, :class_name, :file_name, :table_name,
+                :actions, :options
+
+    def initialize(rails_root, object_name, actions = [], options = {})
+      @rails_root = rails_root
+      @class_name = Inflector.camelize(object_name)
+      @file_name  = Inflector.underscore(@class_name)
+      @table_name = Inflector.pluralize(@file_name)
+      @actions    = actions
+      @options    = options
+
+      # Use local templates if rails_root/generators directory exists.
+      local_template_root = File.join(@rails_root, 'generators')
+      if File.directory?(local_template_root)
+        self.class.template_root = local_template_root
+      end
+    end
+
+    protected
+
+      # Generate a file in a fresh Rails app from an ERB template.
+      # Takes a template path relative to +template_root+, a
+      # destination path relative to +rails_root+, evaluates the template,
+      # and writes the result to the destination.
+      def generate_file(template_file_path, rails_file_path, eval_binding = nil)
+        # Determine full paths for source and destination files.
+        template_path = File.join(template_root, template_file_path)
+        rails_path    = File.join(rails_root, rails_file_path)
+
+        # Create destination directories.
+        FileUtils.mkdir_p(File.dirname(rails_path))
+
+        # Render template and write result.
+        eval_binding ||= binding
+        contents = ERB.new(File.read(template_path), nil, '-').result(eval_binding)
+        File.open(rails_path, 'w') { |file| file.write(contents) }
+      end
+  end
+
+  # Generate controller, helper, functional test, and views.
+  class Controller < Base
+    def generate
+      options[:scaffold] = file_name if options[:scaffold]
+
+      # Controller class.
+      generate_file "controller.erb", "app/controllers/#{file_name}_controller.rb"
+
+      # Helper class.
+      generate_file "helper.erb", "app/helpers/#{file_name}_helper.rb"
+
+      # Function test.
+      generate_file "controller_test.erb", "test/functional/#{file_name}_controller_test.rb"
+
+      # View template for each action.
+      @actions.each do |action|
+        generate_file "controller_view.rhtml",
+                      "app/views/#{file_name}/#{action}.rhtml",
+                      binding
+      end
+    end
+  end
+
+  # Generate model, unit test, and fixtures.
+  class Model < Base
+    def generate
+
+      # Model class.
+      generate_file "model.erb", "app/models/#{file_name}.rb"
+
+      # Model unit test.
+      generate_file "model_test.erb", "test/unit/#{file_name}_test.rb"
+
+      # Test fixtures directory.
+      FileUtils.mkdir_p("test/fixtures/#{table_name}")
+    end
+  end
+
+  # Generate mailer, helper, functional test, and views.
+  class Mailer < Base
+    def generate
+
+      # Mailer class.
+      generate_file "mailer.erb", "app/models/#{file_name}.rb"
+
+      # Mailer unit test.
+      generate_file "mailer_test.erb", "test/unit/#{file_name}_test.rb"
+
+      # Test fixtures directory.
+      FileUtils.mkdir_p("test/fixtures/#{table_name}")
+
+      # View template and fixture for each action.
+      @actions.each do |action|
+        generate_file "mailer_action.rhtml",
+                      "app/views/#{file_name}/#{action}.rhtml",
+                      binding
+        generate_file "mailer_fixture.rhtml",
+                      "test/fixtures/#{table_name}/#{action}",
+                      binding
+      end
+    end
+  end
+end