Class: Rails::Generators::GeneratedAttribute

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

def parse(column_definition)
  name, type, index_type = 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
  index_type, type = type, nil if valid_index_type?(type)

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

  if dangerous_name?(name)
    raise Error, "Could not generate field '#{name}', as it is already defined by Active Record."
  end

  if type && !valid_type?(type)
    raise Error, "Could not generate field '#{name}' with unknown type '#{type}'."
  end

  if index_type && !valid_index_type?(index_type)
    raise Error, "Could not generate field '#{name}' with unknown index '#{index_type}'."
  end

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

  new(name, type, index_type, attr_options)
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 90

def parse_type_and_options(type)
  case type
  when /(text|binary)\{([a-z]+)\}/
    parsed_type, parsed_options = $1, { size: $2.to_sym }
  when /(string|text|binary|integer)\{(\d+)\}/
    parsed_type, parsed_options = $1, { limit: $2.to_i }
  when /decimal\{(\d)[,.-](\d)\}/
    parsed_type, parsed_options = :decimal, { precision: $1.to_i, scale: $2.to_i }
  when /(references|belongs_to)\{(.+)\}/
    parsed_type = $1
    provided_options = $2.split(/[,.-]/)
    parsed_options = Hash[provided_options.map { |opt| [opt.to_sym, true] }]
  else
    parsed_type, parsed_options = type&.remove("!"), {}
  end

  if type&.ends_with?("!")
    parsed_options[:null] = false
  end

  return parsed_type, parsed_options
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 83

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

# File 'railties/lib/rails/generators/generated_attribute.rb', line 79

def valid_index_type?(index_type)
  INDEX_OPTIONS.include?(index_type.to_s)
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 73

def valid_type?(type)
  DEFAULT_TYPES.include?(type.to_s) ||
    !defined?(ActiveRecord::Base) ||
    ActiveRecord::Base.connection_db_config.adapter_class.valid_type?(type)
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 220

def attachments?
  type == :attachments
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 32

attr_reader   :attr_options

# File 'railties/lib/rails/generators/generated_attribute.rb', line 180

def foreign_key?
  name.end_with?("_id")
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 196

def has_index?
  @has_index
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 200

def has_uniq_index?
  @has_uniq_index
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 168

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

# File 'railties/lib/rails/generators/generated_attribute.rb', line 33

attr_writer   :index_name

# File 'railties/lib/rails/generators/generated_attribute.rb', line 31

attr_accessor :name, :type

# File 'railties/lib/rails/generators/generated_attribute.rb', line 204

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

# File 'railties/lib/rails/generators/generated_attribute.rb', line 188

def polymorphic?
  attr_options[:polymorphic]
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 184

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

# File 'railties/lib/rails/generators/generated_attribute.rb', line 192

def required?
  reference? && Rails.application.config.active_record.belongs_to_required_by_default
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 212

def rich_text?
  type == :rich_text
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 208

def token?
  type == :token
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 31

attr_accessor :name, :type

# File 'railties/lib/rails/generators/generated_attribute.rb', line 224

def virtual?
  rich_text? || attachment? || attachments?
end

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

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_fs(:db)
               when :date                        then Date.today.to_fs(:db)
               when :string                      then name == "type" ? "" : "MyString"
               when :text                        then "MyText"
               when :boolean                     then false
               when :references, :belongs_to,
                    :attachment, :attachments,
                    :rich_text                   then nil
               else
                 ""
  end
end

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

def field_type
  @field_type ||= case type
                  when :integer                  then :number_field
                  when :float, :decimal          then :text_field
                  when :time                     then :time_field
                  when :datetime, :timestamp     then :datetime_field
                  when :date                     then :date_field
                  when :text                     then :textarea
                  when :rich_text                then :rich_textarea
                  when :boolean                  then :checkbox
                  when :attachment, :attachments then :file_field
                  else
                    :text_field
  end
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 164

def human_name
  name.humanize
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 232

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

# File 'railties/lib/rails/generators/generated_attribute.rb', line 228

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

# File 'railties/lib/rails/generators/generated_attribute.rb', line 236

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

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

# File 'railties/lib/rails/generators/generated_attribute.rb', line 156

def plural_name
  name.delete_suffix("_id").pluralize
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 259

def print_attribute_options
  if attr_options.empty?
    ""
  elsif attr_options[:size]
    "{#{attr_options[:size]}}"
  elsif attr_options[:limit]
    "{#{attr_options[:limit]}}"
  elsif attr_options[:precision] && attr_options[:scale]
    "{#{attr_options[:precision]},#{attr_options[:scale]}}"
  else
    "{#{attr_options.keys.join(",")}}"
  end
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 160

def singular_name
  name.delete_suffix("_id").singularize
end

# File 'railties/lib/rails/generators/generated_attribute.rb', line 248

def to_s
  if has_uniq_index?
    "#{name}:#{type}#{print_attribute_options}:uniq"
  elsif has_index?
    "#{name}:#{type}#{print_attribute_options}:index"
  else
    "#{name}:#{type}#{print_attribute_options}"
  end
end