diff options
author | David Heinemeier Hansson <david@loudthinking.com> | 2004-11-24 01:04:44 +0000 |
---|---|---|
committer | David Heinemeier Hansson <david@loudthinking.com> | 2004-11-24 01:04:44 +0000 |
commit | db045dbbf60b53dbe013ef25554fd013baf88134 (patch) | |
tree | 257830e3c76458c8ff3d1329de83f32b23926028 /railties/lib/generator.rb | |
download | rails-db045dbbf60b53dbe013ef25554fd013baf88134.tar.gz rails-db045dbbf60b53dbe013ef25554fd013baf88134.tar.bz2 rails-db045dbbf60b53dbe013ef25554fd013baf88134.zip |
Initial
git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@4 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
Diffstat (limited to 'railties/lib/generator.rb')
-rw-r--r-- | railties/lib/generator.rb | 112 |
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 |