Update module behavior to add a JIT Init and Shutdown step

Author: ircmaxell

composer.lock

@@ -12,6 +12,7 @@
 use PHPCfg\Operand;
 use PHPCompiler\Runtime;
 use PHPCompiler\Block;
+use PHPCompiler\Module;
 use PHPCompiler\VM\Variable as VMVariable;
 use PHPTypes\Type;
 
@@ -43,6 +44,7 @@ class Context {
     private array $stringConstant = [];
     private array $builtins;
     private array $stringConstantMap = [];
+    private array $modules = [];
 
     private ?Result $result = null;
     public Builtin\MemoryManager $memory;
@@ -117,6 +119,11 @@ public function popScope(): void {
         $this->scope = array_pop($this->scopeStack);
     }
 
+    public function registerModule(Module $module): void {
+        $this->modules[] = $module;
+        $module->jitInit($this);
+    }
+
     public function registerBuiltin(Builtin $builtin): void {
         $this->builtins[] = $builtin;
     }
@@ -196,6 +203,9 @@ public function compileInPlace() {
     }
 
     private function compileCommon() {
+        foreach ($this->modules as $module) {
+            $module->jitShutdown($this);
+        }
         foreach ($this->builtins as $builtin) {
             $builtin->shutdown();
         }

lib/JIT/Context.php

@@ -19,5 +19,9 @@ public function getFunctions(): array;
 
     public function init(Runtime $runtime): void;
 
+    public function jitInit(JIT\Context $context): void;
+
+    public function jitShutdown(JIT\Context $context): void;
+
     public function shutdown(): void;
 }

lib/Module.php

@@ -30,6 +30,14 @@ public function shutdown(): void {
 
     }
 
+    public function jitInit(JIT\Context $context): void {
+        
+    }
+
+    public function jitShutdown(JIT\Context $context): void {
+
+    }
+
     protected function parseAndCompileFunction(string $name, string $filename): Func {
         $script = $this->runtime->parse(file_get_contents($filename), $filename);
         $func = $this->findFunction($name, $script);

lib/ModuleAbstract.php

@@ -107,6 +107,9 @@ public function loadJitContext(): JITContext {
             if (!is_null($this->debugFile)) {
                 $this->jitContext->setDebugFile($this->debugFile);
             }
+            foreach ($this->modules as $module) {
+                $this->jitContext->registerModule($module);
+            }
         }
         return $this->jitContext;
     }