2010-05-17 16:43:06 -04:00
|
|
|
require 'active_support/time'
|
|
|
|
|
2009-06-23 12:12:37 -04:00
|
|
|
module Rails
|
|
|
|
module Generators
|
|
|
|
class GeneratedAttribute
|
2012-01-14 08:35:57 -05:00
|
|
|
INDEX_OPTIONS = %w(index uniq)
|
|
|
|
UNIQ_INDEX_OPTIONS = %w(uniq)
|
|
|
|
|
2011-12-24 04:38:19 -05:00
|
|
|
attr_accessor :name, :type
|
|
|
|
attr_reader :attr_options
|
2009-06-23 12:12:37 -04:00
|
|
|
|
2011-12-24 04:38:19 -05:00
|
|
|
class << self
|
|
|
|
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
|
2012-01-15 07:49:31 -05:00
|
|
|
has_index, type = type, nil if INDEX_OPTIONS.include?(type)
|
2011-12-24 04:38:19 -05:00
|
|
|
|
|
|
|
type, attr_options = *parse_type_and_options(type)
|
2012-04-22 20:53:18 -04:00
|
|
|
type = type.to_sym if type
|
2012-03-02 20:59:23 -05:00
|
|
|
|
2012-04-22 20:53:18 -04:00
|
|
|
if type && reference?(type)
|
|
|
|
references_index = UNIQ_INDEX_OPTIONS.include?(has_index) ? { :unique => true } : true
|
|
|
|
attr_options[:index] = references_index
|
2012-04-17 16:46:44 -04:00
|
|
|
end
|
2012-03-02 20:59:23 -05:00
|
|
|
|
2011-12-24 04:38:19 -05:00
|
|
|
new(name, type, has_index, attr_options)
|
|
|
|
end
|
|
|
|
|
2012-04-22 20:53:18 -04:00
|
|
|
def reference?(type)
|
|
|
|
[:references, :belongs_to].include? type
|
|
|
|
end
|
|
|
|
|
2011-12-24 04:38:19 -05:00
|
|
|
private
|
|
|
|
|
|
|
|
# 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)
|
|
|
|
case type
|
2011-12-24 06:53:27 -05:00
|
|
|
when /(string|text|binary|integer)\{(\d+)\}/
|
2011-12-24 04:38:19 -05:00
|
|
|
return $1, :limit => $2.to_i
|
2012-01-30 16:13:04 -05:00
|
|
|
when /decimal\{(\d+)[,.-](\d+)\}/
|
2012-01-30 16:54:33 -05:00
|
|
|
return :decimal, :precision => $1.to_i, :scale => $2.to_i
|
2011-12-24 04:38:19 -05:00
|
|
|
else
|
|
|
|
return type, {}
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def initialize(name, type=nil, index_type=false, attr_options={})
|
|
|
|
@name = name
|
2012-04-22 20:53:18 -04:00
|
|
|
@type = type || :string
|
2012-01-14 08:35:57 -05:00
|
|
|
@has_index = INDEX_OPTIONS.include?(index_type)
|
|
|
|
@has_uniq_index = UNIQ_INDEX_OPTIONS.include?(index_type)
|
2011-12-24 04:38:19 -05:00
|
|
|
@attr_options = attr_options
|
2009-06-23 12:12:37 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def field_type
|
|
|
|
@field_type ||= case type
|
2011-05-16 10:43:05 -04: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 12:12:37 -04: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 16:25:20 -05:00
|
|
|
when :string then name == "type" ? "" : "MyString"
|
2009-06-23 12:12:37 -04:00
|
|
|
when :text then "MyText"
|
|
|
|
when :boolean then false
|
2009-07-13 10:18:56 -04:00
|
|
|
when :references, :belongs_to then nil
|
2009-06-23 12:12:37 -04:00
|
|
|
else
|
|
|
|
""
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2009-07-01 12:33:22 -04:00
|
|
|
def human_name
|
|
|
|
name.to_s.humanize
|
|
|
|
end
|
|
|
|
|
2011-12-24 04:38:19 -05:00
|
|
|
def index_name
|
|
|
|
reference? ? "#{name}_id" : name
|
|
|
|
end
|
|
|
|
|
2009-06-23 12:12:37 -04:00
|
|
|
def reference?
|
2012-04-22 20:53:18 -04:00
|
|
|
self.class.reference?(type)
|
2009-06-23 12:12:37 -04:00
|
|
|
end
|
2011-12-24 04:38:19 -05:00
|
|
|
|
2011-08-17 05:16:04 -04:00
|
|
|
def has_index?
|
|
|
|
@has_index
|
|
|
|
end
|
|
|
|
|
|
|
|
def has_uniq_index?
|
|
|
|
@has_uniq_index
|
|
|
|
end
|
|
|
|
|
|
|
|
def inject_options
|
2011-12-24 16:06:25 -05:00
|
|
|
"".tap { |s| @attr_options.each { |k,v| s << ", #{k}: #{v.inspect}" } }
|
2011-08-17 05:16:04 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
def inject_index_options
|
2011-12-24 16:06:25 -05:00
|
|
|
has_uniq_index? ? ", unique: true" : ""
|
2011-08-17 05:16:04 -04:00
|
|
|
end
|
2009-06-23 12:12:37 -04:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|