arel_toolkit 0.2.0 → 0.3.0

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/named_argument.rb

@@ -0,0 +1,29 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    class NamedArgument < Arel::Nodes::Node
+      attr_reader :name
+      attr_reader :value
+
+      def initialize(name, value)
+        @name = name
+        @value = value
+      end
+    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/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/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,40 @@
+# 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
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/overlay.rb

@@ -0,0 +1,44 @@
+# 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
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName
+# rubocop:enable Metrics/AbcSize

data/lib/arel/extensions/position.rb

@@ -0,0 +1,32 @@
+# 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::Node
+      attr_reader :substring
+      attr_reader :string
+
+      def initialize(substring, string)
+        @substring = substring
+        @string = string
+      end
+    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/select_manager.rb

@@ -0,0 +1,9 @@
+Arel::SelectManager.class_eval do
+  def ==(other)
+    @ast == other.ast && @ctx == other.ctx
+  end
+
+  protected
+
+  attr_reader :ctx
+end

data/lib/arel/extensions/substring.rb

@@ -0,0 +1,38 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/10/functions-string.html
+    class Substring < Arel::Nodes::Node
+      attr_reader :string
+      attr_reader :pattern
+      attr_reader :escape
+
+      def initialize(string, pattern, escape)
+        @string = string
+        @pattern = pattern
+        @escape = escape
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Substring(o, collector)
+        collector << 'substring('
+        visit o.string, collector
+        collector << ' from '
+        visit o.pattern, collector
+        unless o.escape.nil?
+          collector << ' for '
+          visit o.escape, collector
+        end
+        collector << ')'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/transaction.rb

@@ -0,0 +1,50 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+# rubocop:disable Metrics/CyclomaticComplexity
+# rubocop:disable Metrics/AbcSize
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/8.3/tutorial-transactions.html
+    class Transaction < Arel::Nodes::Node
+      attr_reader :type
+      attr_reader :options
+
+      def initialize(type, options)
+        @type = type
+        @options = options
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Transaction(o, collector)
+        case o.type
+        when 0
+          collector << 'BEGIN'
+        when 2
+          collector << 'COMMIT'
+        when 3
+          collector << 'ROLLBACK'
+        when 4
+          collector << 'SAVEPOINT '
+          collector << o.options.join(' ')
+        when 5
+          collector << 'RELEASE SAVEPOINT '
+          collector << o.options.join(' ')
+        when 6
+          collector << 'ROLLBACK TO '
+          collector << o.options.join(' ')
+        else
+          raise "Unknown transaction type `#{o.type}`"
+        end
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName
+# rubocop:enable Metrics/CyclomaticComplexity
+# rubocop:enable Metrics/AbcSize

data/lib/arel/extensions/trim.rb

@@ -0,0 +1,36 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/10/functions-string.html
+    class Trim < Arel::Nodes::Node
+      attr_reader :type
+      attr_reader :substring
+      attr_reader :string
+
+      def initialize(type, substring, string)
+        @type = type
+        @substring = substring
+        @string = string
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_Trim(o, collector)
+        collector << "trim(#{o.type} "
+        if o.substring
+          visit o.substring, collector
+          collector << ' from '
+        end
+        visit o.string, collector
+        collector << ')'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/type_cast.rb

@@ -12,6 +12,10 @@ module Arel
         @arg = arg
         @type_name = type_name
       end
+
+      def ==(other)
+        arg == other.arg && type_name == other.type_name
+      end
     end
   end
 

data/lib/arel/{sql_to_arel → extensions}/unbound_column_reference.rb

@@ -1,5 +1,5 @@
 module Arel
-  module SqlToArel
+  module Nodes
     class UnboundColumnReference < ::Arel::Nodes::SqlLiteral; end
   end
 end

data/lib/arel/extensions/update_manager.rb

@@ -0,0 +1,9 @@
+Arel::UpdateManager.class_eval do
+  def ==(other)
+    @ast == other.ast && @ctx == other.ctx
+  end
+
+  protected
+
+  attr_reader :ctx
+end

data/lib/arel/extensions/update_statement.rb

@@ -8,6 +8,14 @@ module Arel
       attr_accessor :with
       attr_accessor :froms
       attr_accessor :returning
+
+      alias_method :old_initialize, :initialize
+      def initialize
+        old_initialize
+
+        @froms = []
+        @returning = []
+      end
     end
   end
 

data/lib/arel/extensions/variable_set.rb

@@ -0,0 +1,46 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.3/sql-set.html
+    class VariableSet < Arel::Nodes::Node
+      attr_reader :type
+      attr_reader :args
+      attr_reader :name
+      attr_reader :local
+
+      def initialize(type, args, name, local)
+        @type = type
+        @args = args
+        @name = name
+        @local = local
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_VariableSet(o, collector)
+        collector << 'SET '
+        collector << 'LOCAL ' if o.local
+
+        if o.name == 'timezone'
+          collector << 'TIME ZONE '
+        else
+          collector << o.name
+          collector << ' TO '
+        end
+
+        if o.args.empty?
+          collector << 'DEFAULT'
+        else
+          inject_join(o.args, collector, ', ')
+        end
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/variable_show.rb

@@ -0,0 +1,31 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Nodes
+    # https://www.postgresql.org/docs/9.1/sql-show.html
+    class VariableShow < Arel::Nodes::Node
+      attr_reader :name
+
+      def initialize(name)
+        @name = name
+      end
+    end
+  end
+
+  module Visitors
+    class ToSql
+      def visit_Arel_Nodes_VariableShow(o, collector)
+        collector << 'SHOW '
+        collector << if o.name == 'timezone'
+                       'TIME ZONE'
+                     else
+                       o.name
+                     end
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions.rb

@@ -44,6 +44,8 @@ require 'arel/extensions/bitwise_xor'
 require 'arel/extensions/exponentiation'
 require 'arel/extensions/contains'
 require 'arel/extensions/contained_by'
+require 'arel/extensions/contained_within_equals'
+require 'arel/extensions/contains_equals'
 require 'arel/extensions/overlap'
 require 'arel/extensions/select_statement'
 require 'arel/extensions/insert_statement'
@@ -64,6 +66,30 @@ require 'arel/extensions/equality'
 require 'arel/extensions/named_function'
 require 'arel/extensions/intersect_all'
 require 'arel/extensions/except_all'
+require 'arel/extensions/select_manager'
+require 'arel/extensions/insert_manager'
+require 'arel/extensions/update_manager'
+require 'arel/extensions/delete_manager'
+require 'arel/extensions/at_time_zone'
+require 'arel/extensions/extract_from'
+require 'arel/extensions/json_get_object'
+require 'arel/extensions/json_get_field'
+require 'arel/extensions/json_path_get_object'
+require 'arel/extensions/json_path_get_field'
+require 'arel/extensions/jsonb_key_exists'
+require 'arel/extensions/jsonb_any_key_exists'
+require 'arel/extensions/jsonb_all_key_exists'
+require 'arel/extensions/transaction'
+require 'arel/extensions/unbound_column_reference'
+require 'arel/extensions/assignment'
+require 'arel/extensions/variable_set'
+require 'arel/extensions/variable_show'
+require 'arel/extensions/position'
+require 'arel/extensions/overlay'
+require 'arel/extensions/substring'
+require 'arel/extensions/overlaps'
+require 'arel/extensions/trim'
+require 'arel/extensions/named_argument'
 
 module Arel
   module Extensions

data/lib/arel/middleware/chain.rb

@@ -0,0 +1,97 @@
+module Arel
+  module Middleware
+    class Chain
+      def initialize(internal_middleware = [], internal_context = {})
+        @internal_middleware = internal_middleware
+        @internal_context = internal_context
+      end
+
+      def execute(sql, binds = [])
+        return sql if internal_middleware.length.zero?
+
+        result = Arel.sql_to_arel(sql, binds: binds)
+        updated_context = context.merge(original_sql: sql)
+
+        internal_middleware.each do |middleware_item|
+          result = result.map do |arel|
+            middleware_item.call(arel, updated_context)
+          end
+        end
+
+        result.to_sql
+      end
+
+      def current
+        internal_middleware.dup
+      end
+
+      def apply(middleware, &block)
+        continue_chain(middleware, internal_context, &block)
+      end
+
+      def only(middleware, &block)
+        continue_chain(middleware, internal_context, &block)
+      end
+
+      def none(&block)
+        continue_chain([], internal_context, &block)
+      end
+
+      def except(without_middleware, &block)
+        new_middleware = internal_middleware.reject do |middleware|
+          middleware == without_middleware
+        end
+
+        continue_chain(new_middleware, internal_context, &block)
+      end
+
+      def insert_before(new_middleware, existing_middleware, &block)
+        index = internal_middleware.index(existing_middleware)
+        updated_middleware = internal_middleware.insert(index, new_middleware)
+        continue_chain(updated_middleware, internal_context, &block)
+      end
+
+      def insert_after(new_middleware, existing_middleware, &block)
+        index = internal_middleware.index(existing_middleware)
+        updated_middleware = internal_middleware.insert(index + 1, new_middleware)
+        continue_chain(updated_middleware, internal_context, &block)
+      end
+
+      def context(new_context = nil, &block)
+        if new_context.nil? && !block.nil?
+          raise 'You cannot do a block statement while calling context without arguments'
+        end
+
+        return internal_context if new_context.nil?
+
+        continue_chain(internal_middleware, new_context, &block)
+      end
+
+      protected
+
+      attr_reader :internal_middleware
+      attr_reader :internal_context
+
+      private
+
+      def continue_chain(middleware, context, &block)
+        new_chain = Arel::Middleware::Chain.new(middleware, context)
+        maybe_execute_block(new_chain, &block)
+      end
+
+      def maybe_execute_block(new_chain, &block)
+        return new_chain if block.nil?
+
+        previous_chain = Middleware.current_chain
+        Arel::Middleware.current_chain = new_chain
+        yield block
+      ensure
+        Arel::Middleware.current_chain = previous_chain
+      end
+
+      def current_chain
+        Arel::Middleware.current_chain
+      end
+    end
+  end
+end