Class: Rails::Generators::GeneratedAttribute

Inherits:
Object
  • Object
show all
Defined in:
lib/rails/generators/generated_attribute.rb

Overview

:nodoc:

Constant Summary collapse

INDEX_OPTIONS =
%w(index uniq)
UNIQ_INDEX_OPTIONS =
%w(uniq)

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(name, type = nil, index_type = false, attr_options = {}) ⇒ GeneratedAttribute

Returns a new instance of GeneratedAttribute.


59
60
61
62
63
64
65
# File 'lib/rails/generators/generated_attribute.rb', line 59

def initialize(name, type=nil, index_type=false, attr_options={})
  @name           = name
  @type           = type || :string
  @has_index      = INDEX_OPTIONS.include?(index_type)
  @has_uniq_index = UNIQ_INDEX_OPTIONS.include?(index_type)
  @attr_options   = attr_options
end

Instance Attribute Details

#attr_optionsObject (readonly)

Returns the value of attribute attr_options


10
11
12
# File 'lib/rails/generators/generated_attribute.rb', line 10

def attr_options
  @attr_options
end

#index_nameObject


109
110
111
112
113
114
115
# File 'lib/rails/generators/generated_attribute.rb', line 109

def index_name
  @index_name ||= if polymorphic?
    %w(id type).map { |t| "#{name}_#{t}" }
  else
    column_name
  end
end

#nameObject

Returns the value of attribute name


9
10
11
# File 'lib/rails/generators/generated_attribute.rb', line 9

def name
  @name
end

#typeObject

Returns the value of attribute type


9
10
11
# File 'lib/rails/generators/generated_attribute.rb', line 9

def type
  @type
end

Class Method Details

.parse(column_definition) ⇒ Object


14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# File 'lib/rails/generators/generated_attribute.rb', line 14

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
  has_index, type = type, nil if INDEX_OPTIONS.include?(type)

  type, attr_options = *parse_type_and_options(type)
  type = type.to_sym if type

  if type && reference?(type)
    if UNIQ_INDEX_OPTIONS.include?(has_index)
      attr_options[:index] = { unique: true }
    end
  end

  new(name, type, has_index, attr_options)
end

.reference?(type) ⇒ Boolean

Returns:

  • (Boolean)

34
35
36
# File 'lib/rails/generators/generated_attribute.rb', line 34

def reference?(type)
  [:references, :belongs_to].include? type
end

Instance Method Details

#column_nameObject


117
118
119
# File 'lib/rails/generators/generated_attribute.rb', line 117

def column_name
  @column_name ||= reference? ? "#{name}_id" : name
end

#defaultObject


81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# File 'lib/rails/generators/generated_attribute.rb', line 81

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)
    when :string                      then name == "type" ? "" : "MyString"
    when :text                        then "MyText"
    when :boolean                     then false
    when :references, :belongs_to     then nil
    else
      ""
  end
end

#field_typeObject


67
68
69
70
71
72
73
74
75
76
77
78
79
# File 'lib/rails/generators/generated_attribute.rb', line 67

def field_type
  @field_type ||= case type
    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
    else
      :text_field
  end
end

#foreign_key?Boolean

Returns:

  • (Boolean)

121
122
123
# File 'lib/rails/generators/generated_attribute.rb', line 121

def foreign_key?
  !!(name =~ /_id$/)
end

#has_index?Boolean

Returns:

  • (Boolean)

137
138
139
# File 'lib/rails/generators/generated_attribute.rb', line 137

def has_index?
  @has_index
end

#has_uniq_index?Boolean

Returns:

  • (Boolean)

141
142
143
# File 'lib/rails/generators/generated_attribute.rb', line 141

def has_uniq_index?
  @has_uniq_index
end

#human_nameObject


105
106
107
# File 'lib/rails/generators/generated_attribute.rb', line 105

def human_name
  name.humanize
end

#inject_index_optionsObject


157
158
159
# File 'lib/rails/generators/generated_attribute.rb', line 157

def inject_index_options
  has_uniq_index? ? ", unique: true" : ""
end

#inject_optionsObject


153
154
155
# File 'lib/rails/generators/generated_attribute.rb', line 153

def inject_options
  "".tap { |s| options_for_migration.each { |k,v| s << ", #{k}: #{v.inspect}" } }
end

#options_for_migrationObject


161
162
163
164
165
166
167
168
169
170
171
172
# File 'lib/rails/generators/generated_attribute.rb', line 161

def options_for_migration
  @attr_options.dup.tap do |options|
    if required?
      options.delete(:required)
      options[:null] = false
    end

    if reference? && !polymorphic?
      options[:foreign_key] = true
    end
  end
end

#password_digest?Boolean

Returns:

  • (Boolean)

145
146
147
# File 'lib/rails/generators/generated_attribute.rb', line 145

def password_digest?
  name == 'password' && type == :digest
end

#plural_nameObject


97
98
99
# File 'lib/rails/generators/generated_attribute.rb', line 97

def plural_name
  name.sub(/_id$/, '').pluralize
end

#polymorphic?Boolean

Returns:

  • (Boolean)

129
130
131
# File 'lib/rails/generators/generated_attribute.rb', line 129

def polymorphic?
  self.attr_options[:polymorphic]
end

#reference?Boolean

Returns:

  • (Boolean)

125
126
127
# File 'lib/rails/generators/generated_attribute.rb', line 125

def reference?
  self.class.reference?(type)
end

#required?Boolean

Returns:

  • (Boolean)

133
134
135
# File 'lib/rails/generators/generated_attribute.rb', line 133

def required?
  self.attr_options[:required]
end

#singular_nameObject


101
102
103
# File 'lib/rails/generators/generated_attribute.rb', line 101

def singular_name
  name.sub(/_id$/, '').singularize
end

#token?Boolean

Returns:

  • (Boolean)

149
150
151
# File 'lib/rails/generators/generated_attribute.rb', line 149

def token?
  type == :token
end