From a292cd32226234a63fea0f1de48c5f38d4307fd1 Mon Sep 17 00:00:00 2001 From: David Heinemeier Hansson Date: Tue, 19 Sep 2006 08:47:31 +0000 Subject: Implemented ./script/generate scaffold_resource comment body:text created_at:datetime active:boolean price:decimal git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@5147 5ecf4fe2-1ee6-0310-87b1-e25e094e27de --- .../scaffold_resource_generator.rb | 129 ++++++++++++--------- 1 file changed, 74 insertions(+), 55 deletions(-) (limited to 'railties/lib/rails_generator/generators/components/scaffold_resource/scaffold_resource_generator.rb') diff --git a/railties/lib/rails_generator/generators/components/scaffold_resource/scaffold_resource_generator.rb b/railties/lib/rails_generator/generators/components/scaffold_resource/scaffold_resource_generator.rb index 462070591a..919f682193 100644 --- a/railties/lib/rails_generator/generators/components/scaffold_resource/scaffold_resource_generator.rb +++ b/railties/lib/rails_generator/generators/components/scaffold_resource/scaffold_resource_generator.rb @@ -1,4 +1,41 @@ class ScaffoldResourceGenerator < Rails::Generator::NamedBase + class ScaffoldAttribute + attr_accessor :name, :type, :column + + def initialize(name, type) + @name, @type = name, type.to_sym + @column = ActiveRecord::ConnectionAdapters::Column.new(name, nil, @type) + end + + def field_type + @field_type ||= case type + when :integer, :float, :decimal then :text_field + when :datetime, :timestamp, :time then :datetime_select + when :date then :date_select + when :string then :text_field + when :text then :text_area + when :boolean then :check_box + else + :text_field + end + end + + def default + @default ||= case type + when :integer then 1 + when :float then 1.5 + when :decimal then "9.99" + when :datetime, :timestamp, :time then Time.now.to_s(:db) + when :date then Date.today.to_s(:db) + when :string then "MyString" + when :text then "MyText" + when :boolean then false + else + "" + end + end + end + attr_reader :controller_name, :controller_class_path, :controller_file_path, @@ -28,68 +65,44 @@ class ScaffoldResourceGenerator < Rails::Generator::NamedBase def manifest recorded_session = record do |m| # Check for class naming collisions. - m.class_collisions controller_class_path, "#{controller_class_name}Controller", - "#{controller_class_name}Helper" - m.class_collisions class_path, "#{class_name}" + m.class_collisions(controller_class_path, "#{controller_class_name}Controller", "#{controller_class_name}Helper") + m.class_collisions(class_path, "#{class_name}") # Controller, helper, views, and test directories. - m.directory File.join('app/models', class_path) - m.directory File.join('app/controllers', controller_class_path) - m.directory File.join('app/helpers', controller_class_path) - m.directory File.join('app/views', controller_class_path, controller_file_name) - m.directory File.join('test/functional', controller_class_path) - m.directory File.join('test/unit', class_path) - - scaffold_views.each do |action| - m.template "view_#{action}.rhtml", - File.join('app/views', - controller_class_path, - controller_file_name, - "#{action}.rhtml"), - :assigns => { :action => action } + m.directory(File.join('app/models', class_path)) + m.directory(File.join('app/controllers', controller_class_path)) + m.directory(File.join('app/helpers', controller_class_path)) + m.directory(File.join('app/views', controller_class_path, controller_file_name)) + m.directory(File.join('test/functional', controller_class_path)) + m.directory(File.join('test/unit', class_path)) + + for action in scaffold_views + m.template( + "view_#{action}.rhtml", + File.join('app/views', controller_class_path, controller_file_name, "#{action}.rhtml") + ) end - m.template 'model.rb', - File.join('app/models', - class_path, - "#{file_name}.rb") - - m.template 'controller.rb', - File.join('app/controllers', - controller_class_path, - "#{controller_file_name}_controller.rb") + m.template('model.rb', File.join('app/models', class_path, "#{file_name}.rb")) - m.template 'functional_test.rb', - File.join('test/functional', - controller_class_path, - "#{controller_file_name}_controller_test.rb") + m.template( + 'controller.rb', File.join('app/controllers', controller_class_path, "#{controller_file_name}_controller.rb") + ) - m.template 'helper.rb', - File.join('app/helpers', - controller_class_path, - "#{controller_file_name}_helper.rb") - - m.template 'unit_test.rb', - File.join('test/unit', - class_path, - "#{file_name}_test.rb") - - m.template 'fixtures.yml', - File.join('test/fixtures', - "#{table_name}.yml") + m.template('functional_test.rb', File.join('test/functional', controller_class_path, "#{controller_file_name}_controller_test.rb")) + m.template('helper.rb', File.join('app/helpers', controller_class_path, "#{controller_file_name}_helper.rb")) + m.template('unit_test.rb', File.join('test/unit', class_path, "#{file_name}_test.rb")) + m.template('fixtures.yml', File.join('test/fixtures', "#{table_name}.yml")) unless options[:skip_migration] - m.migration_template 'migration.rb', 'db/migrate', :assigns => { - :migration_name => "Create#{class_name.pluralize.gsub(/::/, '')}" - }, :migration_file_name => "create_#{file_path.gsub(/\//, '_').pluralize}" - end - - # View template for each action. - [:index, :new, :edit, :show].each do |action| - path = File.join('app/views', class_path, table_name, "#{action}.rhtml") - m.template 'view.rhtml', - path, - :assigns => { :action => action, :path => path } + m.migration_template( + 'migration.rb', 'db/migrate', + :assigns => { + :migration_name => "Create#{class_name.pluralize.gsub(/::/, '')}", + :attributes => attributes + }, + :migration_file_name => "create_#{file_path.gsub(/\//, '_').pluralize}" + ) end end @@ -118,4 +131,10 @@ class ScaffoldResourceGenerator < Rails::Generator::NamedBase def model_name class_name.demodulize end -end + + def attributes + @attributes ||= args.collect do |attribute| + ScaffoldAttribute.new(*attribute.split(":")) + end + end +end \ No newline at end of file -- cgit v1.2.3