From 58581d6bf004210183a6c0e955cdbcc1e3dadca2 Mon Sep 17 00:00:00 2001
From: sergeych <real.sergeych@gmail.com>
Date: Mon, 9 Feb 2026 02:09:29 +0300
Subject: [PATCH] Step 9: module-level bytecode execution

---
 bytecode_migration_plan.md                    |  6 ++---
 .../kotlin/net/sergeych/lyng/Compiler.kt      | 27 +++++++++----------
 .../kotlin/net/sergeych/lyng/Script.kt        |  7 +++++
 3 files changed, 23 insertions(+), 17 deletions(-)

diff --git a/bytecode_migration_plan.md b/bytecode_migration_plan.md
index e3ba613..8c470c8 100644
--- a/bytecode_migration_plan.md
+++ b/bytecode_migration_plan.md
@@ -29,9 +29,9 @@ Goal: migrate the compiler so all values live in frames/bytecode, keeping JVM te
   - [x] Support `ClassScopeMemberRef` without scope-map fallback.
 - [x] Step 8: ObjDynamic member access in bytecode.
   - [x] Allow dynamic receiver field/method lookup without falling back to interpreter.
-- [ ] Step 9: Module-level bytecode execution.
-  - [ ] Compile `Script` bodies to bytecode instead of interpreting at module scope.
-  - [ ] Keep import/module slot seeding in frame-only flow.
+- [x] Step 9: Module-level bytecode execution.
+  - [x] Compile `Script` bodies to bytecode instead of interpreting at module scope.
+  - [x] Keep import/module slot seeding in frame-only flow.
 
 ## Notes
 
diff --git a/lynglib/src/commonMain/kotlin/net/sergeych/lyng/Compiler.kt b/lynglib/src/commonMain/kotlin/net/sergeych/lyng/Compiler.kt
index e771dff..e0b155b 100644
--- a/lynglib/src/commonMain/kotlin/net/sergeych/lyng/Compiler.kt
+++ b/lynglib/src/commonMain/kotlin/net/sergeych/lyng/Compiler.kt
@@ -1526,25 +1526,24 @@ class Compiler(
                     resolutionScriptDepth == 1 &&
                     statements.none { containsUnsupportedForBytecode(it) } &&
                     statements.none { containsDelegatedRefs(it) }
-                val finalStatements = if (wrapScriptBytecode) {
+                val (finalStatements, moduleBytecode) = if (wrapScriptBytecode) {
                     val unwrapped = statements.map { unwrapBytecodeDeep(it) }
                     val block = InlineBlockStatement(unwrapped, start)
-                    listOf(
-                        BytecodeStatement.wrap(
-                            block,
-                            "<script>",
-                            allowLocalSlots = true,
-                            allowedScopeNames = modulePlan.keys,
-                            moduleScopeId = moduleSlotPlan()?.id,
-                            slotTypeByScopeId = slotTypeByScopeId,
-                            knownNameObjClass = knownClassMapForBytecode()
-                        )
-                    )
+                    val bytecodeStmt = BytecodeStatement.wrap(
+                        block,
+                        "<script>",
+                        allowLocalSlots = true,
+                        allowedScopeNames = modulePlan.keys,
+                        moduleScopeId = moduleSlotPlan()?.id,
+                        slotTypeByScopeId = slotTypeByScopeId,
+                        knownNameObjClass = knownClassMapForBytecode()
+                    ) as BytecodeStatement
+                    unwrapped to bytecodeStmt.bytecodeFunction()
                 } else {
-                    statements
+                    statements to null
                 }
                 val moduleRefs = importedModules.map { ImportBindingSource.Module(it.scope.packageName, it.pos) }
-                Script(start, finalStatements, modulePlan, importBindings.toMap(), moduleRefs)
+                Script(start, finalStatements, modulePlan, importBindings.toMap(), moduleRefs, moduleBytecode)
             }.also {
                 // Best-effort script end notification (use current position)
                 miniSink?.onScriptEnd(
diff --git a/lynglib/src/commonMain/kotlin/net/sergeych/lyng/Script.kt b/lynglib/src/commonMain/kotlin/net/sergeych/lyng/Script.kt
index 625ed1e..c0a1d63 100644
--- a/lynglib/src/commonMain/kotlin/net/sergeych/lyng/Script.kt
+++ b/lynglib/src/commonMain/kotlin/net/sergeych/lyng/Script.kt
@@ -20,6 +20,8 @@ package net.sergeych.lyng
 import kotlinx.coroutines.delay
 import kotlinx.coroutines.yield
 import net.sergeych.lyng.Script.Companion.defaultImportManager
+import net.sergeych.lyng.bytecode.CmdFunction
+import net.sergeych.lyng.bytecode.CmdVm
 import net.sergeych.lyng.miniast.*
 import net.sergeych.lyng.obj.*
 import net.sergeych.lyng.pacman.ImportManager
@@ -35,10 +37,12 @@ class Script(
     private val moduleSlotPlan: Map<String, Int> = emptyMap(),
     private val importBindings: Map<String, ImportBinding> = emptyMap(),
     private val importedModules: List<ImportBindingSource.Module> = emptyList(),
+    private val moduleBytecode: CmdFunction? = null,
 //    private val catchReturn: Boolean = false,
 ) : Statement() {
 
     override suspend fun execute(scope: Scope): Obj {
+        scope.pos = pos
         val isModuleScope = scope is ModuleScope
         val shouldSeedModule = isModuleScope || scope.thisObj === ObjVoid
         val moduleTarget = scope
@@ -79,6 +83,9 @@ class Script(
         if (shouldSeedModule) {
             seedModuleSlots(moduleTarget)
         }
+        moduleBytecode?.let { fn ->
+            return CmdVm().execute(fn, scope, scope.args.list)
+        }
         var lastResult: Obj = ObjVoid
         for (s in statements) {
             lastResult = s.execute(scope)