@wrongstack/tools 0.282.2 → 0.283.1

package/dist/builtin.js

@@ -1,6 +1,6 @@
 import { spawn, execFileSync } from 'node:child_process';
 import * as Core from '@wrongstack/core';
-import { buildChildEnv, getDesignKitLoader, isDesignStack, loadActiveKit, applyTokenOverrides, setActiveKit, recordKitChoice, recordOverrides, setDesignOverrides, materializeTokens, runDesignVerify, ToolValidationError, detectNewlineStyle, normalizeToLf, toStyle, atomicWrite, unifiedDiff, ToolError, FetchError, isPrivateIPv4, isPrivateIPv6, assessCommitSafety, compileGlob, expectDefined, recordPackageAction, detectPackageEcosystem, deepMerge, addLinkToTask, addNoteToTask, updateCheckOnTask, addCheckToTask, updateGoalMetricOnTask, addGoalMetricToTask, addDependency, updateTaskAssignment, assignTask, releaseTaskClaim, claimReadyTask, getTaskChain, setTaskChain, removeTask, moveTask, updateTask, getTask, transferTaskToBoard, copyTaskToBoard, mergeTasks, splitTask, addTask, removeColumn, updateColumn, addColumn, getKanbanOrchestrationSnapshot, listReadyTasks, searchKanban, deserializeTaskGraph, syncBoardFromTaskGraph, exportBoardToTaskGraph, serializeTaskGraph, exportBoardAsMarkdown, generateBoardFromDescription, createBoard, parseLinesIntoTasks, getBoard, removeBoard, duplicateBoard, updateBoard, listBoards, mutatePlan, clearPlan, getPlanTemplate, addPlanItem, deriveTodosFromPlanItem, removePlanItem, setPlanItemStatus, mutateTasks, formatTaskList, formatPlan, FsError, toErrorMessage as toErrorMessage$2, computeTaskItemProgress, loadPlan, savePlan, loadTasks, saveTasks, wstackGlobalRoot, resolveWstackPaths, truncate } from '@wrongstack/core';
+import { buildChildEnv, getDesignKitLoader, isDesignStack, loadActiveKit, applyTokenOverrides, setActiveKit, recordKitChoice, recordOverrides, setDesignOverrides, materializeTokens, runDesignVerify, ToolValidationError, detectNewlineStyle, normalizeToLf, toStyle, atomicWrite, unifiedDiff, ToolError, FetchError, isPrivateIPv4, isPrivateIPv6, assessCommitSafety, compileGlob, expectDefined, recordPackageAction, detectPackageEcosystem, deepMerge, addLinkToTask, addNoteToTask, updateCheckOnTask, addCheckToTask, updateGoalMetricOnTask, addGoalMetricToTask, addDependency, getKanbanQueueHealth, listKanbanEvents, recoverStaleTaskAssignments, heartbeatTaskAssignment, updateTaskAssignment, assignTask, releaseTaskClaim, claimReadyTask, getTaskChain, setTaskChain, removeTask, moveTask, updateTask, getTask, transferTaskToBoard, copyTaskToBoard, mergeTasks, splitTask, addTask, removeColumn, updateColumn, addColumn, getKanbanOrchestrationSnapshot, listReadyTasks, searchKanban, deserializeTaskGraph, syncBoardFromTaskGraph, exportBoardToTaskGraph, serializeTaskGraph, exportBoardAsMarkdown, generateBoardFromDescription, createBoard, parseLinesIntoTasks, getBoard, removeBoard, duplicateBoard, updateBoard, listBoards, mutatePlan, clearPlan, getPlanTemplate, addPlanItem, deriveTodosFromPlanItem, removePlanItem, setPlanItemStatus, mutateTasks, formatTaskList, formatPlan, FsError, toErrorMessage as toErrorMessage$2, computeTaskItemProgress, loadPlan, savePlan, loadTasks, saveTasks, wstackGlobalRoot, resolveWstackPaths, truncate } from '@wrongstack/core';
 import * as fs from 'node:fs';
 import { statSync, mkdirSync, createWriteStream } from 'node:fs';
 import * as fs2 from 'node:fs/promises';
@@ -2675,6 +2675,24 @@ var IndexStore = class {
    * When false, ranked search falls back to the LIKE + in-process BM25 path.
    */
   ftsAvailable = false;
+  /**
+   * Cache of prepared statements keyed by their SQL text. `DatabaseSync`
+   * compiles SQL on every `.prepare()` call; for the fixed-SQL methods
+   * (upsertFile, getFileMeta, deleteFile, insertRefs, …) that runs thousands
+   * of times during a full reindex. `StatementSync` objects are reusable
+   * across calls on the same connection, so we compile each distinct SQL once
+   * and reuse it. Cleared in {@link close} when the connection is torn down.
+   */
+  stmtCache = /* @__PURE__ */ new Map();
+  /** Prepare-once helper: compile `sql` on first use, reuse thereafter. */
+  stmt(sql) {
+    let s = this.stmtCache.get(sql);
+    if (s === void 0) {
+      s = this.db.prepare(sql);
+      this.stmtCache.set(sql, s);
+    }
+    return s;
+  }
   /**
    * Execute a SQLite write operation with automatic retry on lock conflicts.
    *
@@ -2844,9 +2862,9 @@ var IndexStore = class {
   deleteSymbolsForFile(file) {
     this.runWithRetry(() => {
       if (this.ftsAvailable) {
-        this.db.prepare("DELETE FROM symbols_fts WHERE rowid IN (SELECT id FROM symbols WHERE file_fk = ?)").run(file);
+        this.stmt("DELETE FROM symbols_fts WHERE rowid IN (SELECT id FROM symbols WHERE file_fk = ?)").run(file);
       }
-      this.db.prepare("DELETE FROM symbols WHERE file_fk = ?").run(file);
+      this.stmt("DELETE FROM symbols WHERE file_fk = ?").run(file);
     });
   }
   /**
@@ -2858,13 +2876,13 @@ var IndexStore = class {
     this.runWithRetry(() => {
       this.deleteRefsForFile(file);
       this.deleteSymbolsForFile(file);
-      this.db.prepare("DELETE FROM files WHERE file = ?").run(file);
+      this.stmt("DELETE FROM files WHERE file = ?").run(file);
     });
   }
   // ─── File metadata ──────────────────────────────────────────────────────────
   upsertFile(meta) {
     this.runWithRetry(() => {
-      this.db.prepare(
+      this.stmt(
         `INSERT INTO files(file, lang, mtime_ms, symbol_count, last_indexed)
          VALUES (?, ?, ?, ?, ?)
          ON CONFLICT(file) DO UPDATE SET
@@ -2876,7 +2894,7 @@ var IndexStore = class {
     });
   }
   getFileMeta(file) {
-    const rows = this.db.prepare(
+    const rows = this.stmt(
       "SELECT file, lang, mtime_ms, symbol_count, last_indexed FROM files WHERE file = ?"
     ).all(file);
     if (!rows.length) return null;
@@ -3099,7 +3117,7 @@ var IndexStore = class {
    */
   insertRefs(fromId, refs) {
     this.runWithRetry(() => {
-      this.db.prepare("DELETE FROM refs WHERE from_id = ?").run(fromId);
+      this.stmt("DELETE FROM refs WHERE from_id = ?").run(fromId);
       if (refs.length === 0) return;
       const stmt = this.db.prepare(
         `INSERT INTO refs(from_id, to_name, to_id, call_type, line)
@@ -3300,6 +3318,7 @@ var IndexStore = class {
     }
   }
   close() {
+    this.stmtCache.clear();
     try {
       this.db.close();
     } catch {
@@ -9330,6 +9349,10 @@ var kanbanTool = {
           "release_task",
           "assign_task",
           "mark_assignment",
+          "heartbeat_assignment",
+          "recover_stale",
+          "events",
+          "queue_health",
           "add_dependency",
           "add_goal_metric",
           "update_goal_metric",
@@ -9376,6 +9399,12 @@ var kanbanTool = {
       fallbackModels: { type: "array", items: { type: "string" } },
       tools: { type: "array", items: { type: "string" } },
       allowedCapabilities: { type: "array", items: { type: "string" } },
+      leaseId: { type: "string" },
+      claimedAt: { type: "string" },
+      heartbeatAt: { type: "string" },
+      leaseExpiresAt: { type: "string" },
+      attempt: { type: "number" },
+      maxAttempts: { type: "number" },
       subagentId: { type: "string" },
       runTaskId: { type: "string" },
       lastResult: { type: "string" },
@@ -9387,6 +9416,8 @@ var kanbanTool = {
       releaseStatus: { type: "string", enum: ["pending", "ready", "blocked"] },
       releaseReason: { type: "string" },
       clearAssignee: { type: "boolean" },
+      recoveryMode: { type: "string", enum: ["release", "retry", "fail"] },
+      recoveryNow: { type: "string" },
       taskGraph: { type: "object" },
       graphId: { type: "string" },
       specId: { type: "string" },
@@ -9802,10 +9833,80 @@ var kanbanTool = {
             ...input.runTaskId !== void 0 ? { runTaskId: input.runTaskId } : {},
             ...input.lastResult !== void 0 ? { lastResult: input.lastResult } : {},
             ...input.error !== void 0 ? { error: input.error } : {},
-            ...input.agentId !== void 0 ? { agentId: input.agentId } : {}
+            ...input.agentId !== void 0 ? { agentId: input.agentId } : {},
+            ...input.leaseId !== void 0 ? { leaseId: input.leaseId } : {},
+            ...input.claimedAt !== void 0 ? { claimedAt: input.claimedAt } : {},
+            ...input.heartbeatAt !== void 0 ? { heartbeatAt: input.heartbeatAt } : {},
+            ...input.leaseExpiresAt !== void 0 ? { leaseExpiresAt: input.leaseExpiresAt } : {},
+            ...input.attempt !== void 0 ? { attempt: input.attempt } : {},
+            ...input.maxAttempts !== void 0 ? { maxAttempts: input.maxAttempts } : {}
           });
           return board ? okBoard(board, "Assignment updated.") : fail("Task not found.");
         }
+        case "heartbeat_assignment": {
+          if (!input.boardId || !input.taskId) {
+            return fail("heartbeat_assignment requires boardId and taskId.");
+          }
+          const board = await heartbeatTaskAssignment(projectRoot, input.boardId, input.taskId, {
+            ...input.heartbeatAt !== void 0 ? { heartbeatAt: input.heartbeatAt } : {},
+            ...input.leaseExpiresAt !== void 0 ? { leaseExpiresAt: input.leaseExpiresAt } : {}
+          });
+          return board ? okBoard(board, "Assignment heartbeat updated.") : fail("Task assignment not found.");
+        }
+        case "recover_stale": {
+          if (!input.boardId) return fail("recover_stale requires boardId.");
+          const policyFields = [
+            input.recoveryPolicyFailOnCostCeiling !== void 0,
+            input.recoveryPolicyReleaseOnFailureKinds !== void 0,
+            input.recoveryPolicyReleaseOnHeartbeatDue !== void 0,
+            input.recoveryPolicyRetryPolicyOverride !== void 0
+          ].some(Boolean);
+          const result = await recoverStaleTaskAssignments(projectRoot, input.boardId, {
+            ...input.recoveryMode !== void 0 ? { mode: input.recoveryMode } : {},
+            ...input.recoveryNow !== void 0 ? { now: input.recoveryNow } : {},
+            ...input.releaseReason !== void 0 ? { reason: input.releaseReason } : {},
+            ...input.clearAssignee !== void 0 ? { clearAssignee: input.clearAssignee } : {},
+            ...policyFields ? {
+              policy: {
+                ...input.recoveryPolicyFailOnCostCeiling !== void 0 ? { failWhenCostCeilingSet: input.recoveryPolicyFailOnCostCeiling } : {},
+                ...input.recoveryPolicyReleaseOnFailureKinds !== void 0 ? {
+                  releaseOnFailureKinds: input.recoveryPolicyReleaseOnFailureKinds
+                } : {},
+                ...input.recoveryPolicyReleaseOnHeartbeatDue !== void 0 ? {
+                  releaseOnHeartbeatDue: input.recoveryPolicyReleaseOnHeartbeatDue
+                } : {},
+                ...input.recoveryPolicyRetryPolicyOverride !== void 0 ? {
+                  retryPolicyOverride: input.recoveryPolicyRetryPolicyOverride
+                } : {}
+              }
+            } : {}
+          });
+          return result ? {
+            ok: true,
+            message: `Recovered ${result.tasks.length} stale assignment(s).`,
+            board: result.board,
+            recoveredTasks: result.tasks
+          } : { ok: true, message: "No stale assignment matched.", recoveredTasks: [] };
+        }
+        case "events": {
+          if (!input.boardId) return fail("events requires boardId.");
+          const eventList = await listKanbanEvents(projectRoot, input.boardId);
+          return {
+            ok: true,
+            message: `${eventList.length} event(s).`,
+            events: eventList
+          };
+        }
+        case "queue_health": {
+          const health = await getKanbanQueueHealth(projectRoot, {
+            ...input.boardId !== void 0 ? { boardId: input.boardId } : {}
+          });
+          return {
+            ok: true,
+            message: `Counts: ready=${health.counts.ready}, running=${health.counts.running}, stale=${health.staleAssignments.count}.`,
+            queueHealth: health
+          };
+        }
         case "add_dependency": {
           if (!input.boardId || !input.taskId || !input.dependencyTaskId) {
             return fail("add_dependency requires boardId, taskId, and dependencyTaskId.");
@@ -9957,11 +10058,20 @@ function assignmentInput(input) {
     fallbackModels: input.fallbackModels,
     tools: input.tools,
     allowedCapabilities: input.allowedCapabilities,
-    assignee: input.assignee
+    assignee: input.assignee,
+    leaseId: input.leaseId,
+    claimedAt: input.claimedAt,
+    heartbeatAt: input.heartbeatAt,
+    leaseExpiresAt: input.leaseExpiresAt,
+    attempt: input.attempt,
+    maxAttempts: input.maxAttempts,
+    costCeilingUsd: input.costCeilingUsd,
+    retryPolicy: input.retryPolicy,
+    lastFailureKind: input.lastFailureKind
   };
 }
 function hasAssignmentInput(input) {
-  return input.agentId !== void 0 || input.name !== void 0 || input.role !== void 0 || input.provider !== void 0 || input.model !== void 0 || input.fallbackProfile !== void 0 || input.fallbackModels !== void 0 || input.tools !== void 0 || input.allowedCapabilities !== void 0 || input.assignee !== void 0 || input.assignmentStatus !== void 0;
+  return input.agentId !== void 0 || input.name !== void 0 || input.role !== void 0 || input.provider !== void 0 || input.model !== void 0 || input.fallbackProfile !== void 0 || input.fallbackModels !== void 0 || input.tools !== void 0 || input.allowedCapabilities !== void 0 || input.assignee !== void 0 || input.leaseId !== void 0 || input.claimedAt !== void 0 || input.heartbeatAt !== void 0 || input.leaseExpiresAt !== void 0 || input.attempt !== void 0 || input.maxAttempts !== void 0 || input.costCeilingUsd !== void 0 || input.retryPolicy !== void 0 || input.lastFailureKind !== void 0 || input.assignmentStatus !== void 0;
 }
 function assignmentForTaskCreate(input) {
   return {
@@ -9974,7 +10084,16 @@ function assignmentForTaskCreate(input) {
     ...input.fallbackProfile !== void 0 ? { fallbackProfile: input.fallbackProfile } : {},
     ...input.fallbackModels !== void 0 ? { fallbackModels: input.fallbackModels } : {},
     ...input.tools !== void 0 ? { tools: input.tools } : {},
-    ...input.allowedCapabilities !== void 0 ? { allowedCapabilities: input.allowedCapabilities } : {}
+    ...input.allowedCapabilities !== void 0 ? { allowedCapabilities: input.allowedCapabilities } : {},
+    ...input.leaseId !== void 0 ? { leaseId: input.leaseId } : {},
+    ...input.claimedAt !== void 0 ? { claimedAt: input.claimedAt } : {},
+    ...input.heartbeatAt !== void 0 ? { heartbeatAt: input.heartbeatAt } : {},
+    ...input.leaseExpiresAt !== void 0 ? { leaseExpiresAt: input.leaseExpiresAt } : {},
+    ...input.attempt !== void 0 ? { attempt: input.attempt } : {},
+    ...input.maxAttempts !== void 0 ? { maxAttempts: input.maxAttempts } : {},
+    ...input.costCeilingUsd !== void 0 ? { costCeilingUsd: input.costCeilingUsd } : {},
+    ...input.retryPolicy !== void 0 ? { retryPolicy: input.retryPolicy } : {},
+    ...input.lastFailureKind !== void 0 ? { lastFailureKind: input.lastFailureKind } : {}
   };
 }