# frozen_string_literal: true

require 'activemodel/associations'
require 'eac_rails_utils/patches/rails_4'
require 'eac_rails_utils/patches/rails_5_2'

module EacRailsUtils
  module Models
    module TablelessAssociations
      module Hooks
        class << self
          def init
            init_rails_4 if ::EacRailsUtils::Patches::Rails4.enabled?
            init_rails_5_2 if ::EacRailsUtils::Patches::Rails52.enabled?
          end

          def init_rails_4
            ActiveSupport.on_load(:active_record) do
              ActiveRecord::Associations::AssociationScope.prepend(
                ::EacRailsUtils::Patches::Rails4::ActiveRecordAssociationsAssociationScope
              )
            end
          end

          def init_rails_5_2
            rails_5_2_fix_activemodel_associations_methods
          end

          def rails_5_2_fix_activemodel_associations_methods
            %i[belongs_to has_many].each do |method|
              ::EacRailsUtils::Patches::Rails52::ActiveModelAssociationMethodFix.new(method)
            end
          end
        end
      end
    end
  end
end