Support BEGIN blocks

lib/typeprof/core/ast.rb

@@ -278,6 +278,7 @@ def self.create_node(raw_node, lenv, use_result = true, allow_meta = false)
       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 :pre_execution_node then PreExecutionNode.new(raw_node, lenv)
       when :post_execution_node then PostExecutionNode.new(raw_node, lenv)
       when :flip_flop_node then FlipFlopNode.new(raw_node, lenv)
       when :shareable_constant_node then create_node(raw_node.write, lenv)

lib/typeprof/core/ast/misc.rb

@@ -23,8 +23,13 @@ def install0(genv)
 
         post_stmts = []
 
+        # BEGIN runs before the surrounding statements, END after them
         @stmts.each do |stmt|
-          next if stmt.nil?
+          stmt.install(genv) if stmt.is_a?(PreExecutionNode)
+        end
+
+        @stmts.each do |stmt|
+          next if stmt.is_a?(PreExecutionNode)
 
           if stmt.is_a?(PostExecutionNode)
             post_stmts << stmt

lib/typeprof/core/ast/variable.rb

@@ -257,7 +257,7 @@ def install0(genv)
       end
     end
 
-    class PostExecutionNode < Node
+    class ExecutionBaseNode < 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)
@@ -273,6 +273,12 @@ def install0(genv)
       end
     end
 
+    class PreExecutionNode < ExecutionBaseNode
+    end
+
+    class PostExecutionNode < ExecutionBaseNode
+    end
+
     class ClassVariableWriteNode < Node
       def initialize(raw_node, rhs, lenv)
         super(raw_node, lenv)

scenario/misc/pre-exec.rb

@@ -0,0 +1,24 @@
+## update
+def check(x)
+end
+
+def check2(x)
+end
+
+y = "str"
+
+BEGIN {
+  def hello = "hello"
+  x = 1
+  y = 1
+}
+
+check(x)
+check2(y)
+
+## assert
+class Object
+  def check: (Integer) -> nil
+  def check2: (String) -> nil
+  def hello: -> String
+end