class ActiveRecord::TableMetadata

def associated_table(table_name)
  reflection = klass._reflect_on_association(table_name) || klass._reflect_on_association(table_name.singularize)
  if !reflection && table_name == arel_table.name
    return self
  end
  if reflection
    association_klass = reflection.klass unless reflection.polymorphic?
  elsif block_given?
    association_klass = yield table_name
  end
  if association_klass
    arel_table = association_klass.arel_table
    arel_table = arel_table.alias(table_name) if arel_table.name != table_name
    TableMetadata.new(association_klass, arel_table, reflection)
  else
    type_caster = TypeCaster::Connection.new(klass, table_name)
    arel_table = Arel::Table.new(table_name, type_caster: type_caster)
    TableMetadata.new(nil, arel_table, reflection)
  end
end

def associated_with?(table_name)
  klass&._reflect_on_association(table_name)
end

def has_column?(column_name)
  klass&.columns_hash&.key?(column_name)
end

def initialize(klass, arel_table, reflection = nil)
  @klass = klass
  @arel_table = arel_table
  @reflection = reflection
end

def polymorphic_association?
  reflection&.polymorphic?
end

def polymorphic_name_association
  reflection&.polymorphic_name
end

def predicate_builder
  if klass
    klass.predicate_builder.with(self)
  else
    PredicateBuilder.new(self)
  end
end

def primary_key
  klass&.primary_key
end

def reflect_on_aggregation(aggregation_name)
  klass&.reflect_on_aggregation(aggregation_name)
end

def through_association?
  reflection&.through_reflection?
end

def type(column_name)
  arel_table.type_for_attribute(column_name)
end