class Countrizable::ActiveRecord::Migration::Migrator

Attributes

model[R]

Public Class Methods

new(model) click to toggle source
# File lib/countrizable/active_record/migration.rb, line 21
def initialize(model)
  @model = model
end

Public Instance Methods

add_country_value_fields() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 87
def add_country_value_fields
  connection.change_table(country_values_table_name) do |t|
    fields.each do |name, options|
      if options.is_a? Hash
        t.column name, options.delete(:type), options
      else
        t.column name, options
      end
    end
  end
end
add_country_value_fields!(fields, options = {}) click to toggle source
# File lib/countrizable/active_record/migration.rb, line 45
def add_country_value_fields!(fields, options = {})
  @fields = fields
  validate_country_fields
  add_country_value_fields
  clear_schema_cache!
  move_data_to_country_table if options[:migrate_data]
  remove_source_columns if options[:remove_source_columns]
  clear_schema_cache!
end
clear_schema_cache!() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 188
def clear_schema_cache!
  connection.schema_cache.clear! if connection.respond_to? :schema_cache
  model::CountryValue.reset_column_information
  model.reset_column_information
end
column_type(name) click to toggle source
# File lib/countrizable/active_record/migration.rb, line 168
def column_type(name)
  columns.detect { |c| c.name == name.to_s }.try(:type) || :string
end
complete_country_fields() click to toggle source

This adds all the current country attributes of the model It's a problem because in early migrations would add all the country attributes

# File lib/countrizable/active_record/migration.rb, line 73
def complete_country_fields
  country_attribute_names.each do |name|
    @fields[name] ||= column_type(name)
  end
end
country_value_country_code_index_name() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 180
def country_value_country_code_index_name
  truncate_index_name "index_#{country_values_table_name}_on_country_code"
end
country_value_index_name() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 176
def country_value_index_name
  truncate_index_name "index_#{country_values_table_name}_on_#{table_name.singularize}_id"
end
country_value_unique_index_name() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 184
def country_value_unique_index_name
  truncate_index_name "index_#{country_values_table_name}_on_#{table_name.singularize}_id_and_country_code"
end
create_country_value_table() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 79
def create_country_value_table
  connection.create_table(country_values_table_name) do |t|
    t.references table_name.sub(/^#{table_name_prefix}/, '').singularize, :null => false, :index => false, :type => column_type(model.primary_key).to_sym
    t.string :country_code, :null => false
    t.timestamps :null => false
  end
end
create_country_value_table!(fields = {}, options = {}) click to toggle source
# File lib/countrizable/active_record/migration.rb, line 29
def create_country_value_table!(fields = {}, options = {})
  extra = options.keys - [:migrate_data, :remove_source_columns, :unique_index]
  if extra.any?
    raise ArgumentError, "Unknown migration #{'option'.pluralize(extra.size)}: #{extra}"
  end
  @fields = fields
  # If we have fields we only want to create the country table with those fields
  complete_country_fields if fields.blank?
  validate_country_fields

  create_country_value_table
  add_country_value_fields!(fields, options)
  create_country_values_index(options)
  clear_schema_cache!
end
create_country_values_index(options) click to toggle source
# File lib/countrizable/active_record/migration.rb, line 99
def create_country_values_index(options)
  foreign_key = "#{table_name.sub(/^#{table_name_prefix}/, "").singularize}_id".to_sym
  connection.add_index(
    country_values_table_name,
    foreign_key,
    :name => country_value_index_name
  )
  # index for select('DISTINCT country_code') call in country_value.rb
  connection.add_index(
    country_values_table_name,
    :country_code,
    :name => country_value_country_code_index_name
  )

  if options[:unique_index]
    connection.add_index(
      country_values_table_name,
      [foreign_key, :country_code],
      :name => country_value_unique_index_name,
      unique: true
    )
  end
end
drop_country_table!(options = {}) click to toggle source
# File lib/countrizable/active_record/migration.rb, line 64
def drop_country_table!(options = {})
  move_data_to_model_table if options[:migrate_data]
  drop_country_values_index
  drop_country_value_table
  clear_schema_cache!
end
drop_country_value_table() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 123
def drop_country_value_table
  connection.drop_table(country_values_table_name)
end
drop_country_values_index() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 127
def drop_country_values_index
  if connection.indexes(country_values_table_name).map(&:name).include?(country_value_index_name)
    connection.remove_index(country_values_table_name, :name => country_value_index_name)
  end
  if connection.indexes(country_values_table_name).map(&:name).include?(country_value_country_code_index_name)
    connection.remove_index(country_values_table_name, :name => country_value_country_code_index_name)
  end
end
fields() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 25
def fields
  @fields ||= complete_country_fields
end
move_data_to_country_value_table() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 136
def move_data_to_country_value_table
  model.find_each do |record|
    country_value = record.country_value_for(I18n.country_code) || record.country_values.build(:country_code => I18n.country_code)
    fields.each do |attribute_name, attribute_type|
      country_value[attribute_name] = record.read_attribute(attribute_name, {:country_attributed => false})
    end
    country_value.save!
  end
end
move_data_to_model_table() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 146
def move_data_to_model_table
  add_missing_columns

  # Find all of the country attributes for all records in the model.
  all_country_attributes = model.all.collect{|m| m.attributes}
  all_country_attributes.each do |country_record|
    # Create a hash containing the country column names and their values.
    country_attribute_names.inject(fields_to_update={}) do |f, name|
      f.update({name.to_sym => country_record[name.to_s]})
    end

    # Now, update the actual model's record with the hash.
    model.where(model.primary_key.to_sym => country_record[model.primary_key]).update_all(fields_to_update)
  end
end
remove_source_columns() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 55
def remove_source_columns
  column_names = *fields.keys
  column_names.each do |column|
    if connection.column_exists?(table_name, column)
      connection.remove_column(table_name, column)
    end
  end
end
valid_field_name?(name) click to toggle source
# File lib/countrizable/active_record/migration.rb, line 172
def valid_field_name?(name)
  country_attribute_names.include?(name)
end
validate_country_fields() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 162
def validate_country_fields
  fields.each do |name, options|
    raise BadFieldName.new(name) unless valid_field_name?(name)
  end
end

Private Instance Methods

add_missing_columns() click to toggle source
# File lib/countrizable/active_record/migration.rb, line 204
def add_missing_columns
  clear_schema_cache!
  country_attribute_names.map(&:to_s).each do |attribute|
    unless model.column_names.include?(attribute)
      connection.add_column(table_name, attribute, model::CountryValue.columns_hash[attribute].type)
    end
  end
end
truncate_index_name(index_name) click to toggle source
# File lib/countrizable/active_record/migration.rb, line 196
def truncate_index_name(index_name)
  if index_name.size < connection.index_name_length
    index_name
  else
    "index_#{Digest::SHA1.hexdigest(index_name)}"[0, connection.index_name_length]
  end
end