@vm0/runner 2.0.4 → 2.1.0

package/index.js

@@ -5,6 +5,8 @@ import { program } from "commander";
 
 // src/commands/start.ts
 import { Command } from "commander";
+import { writeFileSync } from "fs";
+import { dirname, join } from "path";
 
 // src/lib/config.ts
 import { z } from "zod";
@@ -9246,6 +9248,22 @@ async function executeJob(context, config) {
 
 // src/commands/start.ts
 var activeJobs = /* @__PURE__ */ new Set();
+function writeStatusFile(statusFilePath, mode, startedAt) {
+  const status = {
+    mode,
+    active_jobs: activeJobs.size,
+    active_job_ids: Array.from(activeJobs),
+    started_at: startedAt.toISOString(),
+    updated_at: (/* @__PURE__ */ new Date()).toISOString()
+  };
+  try {
+    writeFileSync(statusFilePath, JSON.stringify(status, null, 2));
+  } catch (err) {
+    console.error(
+      `Failed to write status file: ${err instanceof Error ? err.message : "Unknown error"}`
+    );
+  }
+}
 async function executeJob2(context, config) {
   console.log(`  Executing job ${context.runId}...`);
   console.log(`  Prompt: ${context.prompt.substring(0, 100)}...`);
@@ -9280,26 +9298,61 @@ var startCommand = new Command("start").description("Start the runner").option("
     }
     console.log("Setting up network bridge...");
     await setupBridge();
+    const statusFilePath = join(dirname(options.config), "status.json");
+    const startedAt = /* @__PURE__ */ new Date();
+    const state = { mode: "running" };
+    const updateStatus = () => {
+      writeStatusFile(statusFilePath, state.mode, startedAt);
+    };
     console.log(
       `Starting runner '${config.name}' for group '${config.group}'...`
     );
     console.log(`Max concurrent jobs: ${config.sandbox.max_concurrent}`);
+    console.log(`Status file: ${statusFilePath}`);
     console.log("Press Ctrl+C to stop");
     console.log("");
+    updateStatus();
     let running = true;
     process.on("SIGINT", () => {
       console.log("\nShutting down...");
       running = false;
+      state.mode = "stopped";
+      updateStatus();
     });
     process.on("SIGTERM", () => {
       console.log("\nShutting down...");
       running = false;
+      state.mode = "stopped";
+      updateStatus();
+    });
+    process.on("SIGUSR1", () => {
+      if (state.mode === "running") {
+        console.log("\n[Maintenance] Entering drain mode...");
+        console.log(
+          `[Maintenance] Active jobs: ${activeJobs.size} (will wait for completion)`
+        );
+        state.mode = "draining";
+        updateStatus();
+      }
     });
     const jobPromises = /* @__PURE__ */ new Set();
     while (running) {
+      if (state.mode === "draining") {
+        if (activeJobs.size === 0) {
+          console.log("[Maintenance] All jobs completed, exiting drain mode");
+          running = false;
+          break;
+        }
+        if (jobPromises.size > 0) {
+          await Promise.race(jobPromises);
+          updateStatus();
+        }
+        continue;
+      }
       if (activeJobs.size >= config.sandbox.max_concurrent) {
         if (jobPromises.size > 0) {
           await Promise.race(jobPromises);
+          updateStatus();
         }
         continue;
       }
@@ -9316,6 +9369,7 @@ var startCommand = new Command("start").description("Start the runner").option("
           const context = await claimJob(config.server, job.runId);
           console.log(`Claimed job: ${context.runId}`);
           activeJobs.add(context.runId);
+          updateStatus();
           const jobPromise = executeJob2(context, config).catch((error) => {
             console.error(
               `Job ${context.runId} failed:`,
@@ -9324,6 +9378,7 @@ var startCommand = new Command("start").description("Start the runner").option("
           }).finally(() => {
             activeJobs.delete(context.runId);
             jobPromises.delete(jobPromise);
+            updateStatus();
           });
           jobPromises.add(jobPromise);
         } catch (error) {
@@ -9346,6 +9401,8 @@ var startCommand = new Command("start").description("Start the runner").option("
       );
       await Promise.all(jobPromises);
     }
+    state.mode = "stopped";
+    updateStatus();
     console.log("Runner stopped");
     process.exit(0);
   } catch (error) {
@@ -9366,7 +9423,7 @@ var statusCommand = new Command2("status").description("Show runner status").act
 });
 
 // src/index.ts
-var version = true ? "2.0.4" : "0.1.0";
+var version = true ? "2.1.0" : "0.1.0";
 program.name("vm0-runner").version(version).description("Self-hosted runner for VM0 agents");
 program.addCommand(startCommand);
 program.addCommand(statusCommand);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vm0/runner",
-  "version": "2.0.4",
+  "version": "2.1.0",
   "description": "Self-hosted runner for VM0 agents",
   "repository": {
     "type": "git",