module Mobility
  module Backends
    class ActiveRecord::Hstore::QueryMethods < ActiveRecord::QueryMethods
      def initialize(attributes, _)
        super
        attributes_extractor = @attributes_extractor

        define_method :where! do |opts, *rest|
          if i18n_keys = attributes_extractor.call(opts)
            m = arel_table
            locale = Arel::Nodes.build_quoted(Mobility.locale.to_s)
            opts = opts.with_indifferent_access
            infix = Arel::Nodes::InfixOperation

            i18n_query = i18n_keys.inject(nil) { |ops, attr|
              column = m[attr.to_sym]
              value = opts.delete(attr)

              op =
                if value.nil?
                  infix.new(:'?', column, locale).not
                else
                  infix.new(:'->', m[attr.to_sym], locale).eq(value)
                end
              ops ? ops.and(op) : op
            }

            opts.empty? ? super(i18n_query) : super(opts, *rest).where(i18n_query)
          else
            super(opts, *rest)
          end
        end
      end

      def extended(relation)
        super
        attributes_extractor = @attributes_extractor
        m = relation.model.arel_table

        mod = Module.new do
          define_method :not do |opts, *rest|
            if i18n_keys = attributes_extractor.call(opts)
              locale = Arel::Nodes.build_quoted(Mobility.locale.to_s)
              opts = opts.with_indifferent_access
              infix = Arel::Nodes::InfixOperation

              i18n_query = i18n_keys.inject(nil) { |ops, attr|
                column = m[attr.to_sym]
                value = Arel::Nodes.build_quoted(opts.delete(attr).to_s)
                has_key = infix.new(:'?', column, locale)
                not_eq_value = infix.new(:'->', column, locale).not_eq(value)
                op = has_key.and(not_eq_value)
                ops ? ops.and(op) : op
              }

              super(opts, *rest).where(i18n_query)
            else
              super(opts, *rest)
            end
          end
        end
        relation.mobility_where_chain.include(mod)
      end
    end
  end
end