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'
|
2011-12-24 09:38:19 +00:00
|
|
|
require 'active_support/core_ext/object/blank'
|
2010-05-17 20:43:06 +00:00
|
|
|
|
2009-06-23 16:12:37 +00:00
|
|
|
module Rails
|
|
|
|
module Generators
|
|
|
|
class GeneratedAttribute
|
2012-01-14 13:35:57 +00:00
|
|
|
INDEX_OPTIONS = %w(index uniq)
|
|
|
|
UNIQ_INDEX_OPTIONS = %w(uniq)
|
|
|
|
|
2011-12-24 09:38:19 +00:00
|
|
|
attr_accessor :name, :type
|
|
|
|
attr_reader :attr_options
|
2009-06-23 16:12:37 +00:00
|
|
|
|
2011-12-24 09:38:19 +00: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 12:49:31 +00:00
|
|
|
has_index, type = type, nil if INDEX_OPTIONS.include?(type)
|
2011-12-24 09:38:19 +00:00
|
|
|
|
|
|
|
type, attr_options = *parse_type_and_options(type)
|
|
|
|
new(name, type, has_index, attr_options)
|
|
|
|
end
|
|
|
|
|
|
|
|
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 11:53:27 +00:00
|
|
|
when /(string|text|binary|integer)\{(\d+)\}/
|
2011-12-24 09:38:19 +00:00
|
|
|
return $1, :limit => $2.to_i
|
2012-01-30 21:13:04 +00:00
|
|
|
when /decimal\{(\d+)[,.-](\d+)\}/
|
2012-01-30 21:54:33 +00:00
|
|
|
return :decimal, :precision => $1.to_i, :scale => $2.to_i
|
2011-12-24 09:38:19 +00:00
|
|
|
else
|
|
|
|
return type, {}
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def initialize(name, type=nil, index_type=false, attr_options={})
|
|
|
|
@name = name
|
|
|
|
@type = (type.presence || :string).to_sym
|
2012-01-14 13:35:57 +00:00
|
|
|
@has_index = INDEX_OPTIONS.include?(index_type)
|
|
|
|
@has_uniq_index = UNIQ_INDEX_OPTIONS.include?(index_type)
|
2011-12-24 09:38:19 +00:00
|
|
|
@attr_options = attr_options
|
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
|
|
|
|
|
2011-12-24 09:38:19 +00:00
|
|
|
def index_name
|
|
|
|
reference? ? "#{name}_id" : name
|
|
|
|
end
|
|
|
|
|
2009-06-23 16:12:37 +00:00
|
|
|
def reference?
|
2012-01-14 13:35:57 +00:00
|
|
|
self.type.in?(:references, :belongs_to)
|
2009-06-23 16:12:37 +00:00
|
|
|
end
|
2011-12-24 09:38:19 +00:00
|
|
|
|
2011-08-17 09:16:04 +00:00
|
|
|
def has_index?
|
|
|
|
@has_index
|
|
|
|
end
|
|
|
|
|
|
|
|
def has_uniq_index?
|
|
|
|
@has_uniq_index
|
|
|
|
end
|
|
|
|
|
|
|
|
def inject_options
|
2011-12-24 21:06:25 +00:00
|
|
|
"".tap { |s| @attr_options.each { |k,v| s << ", #{k}: #{v.inspect}" } }
|
2011-08-17 09:16:04 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def inject_index_options
|
2011-12-24 21:06:25 +00:00
|
|
|
has_uniq_index? ? ", unique: true" : ""
|
2011-08-17 09:16:04 +00:00
|
|
|
end
|
2009-06-23 16:12:37 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|