Class: AyeVar::Processor

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

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Processor

Returns a new instance of Processor.



37
38
39
40
 
# File 'lib/aye_var.rb', line 37

def initialize
  @context = Set[]
  @annotations = []
end

Instance Attribute Details

#annotations ⇒ Object (readonly)

Returns the value of attribute annotations.



42
43
44
 
# File 'lib/aye_var.rb', line 42

def annotations
  @annotations
end

Class Method Details

.call(source) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
 
# File 'lib/aye_var.rb', line 17

def self.call(source)
  visitor = new
  visitor.visit(Prism.parse(source).value)

  buffer = source.dup

  visitor.annotations.sort_by(&:first).reverse_each do |offset, action, name|
    case action
    when :start
      buffer.insert(offset, "((::Kernel.raise ::AyeVar::NameError.new('Undefined instance variable #{name}') unless defined?(#{name})); ")
    when :end
      buffer.insert(offset, ")")
    else
      raise "Invalid annotation"
    end
  end

  buffer
end

Instance Method Details

#visit_block_node(node) ⇒ Object 



52
53
54
 
# File 'lib/aye_var.rb', line 52

def visit_block_node(node)
  new_context { super }
end

#visit_case_node(node) ⇒ Object 



79
80
81
82
83
84
85
86
87
 
# File 'lib/aye_var.rb', line 79

def visit_case_node(node)
  visit(node.predicate)

  node.conditions.each do |condition|
    branch { visit(condition) }
  end

  branch { visit(node.else_clause) }
end

#visit_class_node(node) ⇒ Object 



44
45
46
 
# File 'lib/aye_var.rb', line 44

def visit_class_node(node)
  new_context { super }
end

#visit_def_node(node) ⇒ Object 



68
69
70
 
# File 'lib/aye_var.rb', line 68

def visit_def_node(node)
  new_context { super }
end

#visit_defined_node(node) ⇒ Object 



60
61
62
63
64
65
66
 
# File 'lib/aye_var.rb', line 60

def visit_defined_node(node)
  if Prism::InstanceVariableReadNode === node.value
    @context << node.value.name
  end

  super
end

#visit_if_node(node) ⇒ Object 



72
73
74
75
76
77
 
# File 'lib/aye_var.rb', line 72

def visit_if_node(node)
  visit(node.predicate)

  branch { visit(node.statements) }
  branch { visit(node.subsequent) }
end

#visit_instance_variable_read_node(node) ⇒ Object 



89
90
91
92
93
94
95
96
97
98
99
100
101
102
 
# File 'lib/aye_var.rb', line 89

def visit_instance_variable_read_node(node)
  name = node.name

  unless @context.include?(name)
    location = node.location

    @context << name

    @annotations << [location.start_character_offset, :start, name]
    @annotations << [location.end_character_offset, :end, name]
  end

  super
end

#visit_module_node(node) ⇒ Object 



48
49
50
 
# File 'lib/aye_var.rb', line 48

def visit_module_node(node)
  new_context { super }
end

#visit_singleton_class_node(node) ⇒ Object 



56
57
58
 
# File 'lib/aye_var.rb', line 56

def visit_singleton_class_node(node)
  new_context { super }
end