arel_toolkit 0.3.0 → 0.4.0

data/lib/arel/enhance/path.rb

@@ -0,0 +1,38 @@
+module Arel
+  module Enhance
+    class Path
+      attr_reader :nodes
+
+      def initialize(nodes = [])
+        @nodes = nodes
+      end
+
+      def append(path_node)
+        Path.new(nodes + [path_node])
+      end
+
+      def dig_send(object)
+        selected_object = object
+        nodes.each do |path_node|
+          selected_object = selected_object.send(*path_node.method)
+        end
+        selected_object
+      end
+
+      def to_a
+        nodes.map(&:value)
+      end
+
+      def current
+        nodes.last
+      end
+
+      def inspect
+        nodes.inspect
+        string = '['
+        string << nodes.map(&:inspect).join(', ')
+        string << ']'
+      end
+    end
+  end
+end

data/lib/arel/enhance/path_node.rb

@@ -0,0 +1,26 @@
+module Arel
+  module Enhance
+    class PathNode
+      attr_reader :method
+      attr_reader :value
+
+      def initialize(method, value)
+        @method = method
+        @value = value
+      end
+
+      def arguments?
+        method.is_a?(Array)
+      end
+
+      def inspect
+        case value
+        when String
+          "'#{value}'"
+        else
+          value.inspect
+        end
+      end
+    end
+  end
+end

data/lib/arel/enhance/query.rb

@@ -0,0 +1,36 @@
+module Arel
+  module Enhance
+    class Query
+      def self.call(node, kwargs)
+        node_attributes = %i[context parent]
+        node_args = kwargs.slice(*node_attributes)
+        object_args = kwargs.except(*node_attributes)
+
+        node.each.select do |child_node|
+          next unless matches?(child_node, node_args)
+
+          matches?(child_node.object, object_args)
+        end
+      end
+
+      def self.matches?(object, test)
+        case test
+        when Hash
+          case object
+          when Hash
+            test <= object
+          else
+            test.all? do |test_key, test_value|
+              next false unless object.respond_to?(test_key)
+
+              object_attribute_value = object.public_send(test_key)
+              matches? object_attribute_value, test_value
+            end
+          end
+        else
+          object == test
+        end
+      end
+    end
+  end
+end

data/lib/arel/enhance/visitor.rb

@@ -0,0 +1,81 @@
+require_relative './context_enhancer/arel_table'
+
+module Arel
+  module Enhance
+    # rubocop:disable Naming/MethodName
+    class Visitor < Arel::Visitors::Dot
+      DEFAULT_CONTEXT_ENHANCERS = {
+        Arel::Table => Arel::Enhance::ContextEnhancer::ArelTable,
+      }.freeze
+
+      attr_reader :context_enhancers
+
+      def accept(object, context_enhancers = DEFAULT_CONTEXT_ENHANCERS)
+        @context_enhancers = context_enhancers
+
+        root_node = Arel::Enhance::Node.new(object)
+        accept_with_root(object, root_node)
+      end
+
+      def accept_with_root(object, root_node, context_enhancers = DEFAULT_CONTEXT_ENHANCERS)
+        @context_enhancers = context_enhancers
+
+        with_node(root_node) do
+          visit object
+        end
+
+        root_node
+      end
+
+      private
+
+      def visit_edge(object, method)
+        arel_node = object.send(method)
+
+        process_node(arel_node, Arel::Enhance::PathNode.new(method, method))
+      end
+
+      def nary(object)
+        visit_edge(object, 'children')
+      end
+      alias visit_Arel_Nodes_And nary
+
+      def visit_Hash(_object)
+        raise 'Hash is not supported'
+      end
+
+      def visit_Array(object)
+        object.each_with_index do |child, index|
+          process_node(child, Arel::Enhance::PathNode.new([:[], index], index))
+        end
+      end
+
+      def process_node(arel_node, path_node)
+        node = Arel::Enhance::Node.new(arel_node)
+        current_node.add(path_node, node)
+
+        update_context(node)
+
+        with_node node do
+          visit arel_node
+        end
+      end
+
+      def visit(object)
+        Arel::Visitors::Visitor.instance_method(:visit).bind(self).call(object)
+      end
+
+      def current_node
+        @node_stack.last
+      end
+
+      def update_context(node)
+        enhancer = context_enhancers[node.object.class]
+        return if enhancer.nil?
+
+        enhancer.call(node)
+      end
+    end
+    # rubocop:enable Naming/MethodName
+  end
+end

data/lib/arel/extensions.rb

@@ -1,3 +1,4 @@
+require 'arel/extensions/dot'
 require 'arel/extensions/unknown'
 require 'arel/extensions/time_with_precision'
 require 'arel/extensions/current_time'
@@ -42,11 +43,15 @@ require 'arel/extensions/modulo'
 require 'arel/extensions/absolute'
 require 'arel/extensions/bitwise_xor'
 require 'arel/extensions/exponentiation'
+
 require 'arel/extensions/contains'
+unless Gem.loaded_specs.key?('postgres_ext')
+  require 'arel/extensions/contained_within_equals'
+  require 'arel/extensions/contains_equals'
+  require 'arel/extensions/overlap'
+end
+
 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'
 require 'arel/extensions/default_values'
@@ -80,7 +85,6 @@ 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'
@@ -90,6 +94,22 @@ require 'arel/extensions/substring'
 require 'arel/extensions/overlaps'
 require 'arel/extensions/trim'
 require 'arel/extensions/named_argument'
+require 'arel/extensions/tree_manager'
+require 'arel/extensions/into'
+require 'arel/extensions/select_core'
+require 'arel/extensions/unary'
+require 'arel/extensions/binary'
+require 'arel/extensions/unary_operation'
+require 'arel/extensions/infix_operation'
+require 'arel/extensions/values_list'
+require 'arel/extensions/case'
+require 'arel/extensions/current_row'
+require 'arel/extensions/false'
+require 'arel/extensions/true'
+require 'arel/extensions/to_sql'
+require 'arel/extensions/prepare'
+require 'arel/extensions/dealocate'
+require 'arel/extensions/active_record_type_caster_map'
 
 module Arel
   module Extensions

data/lib/arel/extensions/active_record_type_caster_map.rb

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

data/lib/arel/extensions/array.rb

@@ -3,14 +3,7 @@
 
 module Arel
   module Nodes
-    class Array < Arel::Nodes::Node
-      attr_reader :items
-
-      def initialize(items)
-        super()
-
-        @items = items
-      end
+    class Array < Arel::Nodes::Unary
     end
   end
 
@@ -18,7 +11,7 @@ module Arel
     class ToSql
       def visit_Arel_Nodes_Array(o, collector)
         collector << 'ARRAY['
-        inject_join(o.items, collector, ', ')
+        inject_join(o.expr, collector, ', ')
         collector << ']'
       end
     end

data/lib/arel/extensions/at_time_zone.rb

@@ -4,12 +4,12 @@
 module Arel
   module Nodes
     # https://www.postgresql.org/docs/9.2/functions-datetime.html#FUNCTIONS-DATETIME-ZONECONVERT
-    class AtTimeZone < Arel::Nodes::Unary
+    class AtTimeZone < Arel::Nodes::Node
       attr_reader :timezone
+      attr_reader :expr
 
       def initialize(expr, timezone)
-        super(expr)
-
+        @expr = expr
         @timezone = timezone
       end
     end
@@ -23,6 +23,13 @@ module Arel
         visit o.timezone, collector
       end
     end
+
+    class Dot
+      def visit_Arel_Nodes_AtTimeZone(o)
+        visit_edge o, 'expr'
+        visit_edge o, 'timezone'
+      end
+    end
   end
 end
 

data/lib/arel/extensions/binary.rb

@@ -0,0 +1,7 @@
+module Arel
+  module Visitors
+    class Dot
+      alias visit_Arel_Nodes_Binary binary
+    end
+  end
+end

data/lib/arel/extensions/bit_string.rb

@@ -3,21 +3,14 @@
 
 module Arel
   module Nodes
-    class BitString < Arel::Nodes::Node
-      attr_reader :str
-
-      def initialize(str)
-        super()
-
-        @str = str
-      end
+    class BitString < Arel::Nodes::Unary
     end
   end
 
   module Visitors
     class ToSql
       def visit_Arel_Nodes_BitString(o, collector)
-        collector << "B'#{o.str[1..-1]}'"
+        collector << "B'#{o.expr[1..-1]}'"
       end
     end
   end

data/lib/arel/extensions/case.rb

@@ -0,0 +1,17 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class Dot
+      def visit_Arel_Nodes_Case(o)
+        visit_edge o, 'case'
+        visit_edge o, 'conditions'
+        visit_edge o, 'default'
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/conflict.rb

@@ -40,6 +40,15 @@ module Arel
       end
       # rubocop:enable Metrics/AbcSize
     end
+
+    class Dot
+      def visit_Arel_Nodes_Conflict(o)
+        visit_edge o, 'action'
+        visit_edge o, 'infer'
+        visit_edge o, 'values'
+        visit_edge o, 'wheres'
+      end
+    end
   end
 end
 

data/lib/arel/extensions/contains.rb

@@ -1,10 +1,32 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
 module Arel
-  module Nodes
-    # https://www.postgresql.org/docs/9.1/functions-array.html
-    class Contains < Arel::Nodes::InfixOperation
-      def initialize(left, right)
-        super(:'@>', left, right)
+  if Gem.loaded_specs.key?('postgres_ext')
+    module Visitors
+      module ContainsPatch
+        def visit_Arel_Nodes_Contains(o, collector)
+          if o.left.is_a?(Arel::Attribute)
+            super
+          else
+            infix_value o, collector, ' @> '
+          end
+        end
+      end
+
+      PostgreSQL.prepend(ContainsPatch)
+    end
+  else
+    module Nodes
+      # https://www.postgresql.org/docs/9.1/functions-array.html
+      class Contains < Arel::Nodes::InfixOperation
+        def initialize(left, right)
+          super(:'@>', left, right)
+        end
       end
     end
   end
 end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/current_catalog.rb

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

data/lib/arel/extensions/current_date.rb

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

data/lib/arel/extensions/current_of_expression.rb

@@ -4,14 +4,7 @@
 module Arel
   module Nodes
     # https://www.postgresql.org/docs/10/sql-update.html
-    class CurrentOfExpression < Arel::Nodes::Node
-      attr_accessor :cursor_name
-
-      def initialize(cursor_name)
-        super()
-
-        @cursor_name = cursor_name
-      end
+    class CurrentOfExpression < Arel::Nodes::Unary
     end
   end
 
@@ -19,7 +12,7 @@ module Arel
     class ToSql
       def visit_Arel_Nodes_CurrentOfExpression(o, collector)
         collector << 'CURRENT OF '
-        collector << o.cursor_name
+        collector << o.expr
       end
     end
   end