Class: Spektr::Extractors::Methods

Inherits:
Prism::Visitor
  • Object
show all
Defined in:
lib/spektr/extractors/methods.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(visibility: :all) ⇒ Methods

Returns a new instance of Methods.



6
7
8
9
10
 
# File 'lib/spektr/extractors/methods.rb', line 6

def initialize(visibility: :all)
  @visibility = visibility
  @current_visibility = :public
  @result = []
end

Instance Attribute Details

#result ⇒ Object

Returns the value of attribute result.



4
5
6
 
# File 'lib/spektr/extractors/methods.rb', line 4

def result
  @result
end

Instance Method Details

#call(ast) ⇒ Object 



12
13
14
15
 
# File 'lib/spektr/extractors/methods.rb', line 12

def call(ast)
  ast.value.accept(self)
  self
end

#visit_call_node(node) ⇒ Object 



17
18
19
20
 
# File 'lib/spektr/extractors/methods.rb', line 17

def visit_call_node(node)
  @current_visibility = node.name if %i[private protected public].include?(node.name)
  super
end

#visit_def_node(node) ⇒ Object 



22
23
24
25
 
# File 'lib/spektr/extractors/methods.rb', line 22

def visit_def_node(node)
  @result << node if @visibility == :all || @current_visibility == @visibility
  super
end