aboutsummaryrefslogtreecommitdiffstats
path: root/lib/active_relation/predicates.rb
blob: e17a9f82fe40e7a05253c542071d9a814ebfbdc7 (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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
module ActiveRelation
  class Predicate
    def ==(other)
      self.class == other.class
    end
  end

  class Binary < Predicate
    attr_reader :operand1, :operand2

    def initialize(operand1, operand2)
      @operand1, @operand2 = operand1, operand2
    end

    def ==(other)
      super and @operand1 == other.operand1 and @operand2 == other.operand2
    end
    
    def bind(relation)
      descend { |x| x.bind(relation) }
    end
    
    def qualify
      descend(&:qualify)
    end

    def to_sql(formatter = nil)
      "#{operand2.format(operand1)} #{predicate_sql} #{operand1.format(operand2)}"
    end
    
    def descend
      self.class.new(yield(operand1), yield(operand2))
    end
  end

  class Equality < Binary
    def ==(other)
      self.class == other.class and
        ((operand1 == other.operand1 and operand2 == other.operand2) or
         (operand1 == other.operand2 and operand2 == other.operand1))
    end

    def predicate_sql
      operand2.equality_predicate_sql
    end
  end

  class GreaterThanOrEqualTo < Binary
    def predicate_sql; '>=' end
  end

  class GreaterThan < Binary
    def predicate_sql; '>' end
  end

  class LessThanOrEqualTo < Binary
    def predicate_sql; '<=' end
  end

  class LessThan < Binary
    def predicate_sql; '<' end
  end

  class Match < Binary
    alias_method :regexp, :operand2
  end
  
  class In < Binary
    def predicate_sql; operand2.inclusion_predicate_sql end
  end
end