42 lines
830 B
Ruby
42 lines
830 B
Ruby
# frozen_string_literal: true
|
|
require 'arel/collectors/sql_string'
|
|
|
|
module Arel
|
|
class TreeManager
|
|
include Arel::FactoryMethods
|
|
|
|
attr_reader :ast, :engine
|
|
|
|
attr_accessor :bind_values
|
|
|
|
def initialize
|
|
@ctx = nil
|
|
@bind_values = []
|
|
end
|
|
|
|
def to_dot
|
|
collector = Arel::Collectors::PlainString.new
|
|
collector = Visitors::Dot.new.accept @ast, collector
|
|
collector.value
|
|
end
|
|
|
|
def to_sql engine = Table.engine
|
|
collector = Arel::Collectors::SQLString.new
|
|
collector = engine.connection.visitor.accept @ast, collector
|
|
collector.value
|
|
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
|