Contents

# frozen_string_literal: true

module Synvert::Core::NodeQuery::Compiler
  # Expression represents a node query expression.
  class Expression
    # Initialize a Expression.
    # @param selector [Synvert::Core::NodeQuery::Compiler::Selector] the selector
    # @param rest [Synvert::Core::NodeQuery::Compiler::Expression] the rest expression
    def initialize(selector: nil, rest: nil)
      @selector = selector
      @rest = rest
    end

    # Check if the node matches the expression.
    # @param node [Parser::AST::Node] the node
    # @return [Boolean]
    def match?(node)
      !query_nodes(node).empty?
    end

    # Query nodes by the selector and the rest expression.
    #
    # @param node [Parser::AST::Node] node to match
    # @param descendant_match [Boolean] whether to match in descendant node
    # @return [Array<Parser::AST::Node>] matching nodes.
    def query_nodes(node, descendant_match = true)
      matching_nodes = find_nodes_by_selector(node, descendant_match)
      return matching_nodes if @rest.nil?

      matching_nodes.flat_map { |matching_node| find_nodes_by_rest(matching_node, descendant_match) }
    end

    def to_s
      result = []
      result << @selector.to_s if @selector
      result << @rest.to_s if @rest
      result.join(' ')
    end

    private

    # Find nodes by @rest
    # @param node [Parser::AST::Node] node to match
    # @param descendant_match [Boolean] whether to match in descendant node
    def find_nodes_by_rest(node, descendant_match = false)
      @rest.query_nodes(node, descendant_match)
    end

    # Find nodes with nil relationship.
    # @param node [Parser::AST::Node] node to match
    # @param descendant_match [Boolean] whether to match in descendant node
    def find_nodes_by_selector(node, descendant_match = true)
      return Array(node) if !@selector

      @selector.query_nodes(node, descendant_match)
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
synvert-core-1.3.1	lib/synvert/core/node_query/compiler/expression.rb
synvert-core-1.3.0	lib/synvert/core/node_query/compiler/expression.rb