arel_toolkit 0.3.0 → 0.4.0

data/lib/arel/extensions/current_role.rb

@@ -13,6 +13,10 @@ module Arel
         collector << 'current_role'
       end
     end
+
+    class Dot
+      alias visit_Arel_Nodes_CurrentRole terminal
+    end
   end
 end
 

data/lib/arel/extensions/current_row.rb

@@ -0,0 +1,7 @@
+module Arel
+  module Visitors
+    class Dot
+      alias visit_Arel_Nodes_CurrentRow terminal
+    end
+  end
+end

data/lib/arel/extensions/current_schema.rb

@@ -13,6 +13,10 @@ module Arel
         collector << 'current_schema'
       end
     end
+
+    class Dot
+      alias visit_Arel_Nodes_CurrentSchema terminal
+    end
   end
 end
 

data/lib/arel/extensions/current_user.rb

@@ -13,6 +13,10 @@ module Arel
         collector << 'current_user'
       end
     end
+
+    class Dot
+      alias visit_Arel_Nodes_CurrentUser terminal
+    end
   end
 end
 

data/lib/arel/extensions/dealocate.rb

@@ -0,0 +1,31 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class Dot
+      def visit_Arel_Nodes_Dealocate(o)
+        visit_edge o, 'name'
+      end
+    end
+
+    class ToSql
+      def visit_Arel_Nodes_Dealocate(o, collector)
+        collector << 'DEALLOCATE ' << (o.name || 'ALL')
+      end
+    end
+  end
+
+  module Nodes
+    class Dealocate < Node
+      attr_reader :name
+
+      def initialize(name)
+        @name = name
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/default_values.rb

@@ -14,6 +14,10 @@ module Arel
         collector << 'DEFAULT VALUES'
       end
     end
+
+    class Dot
+      alias visit_Arel_Nodes_DefaultValues terminal
+    end
   end
 end
 

data/lib/arel/extensions/delete_manager.rb

@@ -1,9 +1,25 @@
-Arel::DeleteManager.class_eval do
-  def ==(other)
-    @ast == other.ast && @ctx == other.ctx
-  end
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  class DeleteManager < Arel::TreeManager
+    def ==(other)
+      other.is_a?(self.class) && @ast == other.ast && @ctx == other.ctx
+    end
+
+    protected
 
-  protected
+    attr_reader :ctx
+  end
 
-  attr_reader :ctx
+  module Visitors
+    class Dot
+      def visit_Arel_DeleteManager(o)
+        visit_edge o, 'ast'
+      end
+    end
+  end
 end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/delete_statement.rb

@@ -3,18 +3,21 @@
 
 module Arel
   module Nodes
-    # https://www.postgresql.org/docs/9.5/sql-insert.html
-    Arel::Nodes::DeleteStatement.class_eval do
-      attr_accessor :using
-      attr_accessor :with
-      attr_accessor :returning
+    # https://www.postgresql.org/docs/10/sql-delete.html
+    class DeleteStatement
+      module DeleteStatementExtension
+        attr_accessor :using
+        attr_accessor :with
+        attr_accessor :returning
 
-      alias_method :old_initialize, :initialize
-      def initialize(relation = nil, wheres = [])
-        old_initialize(relation, wheres)
+        def initialize(relation = nil, wheres = [])
+          super
 
-        @returning = []
+          @returning = []
+        end
       end
+
+      prepend DeleteStatementExtension
     end
   end
 
@@ -49,6 +52,20 @@ module Arel
       end
       # rubocop:enable Metrics/AbcSize
     end
+
+    class Dot
+      module DeleteStatementExtension
+        def visit_Arel_Nodes_DeleteStatement(o)
+          super
+
+          visit_edge o, 'using'
+          visit_edge o, 'with'
+          visit_edge o, 'returning'
+        end
+      end
+
+      prepend(DeleteStatementExtension)
+    end
   end
 end
 

data/lib/arel/extensions/dot.rb

@@ -0,0 +1,11 @@
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class Dot
+      def terminal(_o); end
+    end
+  end
+end
+
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/extract_from.rb

@@ -4,14 +4,7 @@
 module Arel
   module Nodes
     # https://www.postgresql.org/docs/9.1/functions-datetime.html#FUNCTIONS-DATETIME-EXTRACT
-    class ExtractFrom < Arel::Nodes::Unary
-      attr_reader :field
-
-      def initialize(expr, field)
-        super(expr)
-
-        @field = field
-      end
+    class ExtractFrom < Arel::Nodes::Binary
     end
   end
 
@@ -19,9 +12,9 @@ module Arel
     class ToSql
       def visit_Arel_Nodes_ExtractFrom(o, collector)
         collector << 'extract('
-        visit o.field, collector
+        visit o.right, collector
         collector << ' from '
-        visit o.expr, collector
+        visit o.left, collector
         collector << ')'
       end
     end

data/lib/arel/extensions/factorial.rb

@@ -4,11 +4,12 @@
 module Arel
   module Nodes
     # https://www.postgresql.org/docs/9.4/functions-math.html
-    class Factorial < Arel::Nodes::Unary
+    class Factorial < Arel::Nodes::Node
       attr_accessor :prefix
+      attr_accessor :expr
 
       def initialize(expr, prefix)
-        super(expr)
+        @expr = expr
         @prefix = prefix
       end
     end
@@ -26,6 +27,13 @@ module Arel
         end
       end
     end
+
+    class Dot
+      def visit_Arel_Nodes_Factorial(o)
+        visit_edge o, 'expr'
+        visit_edge o, 'prefix'
+      end
+    end
   end
 end
 

data/lib/arel/extensions/false.rb

@@ -0,0 +1,7 @@
+module Arel
+  module Visitors
+    class Dot
+      alias visit_Arel_Nodes_False terminal
+    end
+  end
+end

data/lib/arel/extensions/function.rb

@@ -3,20 +3,27 @@
 module Arel
   module Nodes
     # Postgres: https://www.postgresql.org/docs/9.1/functions-comparison.html
-    Arel::Nodes::Function.class_eval do
-      # postgres only: https://www.postgresql.org/docs/9.5/functions-aggregate.html
-      attr_accessor :orders
-      attr_accessor :filter
-      attr_accessor :within_group
-      attr_accessor :variardic
-
-      def initialize(expr, aliaz = nil)
-        super()
-        @expressions = expr
-        @alias       = aliaz && SqlLiteral.new(aliaz)
-        @distinct    = false
-        @orders      = []
+    class Function
+      module FunctionExtension
+        # postgres only: https://www.postgresql.org/docs/9.5/functions-aggregate.html
+        attr_accessor :orders
+        attr_accessor :filter
+        attr_accessor :within_group
+        attr_accessor :variardic
+        # postgres only: https://www.postgresql.org/docs/10/ddl-schemas.html
+        attr_accessor :schema_name
+
+        def initialize(expr, aliaz = nil)
+          super
+
+          @expressions = expr
+          @alias       = aliaz && SqlLiteral.new(aliaz)
+          @distinct    = false
+          @orders      = []
+        end
       end
+
+      prepend FunctionExtension
     end
   end
 
@@ -26,6 +33,7 @@ module Arel
       # rubocop:disable Metrics/CyclomaticComplexity
       # rubocop:disable Metrics/AbcSize
       def aggregate(name, o, collector)
+        collector << "#{o.schema_name}." if o.schema_name
         collector << "#{name}("
         collector << 'DISTINCT ' if o.distinct
         collector << 'VARIADIC ' if o.variardic
@@ -62,6 +70,27 @@ module Arel
       # rubocop:enable Metrics/CyclomaticComplexity
       # rubocop:enable Metrics/AbcSize
     end
+
+    class Dot
+      module FunctionExtension
+        def function(o)
+          super
+
+          visit_edge o, 'orders'
+          visit_edge o, 'filter'
+          visit_edge o, 'within_group'
+          visit_edge o, 'variardic'
+        end
+
+        alias visit_Arel_Nodes_Exists function
+        alias visit_Arel_Nodes_Min    function
+        alias visit_Arel_Nodes_Max    function
+        alias visit_Arel_Nodes_Avg    function
+        alias visit_Arel_Nodes_Sum    function
+      end
+
+      prepend FunctionExtension
+    end
   end
 end
 

data/lib/arel/extensions/indirection.rb

@@ -3,25 +3,16 @@
 
 module Arel
   module Nodes
-    class Indirection < Arel::Nodes::Node
-      attr_reader :arg
-      attr_reader :indirection
-
-      def initialize(arg, indirection)
-        super()
-
-        @arg = arg
-        @indirection = indirection
-      end
+    class Indirection < Arel::Nodes::Binary
     end
   end
 
   module Visitors
     class ToSql
       def visit_Arel_Nodes_Indirection(o, collector)
-        visit(o.arg, collector)
+        visit(o.left, collector)
         collector << '['
-        visit(o.indirection, collector)
+        visit(o.right, collector)
         collector << ']'
       end
     end

data/lib/arel/extensions/infer.rb

@@ -4,9 +4,9 @@
 module Arel
   module Nodes
     # https://www.postgresql.org/docs/9.5/sql-insert.html
-    class Infer < Arel::Nodes::Node
-      attr_accessor :name
-      attr_accessor :indexes
+    class Infer < Arel::Nodes::Binary
+      alias name left
+      alias indexes right
     end
   end
 
@@ -15,13 +15,13 @@ module Arel
       def visit_Arel_Nodes_Infer(o, collector)
         if o.name
           collector << 'ON CONSTRAINT '
-          collector << o.name
+          collector << o.left
           collector << SPACE
         end
 
-        if o.indexes
+        if o.right
           collector << '('
-          inject_join o.indexes, collector, ', '
+          inject_join o.right, collector, ', '
           collector << ') '
         end
 

data/lib/arel/extensions/infix_operation.rb

@@ -0,0 +1,17 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class Dot
+      def visit_Arel_Nodes_InfixOperation(o)
+        visit_edge o, 'operator'
+        visit_edge o, 'left'
+        visit_edge o, 'right'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/insert_manager.rb

@@ -1,5 +1,21 @@
-Arel::InsertManager.class_eval do
-  def ==(other)
-    @ast == other.ast
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  class InsertManager < Arel::TreeManager
+    def ==(other)
+      other.is_a?(self.class) && @ast == other.ast
+    end
+  end
+
+  module Visitors
+    class Dot
+      def visit_Arel_InsertManager(o)
+        visit_edge o, 'ast'
+      end
+    end
   end
 end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/insert_statement.rb

@@ -3,19 +3,22 @@
 
 module Arel
   module Nodes
-    # https://www.postgresql.org/docs/9.5/sql-insert.html
-    Arel::Nodes::InsertStatement.class_eval do
-      attr_accessor :with
-      attr_accessor :conflict
-      attr_accessor :override
-      attr_accessor :returning
+    class InsertStatement
+      # https://www.postgresql.org/docs/9.5/sql-insert.html
+      module InsertStatementExtensions
+        attr_accessor :with
+        attr_accessor :conflict
+        attr_accessor :override
+        attr_accessor :returning
 
-      alias_method :old_initialize, :initialize
-      def initialize
-        old_initialize
+        def initialize
+          super
 
-        @returning = []
+          @returning = []
+        end
       end
+
+      prepend(InsertStatementExtensions)
     end
   end
 
@@ -57,18 +60,34 @@ module Arel
                       collector
                     end
 
+        visit(o.conflict, collector) if o.conflict
+
         unless o.returning.empty?
           collector << ' RETURNING '
           collector = inject_join o.returning, collector, ', '
         end
 
-        visit(o.conflict, collector) if o.conflict
         collector
       end
       # rubocop:enable Metrics/CyclomaticComplexity
       # rubocop:enable Metrics/AbcSize
       # rubocop:enable Metrics/PerceivedComplexity
     end
+
+    class Dot
+      module InsertStatementExtension
+        def visit_Arel_Nodes_InsertStatement(o)
+          super
+
+          visit_edge o, 'with'
+          visit_edge o, 'conflict'
+          visit_edge o, 'override'
+          visit_edge o, 'returning'
+        end
+      end
+
+      prepend(InsertStatementExtension)
+    end
   end
 end