Add GeneratedAttribute.

2 files changed, 55 insertions, 0 deletions

diff --git a/railties/lib/generator/generated_attribute.rb b/railties/lib/generator/generated_attribute.rb
new file mode 100644
index 0000000000..0ead945281
--- /dev/null
+++ b/railties/lib/generator/generated_attribute.rb
@@ -0,0 +1,44 @@
+module Rails
+  module Generators
+    class GeneratedAttribute
+      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
+
+      def reference?
+        [ :references, :belongs_to ].include?(self.type)
+      end
+    end
+  end
+end
diff --git a/railties/lib/generator/named_base.rb b/railties/lib/generator/named_base.rb
index c35667576a..6035d71060 100644
--- a/railties/lib/generator/named_base.rb
+++ b/railties/lib/generator/named_base.rb
@@ -1,4 +1,5 @@
 require 'generator/base'
+require 'generator/generated_attribute'
 
 module Rails
   module Generators
@@ -7,11 +8,13 @@ module Rails
 
       attr_reader :class_name, :singular_name, :plural_name, :table_name,
                   :class_path, :file_path, :class_nesting, :class_nesting_depth
+
       alias :file_name :singular_name
 
       def initialize(*args)
         super
         assign_names!
+        parse_attributes! if respond_to?(:attributes)
       end
 
       protected
@@ -35,6 +38,14 @@ module Rails
           end
         end
 
+        # Convert attributes hash into an array with GeneratedAttribute objects.
+        #
+        def parse_attributes!
+          attributes.map! do |name, type|
+            Rails::Generator::GeneratedAttribute.new(name, type)
+          end
+        end
+
         # Extract modules from filesystem-style or ruby-style path. Both
         # good/fun/stuff and Good::Fun::Stuff produce the same results.
         #