aboutsummaryrefslogtreecommitdiffstats
path: root/lib/arel/tree_manager.rb
blob: 21a52d8a6006112072ae043210c8771a3843301a (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
module Arel
  class TreeManager
    include Arel::FactoryMethods

    attr_reader :ast, :engine

    def initialize engine
      @engine = engine
      @ctx    = nil
    end

    def to_dot
      Visitors::Dot.new.accept @ast
    end

    def visitor
      engine.connection.visitor
    end

    def to_sql
      visitor.accept @ast
    end

    def initialize_copy other
      super
      @ast = @ast.clone
    end

    def where expr
      if Arel::TreeManager === expr
        expr = expr.ast
      end
      @ctx.wheres << expr
      self
    end
  end
end