# frozen_string_literal: true
require "mobility/backends/active_record/query_methods"

module Mobility
  module Backends
    module ActiveRecord
      class KeyValue::QueryMethods < QueryMethods
        def initialize(attributes, association_name: nil, class_name: nil, **)
          super
          @association_name = association_name

          define_join_method(association_name, class_name)
          define_query_methods(association_name)
        end

        def extended(relation)
          super
          association_name = @association_name
          q = self

          mod = Module.new do
            define_method :not do |opts, *rest|
              if i18n_keys = q.extract_attributes(opts)
                opts = opts.with_indifferent_access
                i18n_keys.each do |attr|
                  opts["#{attr}_#{association_name}"] = { value: q.collapse(opts.delete(attr)) }
                end
                super(opts, *rest).send(:"join_#{association_name}", *i18n_keys)
              else
                super(opts, *rest)
              end
            end
          end
          relation.mobility_where_chain.include(mod)
        end

        private

        def define_join_method(association_name, translation_class)
          define_method :"join_#{association_name}" do |*attributes, **options|
            attributes.inject(self) do |relation, attribute|
              t = translation_class.arel_table.alias("#{attribute}_#{association_name}")
              m = arel_table
              join_type = options[:outer_join] ? Arel::Nodes::OuterJoin : Arel::Nodes::InnerJoin
              relation.joins(m.join(t, join_type).
                             on(t[:key].eq(attribute).
                                and(t[:locale].eq(Mobility.locale).
                                    and(t[:translatable_type].eq(base_class.name).
                                        and(t[:translatable_id].eq(m[:id]))))).join_sources)
            end
          end
        end

        def define_query_methods(association_name)
          q = self

          define_method :where! do |opts, *rest|
            if i18n_keys = q.extract_attributes(opts)
              opts = opts.with_indifferent_access
              i18n_nulls = i18n_keys.reject { |key| opts[key] && [*opts[key]].all? }
              i18n_keys.each do |attr|
                opts["#{attr}_#{association_name}"] = { value: q.collapse(opts.delete(attr)) }
              end
              super(opts, *rest).
                send("join_#{association_name}", *(i18n_keys - i18n_nulls)).
                send("join_#{association_name}", *i18n_nulls, outer_join: true)
            else
              super(opts, *rest)
            end
          end
        end
      end
      KeyValue.private_constant :QueryMethods
    end
  end
end