migration_helpers.rb in stockor-0.5.0

- old
+ new

@@ -41,11 +41,11 @@
             def skr_reference( to_table, *args )
                 options = args.extract_options!
 
                 options[:column] ||= to_table.to_s + '_id'
 
-                column( options[:column], :integer, :null=>options[:null] || false )
+                column( options[:column], (options[:type] || :integer), :null=>options[:null] || false )
                 to_table = options[:to_table] if options.has_key? :to_table
 
                 if options[:single]
                     to_table = to_table.to_s.pluralize
                 end
@@ -73,10 +73,10 @@
                 end
                 definition.skr_foreign_keys.each do |to_table, options |
                     skr_add_foreign_key( table_name, to_table, options )
                 end
                 definition.skr_extra_indexes.each do | index_column, options |
-                    skr_add_index( table_name, index_column, options )
+                    skr_add_index(table_name, index_column, options )
                 end
             end
 
             def skr_add_index( table_name, columns, options={} )
                 table_name = Skr.config.table_prefix + table_name.to_s