arel 1.0.1 → 2.0.0

data/lib/arel/sql/engine.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Sql
+    class Engine
+      def self.new thing
+        #warn "#{caller.first} -- Engine will be removed"
+        thing
+      end
+    end
+  end
+end

data/lib/arel/sql_literal.rb

@@ -1,13 +1,4 @@
 module Arel
-  class SqlLiteral < String
-    def relation
-      nil
-    end
-
-    def to_sql(formatter = nil)
-      self
-    end
-
-    include Attribute::Expressions
+  class SqlLiteral < Nodes::SqlLiteral
   end
 end

data/lib/arel/table.rb

@@ -0,0 +1,126 @@
+module Arel
+  class Table
+    include Arel::Crud
+
+    @engine = nil
+    class << self; attr_accessor :engine; end
+
+    attr_accessor :name, :engine, :aliases, :table_alias
+
+    def initialize name, engine = Table.engine
+      @name    = name
+      @engine  = engine
+      @columns = nil
+      @aliases = []
+      @table_alias = nil
+      @primary_key = nil
+
+      if Hash === engine
+        @engine  = engine[:engine] || Table.engine
+        @columns = attributes_for engine[:columns]
+
+        # Sometime AR sends an :as parameter to table, to let the table know
+        # that it is an Alias.  We may want to override new, and return a
+        # TableAlias node?
+        @table_alias = engine[:as] unless engine[:as].to_s == name.to_s
+      end
+    end
+
+    def primary_key
+      @primary_key ||= begin
+        primary_key_name = @engine.connection.primary_key(name)
+        # some tables might be without primary key
+        primary_key_name && self[primary_key_name]
+      end
+    end
+
+    def alias
+      Nodes::TableAlias.new("#{name}_2", self).tap do |node|
+        @aliases << node
+      end
+    end
+
+    def from table
+      SelectManager.new(@engine, table)
+    end
+
+    def joins manager
+      nil
+    end
+
+    def join relation, klass = Nodes::InnerJoin
+      return from(self) unless relation
+
+      case relation
+      when String, Nodes::SqlLiteral
+        raise if relation.blank?
+        from Nodes::StringJoin.new(self, relation)
+      else
+        from klass.new(self, relation, nil)
+      end
+    end
+
+    def group *columns
+      from(self).group(*columns)
+    end
+
+    def order *expr
+      from(self).order(*expr)
+    end
+
+    def where condition
+      from(self).where condition
+    end
+
+    def project *things
+      from(self).project(*things)
+    end
+
+    def take amount
+      from(self).take amount
+    end
+
+    def having expr
+      from(self).having expr
+    end
+
+    def columns
+      @columns ||=
+        attributes_for @engine.connection.columns(@name, "#{@name} Columns")
+    end
+
+    def [] name
+      return nil unless table_exists?
+
+      name = name.to_sym
+      columns.find { |column| column.name == name }
+    end
+
+    def select_manager
+      SelectManager.new(@engine)
+    end
+
+    private
+
+    def attributes_for columns
+      return nil unless columns
+
+      columns.map do |column|
+        Attributes.for(column).new self, column.name.to_sym, column
+      end
+    end
+
+    def table_exists?
+      @table_exists ||= tables.key?(@name) || engine.connection.table_exists?(name)
+    end
+
+    def tables
+      self.class.table_cache(@engine)
+    end
+
+    @@table_cache = nil
+    def self.table_cache engine # :nodoc:
+      @@table_cache ||= Hash[engine.connection.tables.map { |x| [x,true] }]
+    end
+  end
+end

data/lib/arel/tree_manager.rb

@@ -0,0 +1,26 @@
+module Arel
+  class TreeManager
+    # FIXME: Remove this.
+    include Arel::Relation
+
+    attr_accessor :visitor
+
+    def initialize engine
+      @engine  = engine
+      @visitor = Visitors.visitor_for @engine
+    end
+
+    def to_dot
+      Visitors::Dot.new.accept @head
+    end
+
+    def to_sql
+      @visitor.accept @head
+    end
+
+    def initialize_copy other
+      super
+      @head = @head.clone
+    end
+  end
+end

data/lib/arel/update_manager.rb

@@ -0,0 +1,48 @@
+module Arel
+  class UpdateManager < Arel::TreeManager
+    def initialize engine
+      super
+      @head = Nodes::UpdateStatement.new
+    end
+
+    def take limit
+      @head.limit = limit
+      self
+    end
+
+    def order *expr
+      @head.orders = expr
+      self
+    end
+
+    ###
+    # UPDATE +table+
+    def table table
+      @head.relation = table
+      self
+    end
+
+    def wheres= exprs
+      @head.wheres = exprs
+    end
+
+    def where expr
+      @head.wheres << expr
+      self
+    end
+
+    def set values
+      if String === values
+        @head.values = [values]
+      else
+        @head.values = values.map { |column,value|
+          Nodes::Assignment.new(
+            Nodes::UnqualifiedColumn.new(column),
+            value
+          )
+        }
+      end
+      self
+    end
+  end
+end

data/lib/arel/visitors.rb

@@ -0,0 +1,30 @@
+require 'arel/visitors/to_sql'
+require 'arel/visitors/postgresql'
+require 'arel/visitors/mysql'
+require 'arel/visitors/oracle'
+require 'arel/visitors/join_sql'
+require 'arel/visitors/where_sql'
+require 'arel/visitors/order_clauses'
+require 'arel/visitors/dot'
+
+module Arel
+  module Visitors
+    VISITORS = {
+      'postgresql'      => Arel::Visitors::PostgreSQL,
+      'mysql'           => Arel::Visitors::MySQL,
+      'mysql2'          => Arel::Visitors::MySQL,
+      'oracle_enhanced' => Arel::Visitors::Oracle,
+    }
+
+    ENGINE_VISITORS = Hash.new do |hash, engine|
+      pool         = engine.connection_pool
+      adapter      = pool.spec.config[:adapter]
+      hash[engine] = (VISITORS[adapter] || Visitors::ToSql).new(engine)
+    end
+
+    def self.visitor_for engine
+      ENGINE_VISITORS[engine]
+    end
+    class << self; alias :for :visitor_for; end
+  end
+end

data/lib/arel/visitors/dot.rb

@@ -0,0 +1,233 @@
+module Arel
+  module Visitors
+    class Dot
+      class Node # :nodoc:
+        attr_accessor :name, :id, :fields
+
+        def initialize name, id, fields = []
+          @name   = name
+          @id     = id
+          @fields = fields
+        end
+      end
+
+      class Edge < Struct.new :name, :from, :to # :nodoc:
+      end
+
+      def initialize
+        @nodes      = []
+        @edges      = []
+        @node_stack = []
+        @edge_stack = []
+        @seen       = {}
+      end
+
+      def accept object
+        visit object
+        to_dot
+      end
+
+      private
+      def visit_Arel_Nodes_Grouping o
+        visit_edge o, "expr"
+      end
+
+      def visit_Arel_Nodes_Ordering o
+        visit_edge o, "expr"
+        visit_edge o, "direction"
+      end
+
+      def visit_Arel_Nodes_TableAlias o
+        visit_edge o, "name"
+        visit_edge o, "relation"
+        visit_edge o, "columns"
+      end
+
+      def visit_Arel_Nodes_Sum o
+        visit_edge o, "expressions"
+        visit_edge o, "alias"
+      end
+      alias :visit_Arel_Nodes_Max :visit_Arel_Nodes_Sum
+      alias :visit_Arel_Nodes_Avg :visit_Arel_Nodes_Sum
+
+      def visit_Arel_Nodes_Count o
+        visit_edge o, "expressions"
+        visit_edge o, "distinct"
+      end
+
+      def visit_Arel_Nodes_On o
+        visit_edge o, "expr"
+      end
+
+      def visit_Arel_Nodes_Values o
+        visit_edge o, "expressions"
+      end
+
+      def visit_Arel_Nodes_StringJoin o
+        visit_edge o, "left"
+        visit_edge o, "right"
+      end
+
+      def visit_Arel_Nodes_InnerJoin o
+        visit_edge o, "left"
+        visit_edge o, "right"
+        visit_edge o, "constraint"
+      end
+      alias :visit_Arel_Nodes_OuterJoin :visit_Arel_Nodes_InnerJoin
+
+      def visit_Arel_Nodes_DeleteStatement o
+        visit_edge o, "relation"
+        visit_edge o, "wheres"
+      end
+
+      def visit_Arel_Nodes_UnqualifiedColumn o
+        visit_edge o, "attribute"
+      end
+
+      def visit_Arel_Nodes_Offset o
+        visit_edge o, "value"
+      end
+
+      def visit_Arel_Nodes_InsertStatement o
+        visit_edge o, "relation"
+        visit_edge o, "columns"
+        visit_edge o, "values"
+      end
+
+      def visit_Arel_Nodes_SelectCore o
+        visit_edge o, "froms"
+        visit_edge o, "projections"
+        visit_edge o, "wheres"
+      end
+
+      def visit_Arel_Nodes_SelectStatement o
+        visit_edge o, "cores"
+        visit_edge o, "limit"
+        visit_edge o, "orders"
+        visit_edge o, "offset"
+      end
+
+      def visit_Arel_Nodes_UpdateStatement o
+        visit_edge o, "relation"
+        visit_edge o, "wheres"
+        visit_edge o, "values"
+      end
+
+      def visit_Arel_Table o
+        visit_edge o, "name"
+      end
+
+      def visit_Arel_Attribute o
+        visit_edge o, "relation"
+        visit_edge o, "name"
+      end
+      alias :visit_Arel_Attributes_Integer :visit_Arel_Attribute
+      alias :visit_Arel_Attributes_Float :visit_Arel_Attribute
+      alias :visit_Arel_Attributes_String :visit_Arel_Attribute
+      alias :visit_Arel_Attributes_Time :visit_Arel_Attribute
+      alias :visit_Arel_Attributes_Boolean :visit_Arel_Attribute
+      alias :visit_Arel_Attributes_Attribute :visit_Arel_Attribute
+
+      def visit_Arel_Nodes_Equality o
+        visit_edge o, "left"
+        visit_edge o, "right"
+      end
+      alias :visit_Arel_Nodes_And                :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_Or                 :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_NotEqual           :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_GreaterThan        :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_GreaterThanOrEqual :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_Assignment         :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_In                 :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_LessThan           :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_LessThanOrEqual    :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_Between            :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_NotIn              :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_DoesNotMatch       :visit_Arel_Nodes_Equality
+      alias :visit_Arel_Nodes_Matches            :visit_Arel_Nodes_Equality
+
+      def visit_String o
+        @node_stack.last.fields << o
+      end
+      alias :visit_Time :visit_String
+      alias :visit_Date :visit_String
+      alias :visit_DateTime :visit_String
+      alias :visit_NilClass :visit_String
+      alias :visit_TrueClass :visit_String
+      alias :visit_FalseClass :visit_String
+      alias :visit_Arel_SqlLiteral :visit_String
+      alias :visit_Fixnum :visit_String
+      alias :visit_BigDecimal :visit_String
+      alias :visit_Float :visit_String
+      alias :visit_Symbol :visit_String
+      alias :visit_Arel_Nodes_SqlLiteral :visit_String
+
+      def visit_Hash o
+        o.each_with_index do |pair, i|
+          edge("pair_#{i}")   { visit pair }
+        end
+      end
+
+      def visit_Array o
+        o.each_with_index do |x,i|
+          edge(i) { visit x }
+        end
+      end
+
+      def visit_edge o, method
+        edge(method) { visit o.send(method) }
+      end
+
+      def visit o
+        if node = @seen[o.object_id]
+          @edge_stack.last.to = node
+          return
+        end
+
+        node = Node.new(o.class.name, o.object_id)
+        @seen[node.id] = node
+        @nodes << node
+        with_node node do
+          send "visit_#{o.class.name.gsub('::', '_')}", o
+        end
+      end
+
+      def edge name
+        edge = Edge.new(name, @node_stack.last)
+        @edge_stack.push edge
+        @edges << edge
+        yield
+        @edge_stack.pop
+      end
+
+      def with_node node
+        if edge = @edge_stack.last
+          edge.to = node
+        end
+
+        @node_stack.push node
+        yield
+        @node_stack.pop
+      end
+
+      def quote string
+        string.to_s.gsub('"', '\"')
+      end
+
+      def to_dot
+        "digraph \"ARel\" {\nnode [width=0.375,height=0.25,shape=record];\n" +
+          @nodes.map { |node|
+            label = "<f0>#{node.name}"
+
+            node.fields.each_with_index do |field, i|
+              label << "|<f#{i + 1}>#{quote field}"
+            end
+
+            "#{node.id} [label=\"#{label}\"];"
+          }.join("\n") + "\n" + @edges.map { |edge|
+            "#{edge.from.id} -> #{edge.to.id} [label=\"#{edge.name}\"];"
+          }.join("\n") + "\n}"
+      end
+    end
+  end
+end