@wrongstack/core 0.87.0 → 0.89.1

package/dist/sdd/index.js

@@ -17,7 +17,8 @@ var __export = (target, all) => {
 var atomic_write_exports = {};
 __export(atomic_write_exports, {
   atomicWrite: () => atomicWrite,
-  ensureDir: () => ensureDir
+  ensureDir: () => ensureDir,
+  withFileLock: () => withFileLock
 });
 async function atomicWrite(targetPath, content, opts = {}) {
   const dir = path.dirname(targetPath);
@@ -60,6 +61,49 @@ async function atomicWrite(targetPath, content, opts = {}) {
 async function ensureDir(dir) {
   await fs.mkdir(dir, { recursive: true });
 }
+async function withFileLock(targetPath, fn, opts = {}) {
+  const dir = path.dirname(targetPath);
+  await fs.mkdir(dir, { recursive: true });
+  const lockPath = path.join(dir, `.${path.basename(targetPath)}.lock`);
+  const timeoutMs = opts.timeoutMs ?? 5e3;
+  const staleMs = opts.staleMs ?? 3e4;
+  const started = Date.now();
+  let handle;
+  for (; ; ) {
+    try {
+      handle = await fs.open(lockPath, "wx");
+      await handle.writeFile(`${process.pid}:${Date.now()}`);
+      break;
+    } catch (err) {
+      if (err.code !== "EEXIST") throw err;
+      try {
+        const stat2 = await fs.stat(lockPath);
+        if (Date.now() - stat2.mtimeMs > staleMs) {
+          await fs.unlink(lockPath);
+          continue;
+        }
+      } catch {
+        continue;
+      }
+      if (Date.now() - started >= timeoutMs) {
+        throw new Error(`Timed out waiting for file lock: ${targetPath}`);
+      }
+      await new Promise((resolve) => setTimeout(resolve, 25));
+    }
+  }
+  try {
+    return await fn();
+  } finally {
+    try {
+      await handle?.close();
+    } catch {
+    }
+    try {
+      await fs.unlink(lockPath);
+    } catch {
+    }
+  }
+}
 async function renameWithRetry(from, to) {
   if (process.platform !== "win32") {
     await fs.rename(from, to);
@@ -2591,6 +2635,7 @@ var SddTaskDecomposer = class {
 };
 
 // src/coordination/subagent-budget.ts
+var TIMEOUT_PREEMPT_FRACTION = 0.85;
 var BudgetExceededError = class extends Error {
   kind;
   limit;
@@ -6110,15 +6155,38 @@ var DefaultMultiAgentCoordinator = class _DefaultMultiAgentCoordinator extends E
     }
     const start = Date.now();
     let timer = null;
+    let preemptedForLimit = null;
     const timeoutPromise = new Promise((_, reject) => {
       const armFor = (ms) => {
         if (timer) clearTimeout(timer);
         timer = setTimeout(onTick, Math.max(0, ms));
       };
       const scheduleNext = () => {
-        const wallRemaining = initialTimeoutMs === void 0 ? Number.POSITIVE_INFINITY : (budget.limits.timeoutMs ?? initialTimeoutMs) - (Date.now() - start);
+        const wallLimit = budget.limits.timeoutMs ?? initialTimeoutMs;
+        const wallRemaining = initialTimeoutMs === void 0 ? Number.POSITIVE_INFINITY : wallLimit - (Date.now() - start);
         const idleRemaining = idleLimitMs === void 0 ? Number.POSITIVE_INFINITY : (budget.limits.idleTimeoutMs ?? idleLimitMs) - budget.idleMs();
-        armFor(Math.max(25, Math.min(wallRemaining, idleRemaining)));
+        const preemptRemaining = initialTimeoutMs === void 0 || preemptedForLimit === wallLimit ? Number.POSITIVE_INFINITY : wallLimit * TIMEOUT_PREEMPT_FRACTION - (Date.now() - start);
+        armFor(Math.max(25, Math.min(wallRemaining, idleRemaining, preemptRemaining)));
+      };
+      const negotiateTimeout = async (used, limit) => {
+        const handler = budget.onThreshold;
+        if (!handler) return "stop";
+        const result = handler({
+          kind: "timeout",
+          used,
+          limit,
+          requestDecision: () => new Promise((resolveDecision) => {
+            budget._events?.emit("budget.threshold_reached", {
+              kind: "timeout",
+              used,
+              limit,
+              timeoutMs: 6e4,
+              extend: (extra) => resolveDecision({ extend: extra }),
+              deny: () => resolveDecision("stop")
+            });
+          })
+        });
+        return typeof result === "string" ? result : await result;
       };
       const onTick = async () => {
         const elapsed = Date.now() - start;
@@ -6131,6 +6199,21 @@ var DefaultMultiAgentCoordinator = class _DefaultMultiAgentCoordinator extends E
           reject(new BudgetExceededError("timeout", idleLimit ?? 0, budget.idleMs()));
           return;
         }
+        if (wallLimit !== void 0 && !wallExceeded && budget.onThreshold && preemptedForLimit !== wallLimit && elapsed >= wallLimit * TIMEOUT_PREEMPT_FRACTION) {
+          try {
+            const decision = await negotiateTimeout(elapsed, wallLimit);
+            if (typeof decision !== "string" && decision.extend.timeoutMs !== void 0) {
+              budget.limits.timeoutMs = decision.extend.timeoutMs;
+              preemptedForLimit = null;
+            } else {
+              preemptedForLimit = wallLimit;
+            }
+          } catch {
+            preemptedForLimit = wallLimit;
+          }
+          scheduleNext();
+          return;
+        }
         if (!wallExceeded) {
           scheduleNext();
           return;
@@ -6142,28 +6225,15 @@ var DefaultMultiAgentCoordinator = class _DefaultMultiAgentCoordinator extends E
           return;
         }
         try {
-          const result = budget.onThreshold({
-            kind: "timeout",
-            used: elapsed,
-            limit,
-            requestDecision: () => new Promise((resolveDecision) => {
-              budget._events?.emit("budget.threshold_reached", {
-                kind: "timeout",
-                used: elapsed,
-                limit,
-                timeoutMs: 6e4,
-                extend: (extra) => resolveDecision({ extend: extra }),
-                deny: () => resolveDecision("stop")
-              });
-            })
-          });
-          const decision = typeof result === "string" ? result : await result;
+          const decision = await negotiateTimeout(elapsed, limit);
           if (decision === "continue" || decision === "throw" || decision === "stop") {
+            preemptedForLimit = null;
             armFor(Math.max(1e3, limit));
             return;
           }
           if (decision.extend.timeoutMs !== void 0) {
             budget.limits.timeoutMs = decision.extend.timeoutMs;
+            preemptedForLimit = null;
             scheduleNext();
             return;
           }