require 'everywhere/util'

module ActiveRecord
  class PredicateBuilder
    if ActiveRecord::VERSION::STRING > '3.1'
      class << self
        include Everywhere::Util

        def build_from_hash_with_not_and_like_and_not_like(engine, attributes, default_table)
          attributes_with_not_and_like_and_not_like = attributes.map do |column, value|
            # {key: {not: value}}
            if value.is_a?(Hash) && (value.keys.size == 1) && value.keys.first.in?([:not, :like, :not_like])
              ["#{column}__#{value.keys.first}__", value.values.first]
            else
              [column, value]
            end
          end
          build_from_hash_without_not_and_like_and_not_like(engine, attributes_with_not_and_like_and_not_like, default_table).map do |rel|
            if rel.left.name.to_s.ends_with?('__not__')
              rel.left.name = rel.left.name.to_s.sub(/__not__$/, '').to_sym
              negate rel
            elsif rel.left.name.to_s.ends_with?('__like__')
              rel.left.name = rel.left.name.to_s.sub(/__like__$/, '').to_sym
              Arel::Nodes::Matches.new rel.left, rel.right
            elsif rel.left.name.to_s.ends_with?('__not_like__')
              rel.left.name = rel.left.name.to_s.sub(/__not_like__$/, '').to_sym
              Arel::Nodes::DoesNotMatch.new rel.left, rel.right
            else
              rel
            end
          end
        end
        alias_method_chain :build_from_hash, :not_and_like_and_not_like
      end
    else
      include Everywhere::Util

      def build_from_hash_with_not_and_like_and_not_like(attributes, default_table)
        attributes_with_not_and_like_and_not_like = attributes.map do |column, value|
          # {key: {not: value}}
          if value.is_a?(Hash) && (value.keys.size == 1) && ((value.keys.first == :not) || (value.keys.first == :like) || (value.keys.first == :not_like))
            ["#{column}__#{value.keys.first}__", value.values.first]
          else
            [column, value]
          end
        end
        build_from_hash_without_not_and_like_and_not_like(attributes_with_not_and_like_and_not_like, default_table).map do |rel|
          if rel.left.name.to_s.ends_with?('__not__')
            rel.left.name = rel.left.name.to_s.sub(/__not__$/, '').to_sym
            negate rel
          elsif rel.left.name.to_s.ends_with?('__like__')
            rel.left.name = rel.left.name.to_s.sub(/__like__$/, '').to_sym
            Arel::Nodes::Matches.new rel.left, rel.right
          elsif rel.left.name.to_s.ends_with?('__not_like__')
            rel.left.name = rel.left.name.to_s.sub(/__not_like__$/, '').to_sym
            Arel::Nodes::DoesNotMatch.new rel.left, rel.right
          else
            rel
          end
        end
      end
      alias_method_chain :build_from_hash, :not_and_like_and_not_like
    end
  end
end