Module: RuboCop::AST::NodePattern::Macros
Overview
Helpers for defining methods based on a pattern string
Instance Method Summary
-
#def_node_matcher(method_name, pattern_str, **keyword_defaults)
Define a method which applies a pattern to an AST node.
-
#def_node_search(method_name, pattern_str, **keyword_defaults)
Define a method which recurses over the descendants of an AST node, checking whether any of them match the provided pattern.
-
#def_node_pattern_method_lazily(definer, method_name, pattern_str, keyword_defaults, location)
private
Defines a stub that compiles the pattern and replaces itself on first invocation.
Instance Method Details
#def_node_matcher(method_name, pattern_str, **keyword_defaults)
Define a method which applies a pattern to an AST node
The new method will return nil if the node does not match.
If the node matches, and a block is provided, the new method will
yield to the block (passing any captures as block arguments).
If the node matches, and no block is provided, the new method will
return the captures, or true if there were none.
# File 'lib/rubocop/ast/node_pattern.rb', line 50
def def_node_matcher(method_name, pattern_str, **keyword_defaults) if NodePattern.lazy_compilation def_node_pattern_method_lazily(:def_node_matcher, method_name, pattern_str, keyword_defaults, caller_locations(1, 1).first) else NodePattern.new(pattern_str).def_node_matcher(self, method_name, **keyword_defaults) end end
#def_node_pattern_method_lazily(definer, method_name, pattern_str, keyword_defaults, location) (private)
Defines a stub that compiles the pattern and replaces itself on
first invocation. The compiled method is invoked through the owner’s
instance method rather than a regular dispatch, so that overrides
calling super keep working.
# File 'lib/rubocop/ast/node_pattern.rb', line 80
def def_node_pattern_method_lazily(definer, method_name, pattern_str, keyword_defaults, location) base = self define_method(method_name) do |*args, **kwargs, &block| # Remove the stub before defining the compiled method in its # place, so that no method redefinition warning is emitted. begin base.send(:remove_method, method_name) rescue NameError nil # the method was already compiled concurrently end pattern = NodePattern.new(pattern_str) pattern.definition_location = location pattern.public_send(definer, base, method_name, **keyword_defaults) base.instance_method(method_name).bind_call(self, *args, **kwargs, &block) end method_name end
#def_node_search(method_name, pattern_str, **keyword_defaults)
Define a method which recurses over the descendants of an AST node, checking whether any of them match the provided pattern
If the method name ends with '?', the new method will return true
as soon as it finds a descendant which matches. Otherwise, it will
yield all descendants which match.
# File 'lib/rubocop/ast/node_pattern.rb', line 65
def def_node_search(method_name, pattern_str, **keyword_defaults) if NodePattern.lazy_compilation def_node_pattern_method_lazily(:def_node_search, method_name, pattern_str, keyword_defaults, caller_locations(1, 1).first) else NodePattern.new(pattern_str).def_node_search(self, method_name, **keyword_defaults) end end