square-arel 2.0.9.20110222133018

data/lib/arel/visitors.rb

@@ -0,0 +1,38 @@
+require 'arel/visitors/visitor'
+require 'arel/visitors/depth_first'
+require 'arel/visitors/to_sql'
+require 'arel/visitors/sqlite'
+require 'arel/visitors/postgresql'
+require 'arel/visitors/mysql'
+require 'arel/visitors/mssql'
+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,
+      'mssql'           => Arel::Visitors::MSSQL,
+      'sqlserver'       => Arel::Visitors::MSSQL,
+      'oracle_enhanced' => Arel::Visitors::Oracle,
+      'sqlite'          => Arel::Visitors::SQLite,
+      'sqlite3'         => Arel::Visitors::SQLite,
+    }
+
+    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/depth_first.rb

@@ -0,0 +1,154 @@
+module Arel
+  module Visitors
+    class DepthFirst < Arel::Visitors::Visitor
+      def initialize block = nil
+        @block = block || Proc.new
+      end
+
+      private
+
+      def visit o
+        super
+        @block.call o
+      end
+
+      def unary o
+        visit o.expr
+      end
+      alias :visit_Arel_Nodes_Group             :unary
+      alias :visit_Arel_Nodes_Grouping          :unary
+      alias :visit_Arel_Nodes_Having            :unary
+      alias :visit_Arel_Nodes_Limit             :unary
+      alias :visit_Arel_Nodes_Not               :unary
+      alias :visit_Arel_Nodes_Offset            :unary
+      alias :visit_Arel_Nodes_On                :unary
+      alias :visit_Arel_Nodes_Top               :unary
+      alias :visit_Arel_Nodes_UnqualifiedColumn :unary
+
+      def function o
+        visit o.expressions
+        visit o.alias
+      end
+      alias :visit_Arel_Nodes_Avg    :function
+      alias :visit_Arel_Nodes_Exists :function
+      alias :visit_Arel_Nodes_Max    :function
+      alias :visit_Arel_Nodes_Min    :function
+      alias :visit_Arel_Nodes_Sum    :function
+
+      def visit_Arel_Nodes_Count o
+        visit o.expressions
+        visit o.alias
+        visit o.distinct
+      end
+
+      def join o
+        visit o.left
+        visit o.right
+        visit o.constraint
+      end
+      alias :visit_Arel_Nodes_InnerJoin :join
+      alias :visit_Arel_Nodes_OuterJoin :join
+
+      def binary o
+        visit o.left
+        visit o.right
+      end
+      alias :visit_Arel_Nodes_And                :binary
+      alias :visit_Arel_Nodes_As                 :binary
+      alias :visit_Arel_Nodes_Assignment         :binary
+      alias :visit_Arel_Nodes_Between            :binary
+      alias :visit_Arel_Nodes_DeleteStatement    :binary
+      alias :visit_Arel_Nodes_DoesNotMatch       :binary
+      alias :visit_Arel_Nodes_Equality           :binary
+      alias :visit_Arel_Nodes_GreaterThan        :binary
+      alias :visit_Arel_Nodes_GreaterThanOrEqual :binary
+      alias :visit_Arel_Nodes_In                 :binary
+      alias :visit_Arel_Nodes_LessThan           :binary
+      alias :visit_Arel_Nodes_LessThanOrEqual    :binary
+      alias :visit_Arel_Nodes_Matches            :binary
+      alias :visit_Arel_Nodes_NotEqual           :binary
+      alias :visit_Arel_Nodes_NotIn              :binary
+      alias :visit_Arel_Nodes_Or                 :binary
+      alias :visit_Arel_Nodes_Ordering           :binary
+      alias :visit_Arel_Nodes_StringJoin         :binary
+      alias :visit_Arel_Nodes_TableAlias         :binary
+      alias :visit_Arel_Nodes_Values             :binary
+
+      def visit_Arel_Attribute o
+        visit o.relation
+        visit 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
+      alias :visit_Arel_Attributes_Decimal :visit_Arel_Attribute
+
+      def visit_Arel_Table o
+        visit o.name
+      end
+
+      def terminal o
+      end
+      alias :visit_ActiveSupport_Multibyte_Chars :terminal
+      alias :visit_ActiveSupport_StringInquirer  :terminal
+      alias :visit_Arel_Nodes_Lock               :terminal
+      alias :visit_Arel_Nodes_Node               :terminal
+      alias :visit_Arel_Nodes_SqlLiteral         :terminal
+      alias :visit_Arel_SqlLiteral               :terminal
+      alias :visit_BigDecimal                    :terminal
+      alias :visit_Bignum                        :terminal
+      alias :visit_Class                         :terminal
+      alias :visit_Date                          :terminal
+      alias :visit_DateTime                      :terminal
+      alias :visit_FalseClass                    :terminal
+      alias :visit_Fixnum                        :terminal
+      alias :visit_Float                         :terminal
+      alias :visit_NilClass                      :terminal
+      alias :visit_String                        :terminal
+      alias :visit_Symbol                        :terminal
+      alias :visit_Time                          :terminal
+      alias :visit_TrueClass                     :terminal
+
+      def visit_Arel_Nodes_InsertStatement o
+        visit o.relation
+        visit o.columns
+        visit o.values
+      end
+
+      def visit_Arel_Nodes_SelectCore o
+        visit o.projections
+        visit o.froms
+        visit o.wheres
+        visit o.groups
+        visit o.having
+      end
+
+      def visit_Arel_Nodes_SelectStatement o
+        visit o.cores
+        visit o.orders
+        visit o.limit
+        visit o.lock
+        visit o.offset
+      end
+
+      def visit_Arel_Nodes_UpdateStatement o
+        visit o.relation
+        visit o.values
+        visit o.wheres
+        visit o.orders
+        visit o.limit
+      end
+
+      def visit_Array o
+        o.each { |i| visit i }
+      end
+
+      def visit_Hash o
+        o.each { |k,v| visit(k); visit(v) }
+      end
+    end
+  end
+end

data/lib/arel/visitors/dot.rb

@@ -0,0 +1,230 @@
+module Arel
+  module Visitors
+    class Dot < Arel::Visitors::Visitor
+      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
+        super
+        to_dot
+      end
+
+      private
+      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_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 unary o
+        visit_edge o, "expr"
+      end
+      alias :visit_Arel_Nodes_Group             :unary
+      alias :visit_Arel_Nodes_Grouping          :unary
+      alias :visit_Arel_Nodes_Having            :unary
+      alias :visit_Arel_Nodes_Limit             :unary
+      alias :visit_Arel_Nodes_Not               :unary
+      alias :visit_Arel_Nodes_Offset            :unary
+      alias :visit_Arel_Nodes_On                :unary
+      alias :visit_Arel_Nodes_Top               :unary
+      alias :visit_Arel_Nodes_UnqualifiedColumn :unary
+
+      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
+          super
+        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

data/lib/arel/visitors/join_sql.rb

@@ -0,0 +1,40 @@
+module Arel
+  module Visitors
+    ###
+    # This class produces SQL for JOIN clauses but omits the "single-source"
+    # part of the Join grammar:
+    #
+    #   http://www.sqlite.org/syntaxdiagrams.html#join-source
+    #
+    # This visitor is used in SelectManager#join_sql and is for backwards
+    # compatibility with Arel V1.0
+    class JoinSql < Arel::Visitors::ToSql
+      private
+
+      def visit_Arel_Nodes_SelectCore o
+        [o.froms].grep(Nodes::Join).map { |x| visit x }.join ', '
+      end
+
+      def visit_Arel_Nodes_StringJoin o
+        [
+          (visit o.left if Nodes::Join === o.left),
+          visit(o.right)
+        ].compact.join ' '
+      end
+
+      def visit_Arel_Nodes_OuterJoin o
+        [
+          (visit o.left if Nodes::Join === o.left),
+          "LEFT OUTER JOIN #{visit o.right} #{visit o.constraint if o.constraint}"
+        ].compact.join ' '
+      end
+
+      def visit_Arel_Nodes_InnerJoin o
+        [
+          (visit o.left if Nodes::Join === o.left),
+          "INNER JOIN #{visit o.right} #{visit o.constraint if o.constraint}"
+        ].compact.join ' '
+      end
+    end
+  end
+end