Class: PinkSpoon::ConstantResolver::AssignmentFinder

Inherits:

Prism::Visitor

• Object
• Prism::Visitor
• PinkSpoon::ConstantResolver::AssignmentFinder

Defined in:

lib/pink_spoon/constant_resolver.rb

Instance Method Summary

• #find(ast) ⇒ Object
• #initialize(const_name) ⇒ AssignmentFinder constructor

  A new instance of AssignmentFinder.

• #visit_constant_path_write_node(node) ⇒ Object
• #visit_constant_write_node(node) ⇒ Object

Constructor Details

#initialize(const_name) ⇒ AssignmentFinder

Returns a new instance of AssignmentFinder.

# File 'lib/pink_spoon/constant_resolver.rb', line 381

def initialize(const_name)
  @const_name = const_name
  @result     = nil
end

Instance Method Details

#find(ast) ⇒ Object

# File 'lib/pink_spoon/constant_resolver.rb', line 386

def find(ast)
  visit(ast)
  @result
end

#visit_constant_path_write_node(node) ⇒ Object

# File 'lib/pink_spoon/constant_resolver.rb', line 396

def visit_constant_path_write_node(node)
  @result = node.value if node.target.slice == @const_name
  super
end

#visit_constant_write_node(node) ⇒ Object

# File 'lib/pink_spoon/constant_resolver.rb', line 391

def visit_constant_write_node(node)
  @result = node.value if node.name.to_s == @const_name
  super
end