Tentative support of post_execution_node

lib/typeprof/core/ast.rb

@@ -216,6 +216,7 @@ def self.create_node(raw_node, lenv, use_result = true)
       when :splat_node then SplatNode.new(raw_node, lenv)
       when :for_node then ForNode.new(raw_node, lenv)
       when :alias_global_variable_node then AliasGlobalVariableNode.new(raw_node, lenv)
+      when :post_execution_node then PostExecutionNode.new(raw_node, lenv)
 
       # call
       when :super_node then SuperNode.new(raw_node, lenv)

lib/typeprof/core/ast/variable.rb

@@ -213,6 +213,22 @@ def install0(genv)
       end
     end
 
+    class PostExecutionNode < Node
+      def initialize(raw_node, lenv)
+        super(raw_node, lenv)
+        @body = raw_node.statements ? AST.create_node(raw_node.statements, lenv) : DummyNilNode.new(TypeProf::CodeRange.new(code_range.last, code_range.last), lenv)
+      end
+
+      attr_reader :body
+
+      def subnodes = { body: }
+
+      def install0(genv)
+        @body.install(genv)
+        Source.new(genv.nil_type)
+      end
+    end
+
     class ClassVariableWriteNode < Node
       def initialize(raw_node, rhs, lenv)
         super(raw_node, lenv)

scenario/known-issues/post-exec.rb

@@ -0,0 +1,15 @@
+## update
+def check(x)
+end
+
+def foo
+  x = 1
+  END { check(x) }
+  x = "str"
+end
+
+## assert
+class Object
+  def check: (String) -> nil
+  def foo: -> nil
+end

scenario/misc/post-exec-tmp.rb

@@ -0,0 +1,16 @@
+## update
+def check(x)
+end
+
+def foo
+  x = 1
+  END { check(x) } # TODO: This shoud pass String, but it is not implemented yet
+  # See known-issues/post-exec.rb
+  x = "str"
+end
+
+## assert
+class Object
+  def check: (Integer) -> nil
+  def foo: -> String
+end