aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/relation/predicate_builder.rb
blob: 9e855209f91385a375bd791b4acf36d1369a2703 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
module ActiveRecord
  class PredicateBuilder

    def initialize(engine)
      @engine = engine
    end

    def build_from_hash(attributes, default_table)
      predicates = attributes.map do |column, value|
        table = default_table

        if value.is_a?(Hash)
          table = Arel::Table.new(column, :engine => @engine)
          build_from_hash(value, table)
        else
          column = column.to_s

          if column.include?('.')
            table_name, column = column.split('.', 2)
            table = Arel::Table.new(table_name, :engine => @engine)
          end

          attribute = table[column] || Arel::Attribute.new(table, column.to_sym)

          case value
          when Array, ActiveRecord::Associations::AssociationCollection, ActiveRecord::NamedScope::Scope
            values = value.to_a
            values.any? ? attribute.in(values) : attribute.eq(nil)
          when Range
            # TODO : Arel should handle ranges with excluded end.
            if value.exclude_end?
              [attribute.gteq(value.begin), attribute.lt(value.end)]
            else
              attribute.in(value)
            end
          else
            attribute.eq(value)
          end
        end
      end

      predicates.flatten
    end

  end
end