Contents

# typed: strict
# frozen_string_literal: true

module Packwerk
  # Extracts a constant name from an AST node of type :const
  class ConstNodeInspector
    extend T::Sig
    include ConstantNameInspector

    sig do
      override
        .params(node: AST::Node, ancestors: T::Array[AST::Node])
        .returns(T.nilable(String))
    end
    def constant_name_from_node(node, ancestors:)
      return nil unless NodeHelpers.constant?(node)

      parent = ancestors.first

      # Only process the root `const` node for namespaced constant references. For example, in the
      # reference `Spam::Eggs::Thing`, we only process the const node associated with `Spam`.
      return nil unless root_constant?(parent)

      if parent && constant_in_module_or_class_definition?(node, parent: parent)
        fully_qualify_constant(ancestors)
      else
        begin
          NodeHelpers.constant_name(node)
        rescue NodeHelpers::TypeError
          nil
        end
      end
    end

    private

    sig { params(parent: T.nilable(AST::Node)).returns(T::Boolean) }
    def root_constant?(parent)
      !(parent && NodeHelpers.constant?(parent))
    end

    sig { params(node: AST::Node, parent: AST::Node).returns(T.nilable(T::Boolean)) }
    def constant_in_module_or_class_definition?(node, parent:)
      parent_name = NodeHelpers.module_name_from_definition(parent)
      parent_name && parent_name == NodeHelpers.constant_name(node)
    end

    sig { params(ancestors: T::Array[AST::Node]).returns(String) }
    def fully_qualify_constant(ancestors)
      # We're defining a class with this name, in which case the constant is implicitly fully qualified by its
      # enclosing namespace
      "::" + NodeHelpers.parent_module_name(ancestors: ancestors)
    end
  end

  private_constant :ConstNodeInspector
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
packwerk-3.2.1	lib/packwerk/const_node_inspector.rb
packwerk-3.2.0	lib/packwerk/const_node_inspector.rb
packwerk-3.1.0	lib/packwerk/const_node_inspector.rb
packwerk-3.0.1	lib/packwerk/const_node_inspector.rb
packwerk-3.0.0	lib/packwerk/const_node_inspector.rb