Add GeneratedAttribute.

1 files changed, 11 insertions, 0 deletions

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.
         #