Module: Code0::ZeroTrack::Database::MigrationHelpers::AddColumnEnhancements

Included in:
Code0::ZeroTrack::Database::Migration::V1_0
Defined in:
lib/code0/zero_track/database/migration_helpers/add_column_enhancements.rb

Instance Method Summary collapse

Instance Method Details

#add_column(table_name, column_name, type, *args, **kwargs, &block) ⇒ Object



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# File 'lib/code0/zero_track/database/migration_helpers/add_column_enhancements.rb', line 8

def add_column(table_name, column_name, type, *args, **kwargs, &block)
  helper_context = self

  limit = kwargs.delete(:limit)
  unique = kwargs.delete(:unique)

  super

  return unless type == :text

  quoted_column_name = helper_context.quote_column_name(column_name)

  if limit
    name = helper_context.send(:text_limit_name, table_name, column_name)

    definition = "char_length(#{quoted_column_name}) <= #{limit}"

    add_check_constraint(table_name, definition, name: name)
  end

  if unique.is_a?(Hash)
    unique[:where] = "#{column_name} IS NOT NULL" if unique.delete(:allow_nil_duplicate)
    column_name = "LOWER(#{quoted_column_name})" if unique.delete(:case_insensitive)

    add_index table_name, column_name, unique: true, **unique
  elsif unique
    add_index table_name, column_name, unique: unique
  end
end