arel_toolkit 0.2.0 → 0.4.3

data/lib/arel/extensions/into.rb

@@ -0,0 +1,21 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    class Into < Arel::Nodes::Unary
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Into(o, collector)
+        collector << 'INTO '
+        visit o.expr, collector
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/json_get_field.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.4/functions-json.html#FUNCTIONS-JSON-OP-TABLE
+    class JsonGetField < Arel::Nodes::InfixOperation
+      def initialize(left, right)
+        super(:'->>', left, right)
+      end
+    end
+  end
+end

data/lib/arel/extensions/json_get_object.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.4/functions-json.html#FUNCTIONS-JSON-OP-TABLE
+    class JsonGetObject < Arel::Nodes::InfixOperation
+      def initialize(left, right)
+        super(:'->', left, right)
+      end
+    end
+  end
+end

data/lib/arel/extensions/json_path_get_field.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.4/functions-json.html#FUNCTIONS-JSON-OP-TABLE
+    class JsonPathGetField < Arel::Nodes::InfixOperation
+      def initialize(left, right)
+        super(:'#>>', left, right)
+      end
+    end
+  end
+end

data/lib/arel/extensions/json_path_get_object.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.4/functions-json.html#FUNCTIONS-JSON-OP-TABLE
+    class JsonPathGetObject < Arel::Nodes::InfixOperation
+      def initialize(left, right)
+        super(:'#>', left, right)
+      end
+    end
+  end
+end

data/lib/arel/extensions/jsonb_all_key_exists.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.4/functions-json.html#FUNCTIONS-JSONB-OP-TABLE
+    class JsonbAllKeyExists < Arel::Nodes::InfixOperation
+      def initialize(left, right)
+        super(:'?&', left, right)
+      end
+    end
+  end
+end

data/lib/arel/extensions/jsonb_any_key_exists.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.4/functions-json.html#FUNCTIONS-JSONB-OP-TABLE
+    class JsonbAnyKeyExists < Arel::Nodes::InfixOperation
+      def initialize(left, right)
+        super(:'?|', left, right)
+      end
+    end
+  end
+end

data/lib/arel/extensions/jsonb_key_exists.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.4/functions-json.html#FUNCTIONS-JSONB-OP-TABLE
+    class JsonbKeyExists < Arel::Nodes::InfixOperation
+      def initialize(left, right)
+        super(:'?', left, right)
+      end
+    end
+  end
+end

data/lib/arel/extensions/least.rb

@@ -1,9 +1,23 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
 module Arel
   module Nodes
-    class Least < Arel::Nodes::NamedFunction
-      def initialize(args)
-        super 'LEAST', args
+    # https://www.postgresql.org/docs/10/functions-conditional.html
+    class Least < Arel::Nodes::Unary
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Least(o, collector)
+        collector << 'LEAST('
+        collector = inject_join(o.expr, collector, ', ')
+        collector << ')'
       end
     end
   end
 end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/named_argument.rb

@@ -0,0 +1,24 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    class NamedArgument < Arel::Nodes::Binary
+      alias name left
+      alias value right
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_NamedArgument(o, collector)
+        collector << o.name
+        collector << ' => '
+        visit(o.value, collector)
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/named_function.rb

@@ -8,6 +8,13 @@ module Arel
         aggregate(o.name, o, collector)
       end
     end
+
+    class Dot
+      def visit_Arel_Nodes_NamedFunction(o)
+        visit_edge o, 'name'
+        function(o)
+      end
+    end
   end
 end
 

data/lib/arel/extensions/node.rb

@@ -0,0 +1,10 @@
+module Arel
+  module Nodes
+    class Node
+      def to_sql_and_binds(engine = Arel::Table.engine)
+        collector = engine.connection.send(:collector)
+        engine.connection.visitor.accept(self, collector).value
+      end
+    end
+  end
+end

data/lib/arel/extensions/not_distinct_from.rb

@@ -1,22 +1,9 @@
-# rubocop:disable Naming/MethodName
-# rubocop:disable Naming/UncommunicativeMethodParamName
-
 module Arel
   module Nodes
-    class NotDistinctFrom < Arel::Nodes::Binary
-    end
-  end
-
-  module Visitors
-    class ToSql
-      def visit_Arel_Nodes_NotDistinctFrom(o, collector)
-        visit o.left, collector
-        collector << ' IS NOT DISTINCT FROM '
-        visit o.right, collector
+    class NotDistinctFrom < Arel::Nodes::InfixOperation
+      def initialize(left, right)
+        super(:'IS NOT DISTINCT FROM', left, right)
       end
     end
   end
 end
-
-# rubocop:enable Naming/MethodName
-# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/not_equal.rb

@@ -6,15 +6,13 @@ module Arel
     class ToSql
       def visit_Arel_Nodes_NotEqual(o, collector)
         right = o.right
-
         collector = visit o.left, collector
 
-        case right
-        when Arel::Nodes::Unknown, Arel::Nodes::False, Arel::Nodes::True
+        if [Arel::Nodes::Unknown, Arel::Nodes::False, Arel::Nodes::True].include?(right.class)
           collector << ' IS NOT '
           visit right, collector
 
-        when NilClass
+        elsif right.nil?
           collector << ' IS NOT NULL'
 
         else

data/lib/arel/extensions/ordering.rb

@@ -3,15 +3,19 @@
 
 module Arel
   module Nodes
-    Arel::Nodes::Ordering.class_eval do
-      # Postgres: https://www.postgresql.org/docs/9.4/queries-order.html
-      attr_accessor :nulls
+    class Ordering
+      module OrderingExtension
+        # Postgres: https://www.postgresql.org/docs/9.4/queries-order.html
+        attr_accessor :nulls
 
-      def initialize(expr, nulls = 0)
-        super(expr)
+        def initialize(expr, nulls = 0)
+          super(expr)
 
-        @nulls = nulls
+          @nulls = nulls
+        end
       end
+
+      prepend OrderingExtension
     end
   end
 
@@ -40,6 +44,17 @@ module Arel
         end
       end
     end
+
+    class Dot
+      module OrderingExtension
+        def visit_Arel_Nodes_Ordering(o)
+          super
+          visit_edge o, 'nulls'
+        end
+      end
+
+      prepend OrderingExtension
+    end
   end
 end
 

data/lib/arel/extensions/overlap.rb

@@ -1,7 +1,7 @@
 module Arel
   module Nodes
     # https://www.postgresql.org/docs/9.1/functions-array.html
-    class Overlap < InfixOperation
+    class Overlap < Arel::Nodes::InfixOperation
       def initialize(left, right)
         super(:'&&', left, right)
       end

data/lib/arel/extensions/overlaps.rb

@@ -0,0 +1,49 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/10/functions-string.html
+    class Overlaps < Arel::Nodes::Node
+      attr_reader :start1
+      attr_reader :end1
+      attr_reader :start2
+      attr_reader :end2
+
+      def initialize(start1, end1, start2, end2)
+        @start1 = start1
+        @end1 = end1
+        @start2 = start2
+        @end2 = end2
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Overlaps(o, collector)
+        collector << '('
+        visit o.start1, collector
+        collector << ', '
+        visit o.end1, collector
+        collector << ') OVERLAPS ('
+        visit o.start2, collector
+        collector << ', '
+        visit o.end2, collector
+        collector << ')'
+      end
+    end
+
+    class Dot
+      def visit_Arel_Nodes_Overlaps(o)
+        visit_edge o, 'start1'
+        visit_edge o, 'end1'
+        visit_edge o, 'start2'
+        visit_edge o, 'end2'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/overlay.rb

@@ -0,0 +1,53 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+# rubocop:disable Metrics/AbcSize
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/10/functions-string.html
+    class Overlay < Arel::Nodes::Node
+      attr_reader :string
+      attr_reader :substring
+      attr_reader :start
+      attr_reader :length
+
+      def initialize(string, substring, start, length = nil)
+        @string = string
+        @substring = substring
+        @start = start
+        @length = length
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Overlay(o, collector)
+        collector << 'overlay('
+        visit o.string, collector
+        collector << ' placing '
+        visit o.substring, collector
+        collector << ' from '
+        visit o.start, collector
+        unless o.length.nil?
+          collector << ' for '
+          visit o.length, collector
+        end
+        collector << ')'
+      end
+    end
+
+    class Dot
+      def visit_Arel_Nodes_Overlay(o)
+        visit_edge o, 'string'
+        visit_edge o, 'substring'
+        visit_edge o, 'start'
+        visit_edge o, 'length'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName
+# rubocop:enable Metrics/AbcSize

data/lib/arel/extensions/position.rb

@@ -0,0 +1,27 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.1/functions-string.html#FUNCTIONS-STRING-SQL
+    class Position < Arel::Nodes::Binary
+      alias substring left
+      alias string right
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Position(o, collector)
+        collector << 'position('
+        visit o.substring, collector
+        collector << ' in '
+        visit o.string, collector
+        collector << ')'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/prepare.rb

@@ -0,0 +1,39 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class Dot
+      def visit_Arel_Nodes_Prepare(o)
+        visit_edge o, 'name'
+        visit_edge o, 'argtypes'
+        visit_edge o, 'query'
+      end
+    end
+
+    class ToSql
+      def visit_Arel_Nodes_Prepare(o, collector)
+        collector << "PREPARE #{o.name}"
+        collector << " (#{o.argtypes.join(', ')})" if o.argtypes
+        collector << ' AS ('
+        visit(o.query, collector)
+        collector << ')'
+      end
+    end
+  end
+
+  module Nodes
+    class Prepare < Node
+      attr_reader :name, :query, :argtypes
+
+      def initialize(name, argtypes, query)
+        @name = name
+        @query = query
+        @argtypes = argtypes
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName