Contents

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

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
arel_toolkit-0.4.0	lib/arel/enhance/visitor.rb