Support arguments for break

Author: mame

doc/doc.ja.md

@@ -131,4 +131,4 @@ Procオブジェクトは、ラムダ式（`-> { ... }`）やブロック仮引
 これらは抽象化されず、コード片と結びついた具体的な値として扱われます。
 これらに渡された引数や返された値によってRBS出力されます。
 
-TBD
+TODO: write more

lib/typeprof/core/ast/call.rb

@@ -124,7 +124,15 @@ def install0(genv)
 
         a_args = ActualArguments.new(positional_args, @splat_flags, keyword_args, blk_ty)
         box = @changes.add_method_call_box(genv, recv, @mid, a_args, !@recv)
-        box.ret
+
+        if @block_body && @block_body.lenv.break_vtx
+          ret = Vertex.new(self)
+          @changes.add_edge(genv, box.ret, ret)
+          @changes.add_edge(genv, @block_body.lenv.break_vtx, ret)
+          ret
+        else
+          box.ret
+        end
       end
 
       def block_last_stmt_code_range

lib/typeprof/core/ast/control.rb

@@ -176,16 +176,17 @@ class UntilNode < LoopNode
     class BreakNode < Node
       def initialize(raw_node, lenv)
         super(raw_node, lenv)
-        @arg = raw_node.arguments ? AST.create_node(raw_node.arguments.arguments.first, lenv) : nil
+        @arg = AST.parse_return_arguments(raw_node, lenv, code_range)
       end
 
       attr_reader :arg
 
       def subnodes = { arg: }
 
       def install0(genv)
-        _arg = @arg ? @arg.install(genv) : Source.new(genv.nil_type)
-        # TODO: implement!
+        arg = @arg.install(genv)
+        @changes.add_edge(genv, arg, @lenv.get_break_vtx)
+        Source.new()
       end
     end
 

lib/typeprof/core/env.rb

@@ -282,11 +282,12 @@ def initialize(path, cref, locals, return_boxes)
       @cref = cref
       @locals = locals
       @return_boxes = return_boxes
+      @break_vtx = nil
       @next_boxes = []
       @filters = {}
     end
 
-    attr_reader :path, :cref, :locals, :return_boxes, :next_boxes
+    attr_reader :path, :cref, :locals, :return_boxes, :break_vtx, :next_boxes
 
     def new_var(name, node)
       @locals[name] = Vertex.new(node)
@@ -312,6 +313,11 @@ def add_next_box(box)
       @next_boxes << box
     end
 
+    def get_break_vtx
+      @break_vtx ||= Vertex.new(:break_vtx)
+    end
+
+
     def push_read_filter(name, type)
       (@filters[name] ||= []) << type
     end

lib/typeprof/core/graph/vertex.rb

@@ -146,6 +146,7 @@ def initialize(origin)
       when ValueEntity
       when ActualArguments
       when Array
+      when Symbol
       else
         raise "unknown class: #{ origin.class }"
       end

scenario/control/break.rb

@@ -16,5 +16,17 @@ def bar
 ## assert
 class Object
   def foo: { (Integer) -> Float } -> String
-  def bar: -> String
+  def bar: -> String?
+end
+
+## update: test.rb
+def foo
+  loop do
+    break 1
+  end
+end
+
+## assert
+class Object
+  def foo: -> Integer
 end