Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions lib/typeprof/core/ast.rb
Original file line number Diff line number Diff line change
Expand Up @@ -92,6 +92,8 @@ def self.create_node(raw_node, lenv, use_result = true)
# variables
when :local_variable_read_node
LocalVariableReadNode.new(raw_node, lenv)
when :it_local_variable_read_node
ItLocalVariableReadNode.new(raw_node, lenv)
when :local_variable_write_node
LocalVariableWriteNode.new(raw_node, AST.create_node(raw_node.value, lenv), lenv)
when :local_variable_operator_write_node
Expand Down
2 changes: 2 additions & 0 deletions lib/typeprof/core/ast/call.rb
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,8 @@ def initialize(raw_node, recv, mid, mid_code_range, raw_args, last_arg, raw_bloc
raw_block.parameters.parameters.requireds.map {|n| n.is_a?(Prism::MultiTargetNode) ? nil : n.name }
when Prism::NumberedParametersNode
1.upto(raw_block.parameters.maximum).map { |n| :"_#{n}" }
when Prism::ItParametersNode
[:it]
when nil
[]
else
Expand Down
16 changes: 16 additions & 0 deletions lib/typeprof/core/ast/variable.rb
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,22 @@ def retrieve_at(pos)
end
end

class ItLocalVariableReadNode < Node
def initialize(raw_node, lenv)
super(raw_node, lenv)
end

def attrs = {}

def install0(genv)
@lenv.get_var(:it)
end

def retrieve_at(pos)
yield self if code_range.include?(pos)
end
end

class LocalVariableWriteNode < Node
def initialize(raw_node, rhs, lenv)
super(raw_node, lenv)
Expand Down
13 changes: 13 additions & 0 deletions scenario/block/it.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
## update
def foo(&b)
b.call(1)
end

foo do
it
end

## assert
class Object
def foo: { (Integer) -> Integer } -> Integer
end