Class: PinkSpoon::ConstantResolver::IvarTypeFinder

Inherits:
Prism::Visitor
  • Object
show all
Defined in:
lib/pink_spoon/constant_resolver.rb

Overview

Collects all RHS nodes for ‘@ivar = rhs` and `@ivar ||= rhs` writes.

Instance Method Summary collapse

Constructor Details

#initialize(ivar_name) ⇒ IvarTypeFinder

Returns a new instance of IvarTypeFinder.



422
423
424
425
 
# File 'lib/pink_spoon/constant_resolver.rb', line 422

def initialize(ivar_name)
  @ivar_name = ivar_name
  @results   = []
end

Instance Method Details

#find_all(ast) ⇒ Object 



427
428
429
430
 
# File 'lib/pink_spoon/constant_resolver.rb', line 427

def find_all(ast)
  visit(ast)
  @results
end

#visit_instance_variable_or_write_node(node) ⇒ Object 



437
438
439
440
 
# File 'lib/pink_spoon/constant_resolver.rb', line 437

def visit_instance_variable_or_write_node(node)
  @results << node.value if node.name.to_s == @ivar_name
  super
end

#visit_instance_variable_write_node(node) ⇒ Object 



432
433
434
435
 
# File 'lib/pink_spoon/constant_resolver.rb', line 432

def visit_instance_variable_write_node(node)
  @results << node.value if node.name.to_s == @ivar_name
  super
end