module Arel::TreeManager::StatementMethods

def key
  @ast.key
end

def key=(key)
  @ast.key = Nodes.build_quoted(key)
end

def offset(offset)
  @ast.offset = Nodes::Offset.new(Nodes.build_quoted(offset)) if offset
  self
end

def order(*expr)
  @ast.orders = expr
  self
end

def take(limit)
  @ast.limit = Nodes::Limit.new(Nodes.build_quoted(limit)) if limit
  self
end

def where(expr)
  @ast.wheres << expr
  self
end

def wheres=(exprs)
  @ast.wheres = exprs
end