Force to make the result of "initialize" method void

Author: mame

lib/typeprof/core/ast/base.rb

@@ -92,6 +92,7 @@ def undefine0(genv)
         end
       end
 
+      #: (TypeProf::Core::GlobalEnv) -> TypeProf::Core::BasicVertex
       def install(genv)
         @ret = install0(genv)
         @changes.reinstall(genv)
@@ -111,6 +112,7 @@ def install_copy(genv)
         @ret = (@prev_node || raise).ret
       end
 
+      #: (TypeProf::Core::GlobalEnv) -> untyped
       def install0(_)
         raise "should override"
       end

lib/typeprof/core/ast/call.rb

@@ -190,10 +190,12 @@ def retrieve_at(pos, &blk)
       def modified_vars(tbl, vars)
         subnodes.each do |key, subnode|
           next unless subnode
-          if key == :block_body
-            subnode.modified_vars(tbl - self.block_tbl, vars)
-          elsif subnode.is_a?(AST::Node)
-            subnode.modified_vars(tbl, vars)
+          if subnode.is_a?(AST::Node)
+            if key == :block_body
+              subnode.modified_vars(tbl - self.block_tbl, vars)
+            else
+              subnode.modified_vars(tbl, vars)
+            end
           else
             subnode.each {|n| n.modified_vars(tbl, vars) }
           end

lib/typeprof/core/env/method.rb

@@ -82,6 +82,7 @@ def get_keyword_arg(genv, changes, name)
   end
 
   class Block
+    #: (AST::CallBaseNode, Vertex, Array[Vertex], Array[EscapeBox]) -> void
     def initialize(node, f_ary_arg, f_args, next_boxes)
       @node = node
       @f_ary_arg = f_ary_arg

lib/typeprof/core/graph/box.rb

@@ -664,7 +664,7 @@ def show(output_parameter_names)
       args = args.join(", ")
       s = args.empty? ? [] : ["(#{ args })"]
       s << "#{ block_show.sort.join(" | ") }" unless block_show.empty?
-      s << "-> #{ @ret.show }"
+      s << "-> #{ @mid == :initialize ? "void" : @ret.show }"
       s.join(" ")
     end
   end

lib/typeprof/core/graph/vertex.rb

@@ -63,6 +63,10 @@ def show
         end
       end
     end
+
+    def new_vertex(genv, origin)
+      raise NotImplementedError
+    end
   end
 
   class Source < BasicVertex

scenario/args/keyword-twice.rb

@@ -12,7 +12,7 @@ class Bar < Foo
 
 ## assert
 class Foo
-  def initialize: (k: Integer) -> nil
+  def initialize: (k: Integer) -> void
 end
 class Bar < Foo
 end
@@ -31,7 +31,7 @@ class Bar < Foo
 
 ## assert
 class Foo
-  def initialize: (?k: :default | Integer) -> nil
+  def initialize: (?k: :default | Integer) -> void
 end
 class Bar < Foo
 end

scenario/args/positionals_rb_to_rb_block.rb

@@ -16,5 +16,5 @@ def initialize(n)
 
 ## assert
 class Foo
-  def initialize: (Integer) { (String, Float) -> (Integer | String) } -> nil
+  def initialize: (Integer) { (String, Float) -> (Integer | String) } -> void
 end

scenario/class/basic1.rb

@@ -13,6 +13,6 @@ def foo(n)
 
 ## assert
 class C
-  def initialize: (Integer) -> Integer
+  def initialize: (Integer) -> void
   def foo: (String) -> singleton(C)
 end

scenario/class/initialize.rb

@@ -19,7 +19,7 @@ def foo
 class A
 end
 class B
-  def initialize: (Integer) -> Integer
+  def initialize: (Integer) -> void
 end
 class C
 end
@@ -31,4 +31,4 @@ def foo: -> B
 Integer
 
 ## hover: test.rb:14:6
-B#initialize : (Integer) -> Integer
+B#initialize : (Integer) -> void

scenario/class/initialize2.rb

@@ -18,10 +18,10 @@ def my_new(klass)
 
 ## assert
 class Foo
-  def initialize: (Integer) -> nil
+  def initialize: (Integer) -> void
 end
 class Bar
-  def initialize: (Integer) -> nil
+  def initialize: (Integer) -> void
 end
 class Object
   def my_new: (singleton(Bar) | singleton(Foo)) -> (Bar | Foo)