# frozen_string_literal: true

if defined?(::ActiveRecord)
  ::ActiveRecord::Associations::Builder::HasOne.class_eval do
    # Based on
    # https://github.com/rails/rails/blob/master/activerecord/lib/active_record/associations/builder/collection_association.rb#L50
    # https://github.com/rails/rails/blob/master/activerecord/lib/active_record/associations/builder/singular_association.rb#L11
    def self.define_accessors(mixin, reflection)
      super
      name = reflection.name
      mixin.class_eval <<-CODE, __FILE__, __LINE__ + 1
        def #{name}_id
          # if an attribute is already defined with this methods name we should just use it
          return read_attribute(__method__) if has_attribute?(__method__)
          association(:#{name}).reader.try(:id)
        end
      CODE
    end
  end
end