arel_toolkit 0.3.0 → 0.4.4

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,38 @@
+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
+        when Arel::Enhance::QueryMethods::QueryMethod
+          test.matches?(object)
+        else
+          object == test
+        end
+      end
+    end
+  end
+end

data/lib/arel/enhance/query_methods.rb

@@ -0,0 +1,23 @@
+module Arel
+  module Enhance
+    module QueryMethods
+      class QueryMethod
+        attr_reader :subject
+
+        def initialize(subject)
+          @subject = subject
+        end
+      end
+
+      class Ancestors < QueryMethod
+        def matches?(other)
+          other <= subject
+        end
+      end
+
+      def self.in_ancestors?(object)
+        Ancestors.new(object)
+      end
+    end
+  end
+end

data/lib/arel/enhance/visitor.rb

@@ -0,0 +1,97 @@
+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)
+        object.each do |key, child|
+          process_node(child, Arel::Enhance::PathNode.new([:[], key], key))
+        end
+      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
+
+      # rubocop:disable Metrics/AbcSize
+      # arel/lib/arel/visitors/visitor.rb:29
+      def visit(object)
+        dispatch_method = dispatch[object.class]
+        send dispatch_method, object
+      rescue NoMethodError => e
+        raise e if respond_to?(dispatch_method, true)
+
+        superklass = object.class.ancestors.find do |klass|
+          respond_to?(dispatch[klass], true)
+        end
+        raise(TypeError, "Cannot visit #{object.class}") unless superklass
+
+        dispatch[object.class] = dispatch[superklass]
+        retry
+      end
+      # rubocop:enable Metrics/AbcSize
+
+      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'
@@ -20,6 +21,7 @@ require 'arel/extensions/lateral'
 require 'arel/extensions/range_function'
 require 'arel/extensions/with_ordinality'
 require 'arel/extensions/table'
+require 'arel/extensions/attributes_attribute'
 require 'arel/extensions/row'
 require 'arel/extensions/ordering'
 require 'arel/extensions/all'
@@ -42,11 +44,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'
@@ -58,8 +64,6 @@ require 'arel/extensions/current_of_expression'
 require 'arel/extensions/delete_statement'
 require 'arel/extensions/least'
 require 'arel/extensions/greatest'
-require 'arel/extensions/generate_series'
-require 'arel/extensions/rank'
 require 'arel/extensions/coalesce'
 require 'arel/extensions/not_equal'
 require 'arel/extensions/equality'
@@ -80,7 +84,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 +93,29 @@ 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'
+require 'arel/extensions/active_record_type_caster_connection'
+require 'arel/extensions/active_record_relation_query_attribute'
+require 'arel/extensions/active_model_attribute_with_cast_value'
+require 'arel/extensions/exists'
+require 'arel/extensions/bind_param'
+require 'arel/extensions/node'
+require 'arel/extensions/top'
 
 module Arel
   module Extensions

data/lib/arel/extensions/active_model_attribute_with_cast_value.rb

@@ -0,0 +1,22 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class Dot
+      def visit_ActiveModel_Attribute_WithCastValue(o)
+        visit_edge o, 'name'
+        visit_edge o, 'value_before_type_cast'
+      end
+    end
+
+    class ToSql
+      def visit_ActiveModel_Attribute_WithCastValue(_o, collector)
+        collector
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/active_record_relation_query_attribute.rb

@@ -0,0 +1,22 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Visitors
+    class Dot
+      def visit_ActiveRecord_Relation_QueryAttribute(o)
+        visit_edge o, 'name'
+        visit_edge o, 'value_before_type_cast'
+      end
+    end
+
+    class ToSql
+      def visit_ActiveRecord_Relation_QueryAttribute(_o, collector)
+        collector
+      end
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName

data/lib/arel/extensions/active_record_type_caster_connection.rb

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

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

@@ -0,0 +1,47 @@
+# rubocop:disable Naming/MethodName
+# rubocop:disable Naming/UncommunicativeMethodParamName
+
+module Arel
+  module Attributes
+    class Attribute
+      module AttributeExtension
+        # postgres only: https://www.postgresql.org/docs/10/ddl-schemas.html
+        attr_accessor :schema_name
+        attr_accessor :database
+      end
+
+      prepend AttributeExtension
+    end
+  end
+
+  module Visitors
+    class ToSql
+      module AttributesAttributeExtension
+        def visit_Arel_Attributes_Attribute(o, collector)
+          collector << "#{quote_table_name(o.database)}." if o.database
+          collector << "#{quote_table_name(o.schema_name)}." if o.schema_name
+
+          super
+        end
+      end
+
+      prepend AttributesAttributeExtension
+    end
+
+    class Dot
+      module AttributesAttributeExtension
+        def visit_Arel_Attributes_Attribute(o)
+          super
+
+          visit_edge o, 'schema_name'
+          visit_edge o, 'database'
+        end
+      end
+
+      prepend AttributesAttributeExtension
+    end
+  end
+end
+
+# rubocop:enable Naming/MethodName
+# rubocop:enable Naming/UncommunicativeMethodParamName