labgate 0.5.21 → 0.5.23

package/dist/mcp-bundles/slurm-mcp.bundle.mjs

@@ -29945,7 +29945,7 @@ var StdioServerTransport = class {
 };
 
 // src/lib/slurm-db.ts
-import { existsSync, mkdirSync, chmodSync } from "fs";
+import { existsSync, mkdirSync, chmodSync, readFileSync, writeFileSync } from "fs";
 import { dirname } from "path";
 var TERMINAL_STATES = /* @__PURE__ */ new Set([
   "COMPLETED",
@@ -29998,30 +29998,43 @@ CREATE INDEX IF NOT EXISTS idx_slurm_jobs_submit_time ON slurm_jobs(submit_time)
 `;
 var SlurmJobDB = class {
   db;
+  useJsonFallback = false;
+  jsonStorePath = null;
+  jsonJobs = /* @__PURE__ */ new Map();
   constructor(dbPath) {
-    if (!Database) {
-      throw new Error(
-        'SLURM tracking requires the "better-sqlite3" package. Install it with: npm install better-sqlite3'
-      );
-    }
     const dir = dirname(dbPath);
     if (!existsSync(dir)) {
       mkdirSync(dir, { recursive: true });
     }
-    this.db = new Database(dbPath);
-    try {
-      chmodSync(dbPath, 384);
-    } catch {
-    }
-    this.db.pragma("journal_mode = WAL");
-    this.db.pragma("busy_timeout = 5000");
-    this.db.exec(SCHEMA);
-    try {
-      this.db.exec("ALTER TABLE slurm_jobs ADD COLUMN notes TEXT");
-    } catch {
+    const forceJson = process.env.LABGATE_SLURM_DB_FORCE_JSON === "1";
+    if (Database && !forceJson) {
+      try {
+        this.db = new Database(dbPath);
+        try {
+          chmodSync(dbPath, 384);
+        } catch {
+        }
+        this.db.pragma("journal_mode = WAL");
+        this.db.pragma("busy_timeout = 5000");
+        this.db.exec(SCHEMA);
+        try {
+          this.db.exec("ALTER TABLE slurm_jobs ADD COLUMN notes TEXT");
+        } catch {
+        }
+        return;
+      } catch {
+      }
     }
+    this.useJsonFallback = true;
+    this.jsonStorePath = `${dbPath}.json`;
+    this.loadJsonStore();
+    this.flushJsonStore();
   }
   upsertJob(job) {
+    if (this.useJsonFallback) {
+      this.upsertJobJson(job);
+      return;
+    }
     const now = (/* @__PURE__ */ new Date()).toISOString();
     const existing = this.getJob(job.job_id);
     if (existing) {
@@ -30110,14 +30123,37 @@ var SlurmJobDB = class {
     }
   }
   updateJobNotes(jobId, notes) {
+    if (this.useJsonFallback) {
+      const existing = this.jsonJobs.get(jobId);
+      if (!existing) return false;
+      this.jsonJobs.set(jobId, { ...existing, notes, updated_at: (/* @__PURE__ */ new Date()).toISOString() });
+      this.flushJsonStore();
+      return true;
+    }
     const now = (/* @__PURE__ */ new Date()).toISOString();
     const result = this.db.prepare("UPDATE slurm_jobs SET notes = ?, updated_at = ? WHERE job_id = ?").run(notes, now, jobId);
     return result.changes > 0;
   }
   getJob(jobId) {
+    if (this.useJsonFallback) {
+      const job = this.jsonJobs.get(jobId);
+      return job ? { ...job } : null;
+    }
     return this.db.prepare("SELECT * FROM slurm_jobs WHERE job_id = ?").get(jobId) ?? null;
   }
   listJobs(opts) {
+    if (this.useJsonFallback) {
+      const limitRaw2 = opts?.limit ?? 100;
+      const offsetRaw2 = opts?.offset ?? 0;
+      const limit2 = Number.isFinite(limitRaw2) ? Math.min(1e3, Math.max(1, Math.floor(limitRaw2))) : 100;
+      const offset2 = Number.isFinite(offsetRaw2) ? Math.max(0, Math.floor(offsetRaw2)) : 0;
+      const search = opts?.search?.toLowerCase();
+      const rows = [...this.jsonJobs.values()].filter((j) => !opts?.state || j.state === opts.state).filter((j) => !opts?.session_id || j.session_id === opts.session_id).filter((j) => {
+        if (!search) return true;
+        return j.job_id.toLowerCase().includes(search) || (j.name || "").toLowerCase().includes(search) || (j.command || "").toLowerCase().includes(search) || (j.notes || "").toLowerCase().includes(search);
+      }).sort((a, b) => a.submit_time < b.submit_time ? 1 : a.submit_time > b.submit_time ? -1 : 0).slice(offset2, offset2 + limit2);
+      return rows.map((r) => ({ ...r }));
+    }
     const conditions = [];
     const values = [];
     if (opts?.state) {
@@ -30141,18 +30177,138 @@ var SlurmJobDB = class {
     return this.db.prepare(`SELECT * FROM slurm_jobs ${where} ORDER BY submit_time DESC LIMIT ? OFFSET ?`).all(...values, limit, offset);
   }
   getActiveJobIds() {
+    if (this.useJsonFallback) {
+      return [...this.jsonJobs.values()].filter((j) => ["PENDING", "RUNNING", "COMPLETING", "SUSPENDED"].includes(j.state)).map((j) => j.job_id);
+    }
     const rows = this.db.prepare("SELECT job_id FROM slurm_jobs WHERE state IN ('PENDING', 'RUNNING', 'COMPLETING', 'SUSPENDED')").all();
     return rows.map((r) => r.job_id);
   }
   getJobCount(state) {
+    if (this.useJsonFallback) {
+      if (state) {
+        let count = 0;
+        for (const job of this.jsonJobs.values()) {
+          if (job.state === state) count++;
+        }
+        return count;
+      }
+      return this.jsonJobs.size;
+    }
     if (state) {
       return this.db.prepare("SELECT COUNT(*) as count FROM slurm_jobs WHERE state = ?").get(state).count;
     }
     return this.db.prepare("SELECT COUNT(*) as count FROM slurm_jobs").get().count;
   }
   close() {
+    if (this.useJsonFallback) {
+      this.flushJsonStore();
+      return;
+    }
     this.db.close();
   }
+  loadJsonStore() {
+    if (!this.jsonStorePath) return;
+    if (!existsSync(this.jsonStorePath)) return;
+    try {
+      const raw = readFileSync(this.jsonStorePath, "utf-8");
+      const parsed = JSON.parse(raw || "{}");
+      const jobs = Array.isArray(parsed?.jobs) ? parsed.jobs : [];
+      for (const row of jobs) {
+        const job = row;
+        if (!job?.job_id) continue;
+        this.jsonJobs.set(job.job_id, job);
+      }
+    } catch {
+      this.jsonJobs.clear();
+    }
+  }
+  flushJsonStore() {
+    if (!this.jsonStorePath) return;
+    try {
+      const payload = {
+        version: 1,
+        jobs: [...this.jsonJobs.values()]
+      };
+      writeFileSync(this.jsonStorePath, JSON.stringify(payload, null, 2) + "\n", { encoding: "utf-8", mode: 384 });
+      try {
+        chmodSync(this.jsonStorePath, 384);
+      } catch {
+      }
+    } catch {
+    }
+  }
+  upsertJobJson(job) {
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const existing = this.jsonJobs.get(job.job_id);
+    if (existing) {
+      const fields = [
+        "name",
+        "state",
+        "submit_time",
+        "start_time",
+        "end_time",
+        "partition",
+        "nodes",
+        "num_nodes",
+        "num_tasks",
+        "cpus_per_task",
+        "mem",
+        "stdout_path",
+        "stderr_path",
+        "stdout_size",
+        "stderr_size",
+        "command",
+        "script_path",
+        "session_id",
+        "workdir",
+        "exit_code",
+        "account",
+        "qos",
+        "time_limit",
+        "notes"
+      ];
+      const updated = { ...existing, updated_at: now };
+      for (const field of fields) {
+        if (job[field] !== void 0) {
+          updated[field] = job[field];
+        }
+      }
+      this.jsonJobs.set(job.job_id, updated);
+      this.flushJsonStore();
+      return;
+    }
+    const inserted = {
+      job_id: job.job_id,
+      name: job.name ?? null,
+      state: job.state ?? "UNKNOWN",
+      submit_time: job.submit_time ?? now,
+      start_time: job.start_time ?? null,
+      end_time: job.end_time ?? null,
+      partition: job.partition ?? null,
+      nodes: job.nodes ?? null,
+      num_nodes: job.num_nodes ?? null,
+      num_tasks: job.num_tasks ?? null,
+      cpus_per_task: job.cpus_per_task ?? null,
+      mem: job.mem ?? null,
+      stdout_path: job.stdout_path ?? null,
+      stderr_path: job.stderr_path ?? null,
+      stdout_size: job.stdout_size ?? null,
+      stderr_size: job.stderr_size ?? null,
+      command: job.command ?? null,
+      script_path: job.script_path ?? null,
+      session_id: job.session_id ?? null,
+      workdir: job.workdir ?? null,
+      exit_code: job.exit_code ?? null,
+      account: job.account ?? null,
+      qos: job.qos ?? null,
+      time_limit: job.time_limit ?? null,
+      notes: job.notes ?? null,
+      discovered_at: now,
+      updated_at: now
+    };
+    this.jsonJobs.set(job.job_id, inserted);
+    this.flushJsonStore();
+  }
 };
 
 // src/lib/config.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "labgate",
-  "version": "0.5.21",
+  "version": "0.5.23",
   "description": "Policy-controlled sandboxes for AI coding agents — https://labgate.dev",
   "homepage": "https://labgate.dev",
   "keywords": [