2010-05-17 20:43:06 +00:00
|
|
|
require 'active_support/time'
|
2011-04-10 16:52:42 +00:00
|
|
|
require 'active_support/core_ext/object/inclusion'
|
2010-05-17 20:43:06 +00:00
|
|
|
|
2009-06-23 16:12:37 +00:00
|
|
|
module Rails
|
|
|
|
module Generators
|
|
|
|
class GeneratedAttribute
|
2011-08-17 09:16:04 +00:00
|
|
|
attr_accessor :name, :type, :has_index, :attr_options
|
2009-06-23 16:12:37 +00:00
|
|
|
|
2011-08-17 09:16:04 +00:00
|
|
|
def initialize(column_definition)
|
|
|
|
parse column_definition
|
2009-06-23 16:12:37 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def field_type
|
|
|
|
@field_type ||= case type
|
2011-05-16 14:43:05 +00:00
|
|
|
when :integer then :number_field
|
|
|
|
when :float, :decimal then :text_field
|
|
|
|
when :time then :time_select
|
|
|
|
when :datetime, :timestamp then :datetime_select
|
|
|
|
when :date then :date_select
|
|
|
|
when :text then :text_area
|
|
|
|
when :boolean then :check_box
|
2009-06-23 16:12:37 +00:00
|
|
|
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)
|
2011-03-05 21:25:20 +00:00
|
|
|
when :string then name == "type" ? "" : "MyString"
|
2009-06-23 16:12:37 +00:00
|
|
|
when :text then "MyText"
|
|
|
|
when :boolean then false
|
2009-07-13 14:18:56 +00:00
|
|
|
when :references, :belongs_to then nil
|
2009-06-23 16:12:37 +00:00
|
|
|
else
|
|
|
|
""
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-07-01 16:33:22 +00:00
|
|
|
def human_name
|
|
|
|
name.to_s.humanize
|
|
|
|
end
|
|
|
|
|
2009-06-23 16:12:37 +00:00
|
|
|
def reference?
|
2011-04-12 16:04:40 +00:00
|
|
|
self.type.in?([:references, :belongs_to])
|
2009-06-23 16:12:37 +00:00
|
|
|
end
|
2011-08-17 09:16:04 +00:00
|
|
|
|
|
|
|
def has_index?
|
|
|
|
@has_index
|
|
|
|
end
|
|
|
|
|
|
|
|
def has_uniq_index?
|
|
|
|
@has_uniq_index
|
|
|
|
end
|
|
|
|
|
|
|
|
def parse(column_definition)
|
|
|
|
name, type, has_index = column_definition.split(':')
|
|
|
|
# if user provided "name:index" instead of "name:string:index" type should be set blank
|
|
|
|
# so GeneratedAttribute's constructor could set it to :string
|
|
|
|
if type =~ /index|uniq|unique/i
|
|
|
|
has_index = type
|
|
|
|
type = nil
|
|
|
|
end
|
|
|
|
type = :string if type.blank?
|
|
|
|
|
|
|
|
@name = name
|
|
|
|
@type, @attr_options = *parse_type_and_options(type)
|
|
|
|
@has_index = ['index','uniq','unique'].include?(has_index)
|
|
|
|
@has_uniq_index = ['uniq','unique'].include?(has_index)
|
|
|
|
end
|
|
|
|
|
|
|
|
# parse possible attribute options like :limit for string/text/binary/integer or :precision/:scale for decimals
|
|
|
|
# when declaring options curly brackets should be used
|
|
|
|
def parse_type_and_options(type)
|
|
|
|
attribute_options = case type
|
|
|
|
when /(string|text|binary|integer){(\d+)}/
|
|
|
|
{:limit => $2.to_i}
|
|
|
|
when /decimal{(\d+),(\d+)}/
|
|
|
|
{:precision => $1.to_i, :scale => $2.to_i}
|
|
|
|
else; {}
|
|
|
|
end
|
|
|
|
[type.to_s.gsub(/{.*}/,'').to_sym, attribute_options]
|
|
|
|
end
|
|
|
|
|
|
|
|
def inject_options
|
|
|
|
@attr_options.blank? ? '' : ", #{@attr_options.to_s.gsub(/[{}]/, '')}"
|
|
|
|
end
|
|
|
|
|
|
|
|
def inject_index_options
|
|
|
|
has_uniq_index? ? ", :unique => true" : ''
|
|
|
|
end
|
2009-06-23 16:12:37 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|