class ActiveRecord::Reflection::BelongsToReflection

def association_class
  if polymorphic?
    Associations::BelongsToPolymorphicAssociation
  else
    Associations::BelongsToAssociation
  end
end

def association_primary_key(klass = nil)
  if primary_key = options[:primary_key]
    @association_primary_key ||= if primary_key.is_a?(Array)
      primary_key.map { |pk| pk.to_s.freeze }.freeze
    else
      -primary_key.to_s
    end
  elsif (klass || self.klass).has_query_constraints? || options[:query_constraints]
    (klass || self.klass).composite_query_constraints_list
  elsif (klass || self.klass).composite_primary_key?
    # If klass has composite primary key of shape [:<tenant_key>, :id], infer primary_key as :id
    primary_key = (klass || self.klass).primary_key
    primary_key.include?("id") ? "id" : primary_key
  else
    primary_key(klass || self.klass)
  end
end

def belongs_to?; true; end

def can_find_inverse_of_automatically?(*)
  !polymorphic? && super
end

def join_foreign_key
  foreign_key
end

def join_foreign_type
  foreign_type
end

def join_primary_key(klass = nil)
  polymorphic? ? association_primary_key(klass) : association_primary_key
end

def macro; :belongs_to; end