Sha256: 9231e49704ae435e836f6c17661c832181d1ff9ffa7dcfd84b3d3d8e9376c924
Contents?: true
Size: 1.77 KB
Versions: 6
Compression:
Stored size: 1.77 KB
Contents
# frozen_string_literal: true
require 'ast'
require 'set'
module I18n::Tasks::Scanners
class RubyAstCallFinder
include AST::Processor::Mixin
# @param messages [Array<Symbol>] method names to intercept.
# @param receivers [Array<nil, AST::Node>] receivers of the `messages` to intercept.
def initialize(messages:, receivers:)
super()
@messages = Set.new(messages).freeze
@receivers = Set.new(receivers).freeze
end
# @param root_node [Parser::AST:Node]
# @yieldparam send_node [Parser::AST:Node]
# @yieldparam method_name [nil, String] the surrounding method's name.
def find_calls(root_node, &block)
@callback = block
process root_node
ensure
@callback = nil
end
# @param root_node (see #find_calls)
# @yieldparam (see #find_calls)
# @return [Array<block return values excluding nils>]
def collect_calls(root_node)
results = []
find_calls root_node do |send_node, method_name|
result = yield send_node, method_name
results << result if result
end
results
end
def on_def(node)
@method_name = node.children[0]
handler_missing node
ensure
@method_name = nil
end
def on_send(send_node)
receiver = send_node.children[0]
message = send_node.children[1]
if @messages.include?(message) &&
# use `any?` because `include?` checks type equality, but the receiver is a Parser::AST::Node != AST::Node.
@receivers.any? { |r| r == receiver }
@callback.call(send_node, @method_name)
else
handler_missing send_node
end
nil
end
def handler_missing(node)
node.children.each { |child| process(child) if child.is_a?(::Parser::AST::Node) }
nil
end
end
end
Version data entries
6 entries across 6 versions & 1 rubygems