class ActiveRecord::ConnectionAdapters::AlterTable

def add_check_constraint(expression, options)
  @check_constraint_adds << @td.new_check_constraint_definition(expression, options)
end

def add_column(name, type, **options)
  name = name.to_s
  type = type.to_sym
  @adds << AddColumnDefinition.new(@td.new_column_definition(name, type, **options))
end

def add_foreign_key(to_table, options)
  @foreign_key_adds << @td.new_foreign_key_definition(to_table, options)
end

def drop_check_constraint(constraint_name)
  @check_constraint_drops << constraint_name
end

def drop_foreign_key(name)
  @foreign_key_drops << name
end

def initialize(td)
  @td   = td
  @adds = []
  @foreign_key_adds = []
  @foreign_key_drops = []
  @check_constraint_adds = []
  @check_constraint_drops = []
end

def name; @td.name; end