feat: integrate JiT compilation and finalize test suite

* feat: enable JiT compilation in CLI run, build and compile commands

- Connect main CLI execution flow to the JiT worker process

- Add comprehensive JiT test suite

- Finalize AoT build logic to preserve JiT-specific fields

- Update CLI console and entry points for integrated JiT support

Author: rafal-hawrylak

cli/BUILD

@@ -65,15 +65,13 @@ ts_test_suite(
         "index_project_test.ts",
         "index_compile_test.ts",
         "index_run_e2e_test.ts",
-        "index_jit_main_test.ts",
-        "index_jit_advanced_test.ts",
-        "index_jit_dependency_test.ts",
-        "index_jit_runtime_test.ts",
+        "tests/jit/index_jit_main_test.ts",
+        "tests/jit/index_jit_advanced_test.ts",
+        "tests/jit/index_jit_dependency_test.ts",
+        "tests/jit/index_jit_runtime_test.ts",
         "util_test.ts",
-        "jit_build_test.ts",
-        "jit_rpc_test.ts",
-        "jit_run_test.ts",
-        "bigquery_test.ts"
+        "tests/jit/jit_build_test.ts",
+        "tests/jit/jit_run_test.ts",
     ],
     data = [
         ":node_modules",

cli/api/commands/base_worker.ts

@@ -15,14 +15,15 @@ export abstract class BaseWorker<TResponse, TMessage = any> {
 
     return new Promise((resolve, reject) => {
       let completed = false;
+      let booted = false;
 
       const terminate = (fn: () => void) => {
         if (completed) {
           return;
         }
         completed = true;
         clearTimeout(timeout);
-        child.kill();
+        child.kill("SIGKILL");
         fn();
       };
 
@@ -34,7 +35,10 @@ export abstract class BaseWorker<TResponse, TMessage = any> {
 
       child.on("message", (message: any) => {
         if (message.type === "worker_booted") {
-          onBoot(child);
+          if (!booted) {
+            booted = true;
+            onBoot(child);
+          }
           return;
         }
         onMessage(message, child, (res) => terminate(() => resolve(res)), (err) => terminate(() => reject(err)));

cli/api/commands/build.ts

@@ -83,9 +83,7 @@ export class Builder {
       ...this.toPartialExecutionAction(table),
       type: "table",
       tableType: utils.tableTypeEnumToString(table.enumType),
-      tasks: table.disabled
-        ? []
-        : this.executionSql.publishTasks(table, runConfig, tableMetadata).build(),
+      tasks: this.executionSql.createTableTasks(table, runConfig, tableMetadata),
       hermeticity: table.hermeticity || dataform.ActionHermeticity.HERMETIC
     };
   }
@@ -94,9 +92,7 @@ export class Builder {
     return {
       ...this.toPartialExecutionAction(operation),
       type: "operation",
-      tasks: operation.disabled
-        ? []
-        : operation.queries.map(statement => ({ type: "statement", statement })),
+      tasks: this.executionSql.createOperationTasks(operation),
       hermeticity: operation.hermeticity || dataform.ActionHermeticity.NON_HERMETIC
     };
   }
@@ -105,9 +101,7 @@ export class Builder {
     return {
       ...this.toPartialExecutionAction(assertion),
       type: "assertion",
-      tasks: assertion.disabled
-        ? []
-        : this.executionSql.assertTasks(assertion, this.prunedGraph.projectConfig).build(),
+      tasks: this.executionSql.createAssertionTasks(assertion),
       hermeticity: assertion.hermeticity || dataform.ActionHermeticity.HERMETIC
     };
   }
@@ -126,9 +120,6 @@ export class Builder {
     if (jitCode) {
       executionAction.jitCode = jitCode;
     }
-    if (this.prunedGraph.jitData) {
-      executionAction.jitData = this.prunedGraph.jitData;
-    }
     return executionAction;
   }
 }