@wrongstack/tools 0.282.2 → 0.283.1

package/dist/index.d.ts

@@ -5,7 +5,7 @@ export { batchToolUseTool } from './batch-tool-use.js';
 export { OPTIONAL_TOOLS, TIER1_TOOLS, TIER2_TOOLS, TIER3_TOOLS, builtinTools } from './builtin.js';
 export { CircuitBreaker, CircuitBreakerConfig, CircuitBreakerSnapshot } from './circuit-breaker.js';
 export { C as CircuitOpenError, e as CircuitSnapshot, f as CircuitState, h as IndexCircuitBreaker, i as IndexTimeoutError, k as cancelPendingReindexes, l as codebaseIndexStats, m as codebaseIndexTool, n as codebaseSearchTool, o as codebaseStatsTool, p as enqueueReindex, q as getIndexState, r as indexCircuitBreaker, s as isIndexReady, t as isIndexableFile, u as isIndexing, v as onIndexStateChange, w as resetIndexCircuitBreaker, x as runStartupIndex, y as searchCodebaseIndex, z as shutdownCodebaseIndexHost } from './background-indexer-BeDBxfSh.js';
-import { Tool, AssignKanbanTaskInput, KanbanTaskPriority, KanbanTaskStatus, KanbanAgentRunStatus, KanbanBoard, KanbanBoardSummary, KanbanTask, KanbanSearchResult, KanbanOrchestrationSnapshot, SerializedTaskGraph, SkillLoader } from '@wrongstack/core';
+import { Tool, AssignKanbanTaskInput, KanbanTaskPriority, KanbanTaskStatus, KanbanAgentRunStatus, KanbanBoard, KanbanBoardSummary, KanbanTask, KanbanSearchResult, KanbanEvent, KanbanQueueHealth, KanbanOrchestrationSnapshot, SerializedTaskGraph, SkillLoader } from '@wrongstack/core';
 export { diffTool } from './diff.js';
 export { documentTool } from './document.js';
 export { editTool } from './edit.js';
@@ -184,7 +184,7 @@ interface DesignOutput {
  */
 declare const designTool: Tool<DesignInput, DesignOutput>;
 
-type KanbanAction = 'list_boards' | 'get_board' | 'create_board' | 'duplicate_board' | 'update_board' | 'delete_board' | 'generate_board' | 'export_markdown' | 'export_task_graph' | 'sync_task_graph' | 'search_tasks' | 'ready_tasks' | 'snapshot' | 'add_column' | 'update_column' | 'delete_column' | 'add_task' | 'split_task' | 'merge_tasks' | 'copy_task' | 'transfer_task' | 'get_task' | 'update_task' | 'move_task' | 'delete_task' | 'set_chain' | 'get_chain' | 'claim_task' | 'release_task' | 'assign_task' | 'mark_assignment' | 'add_dependency' | 'add_goal_metric' | 'update_goal_metric' | 'add_check' | 'update_check' | 'add_note' | 'add_link';
+type KanbanAction = 'list_boards' | 'get_board' | 'create_board' | 'duplicate_board' | 'update_board' | 'delete_board' | 'generate_board' | 'export_markdown' | 'export_task_graph' | 'sync_task_graph' | 'search_tasks' | 'ready_tasks' | 'snapshot' | 'add_column' | 'update_column' | 'delete_column' | 'add_task' | 'split_task' | 'merge_tasks' | 'copy_task' | 'transfer_task' | 'get_task' | 'update_task' | 'move_task' | 'delete_task' | 'set_chain' | 'get_chain' | 'claim_task' | 'release_task' | 'assign_task' | 'mark_assignment' | 'heartbeat_assignment' | 'recover_stale' | 'events' | 'queue_health' | 'add_dependency' | 'add_goal_metric' | 'update_goal_metric' | 'add_check' | 'update_check' | 'add_note' | 'add_link';
 interface KanbanToolInput extends Omit<AssignKanbanTaskInput, 'status'> {
     action: KanbanAction;
     boardId?: string | undefined;
@@ -237,6 +237,15 @@ interface KanbanToolInput extends Omit<AssignKanbanTaskInput, 'status'> {
     includeCompletedTasks?: boolean | undefined;
     preserveAssignment?: boolean | undefined;
     preserveDependencies?: boolean | undefined;
+    leaseId?: string | undefined;
+    claimedAt?: string | undefined;
+    heartbeatAt?: string | undefined;
+    leaseExpiresAt?: string | undefined;
+    attempt?: number | undefined;
+    maxAttempts?: number | undefined;
+    costCeilingUsd?: number | undefined;
+    retryPolicy?: 'off' | 'incremental' | 'exponential' | undefined;
+    lastFailureKind?: string | undefined;
     subagentId?: string | undefined;
     runTaskId?: string | undefined;
     lastResult?: string | undefined;
@@ -245,6 +254,12 @@ interface KanbanToolInput extends Omit<AssignKanbanTaskInput, 'status'> {
     releaseStatus?: 'pending' | 'ready' | 'blocked' | undefined;
     releaseReason?: string | undefined;
     clearAssignee?: boolean | undefined;
+    recoveryMode?: 'auto' | 'release' | 'retry' | 'fail' | undefined;
+    recoveryNow?: string | undefined;
+    recoveryPolicyFailOnCostCeiling?: boolean | undefined;
+    recoveryPolicyReleaseOnFailureKinds?: string[] | undefined;
+    recoveryPolicyReleaseOnHeartbeatDue?: boolean | undefined;
+    recoveryPolicyRetryPolicyOverride?: 'off' | 'incremental' | 'exponential' | undefined;
     taskGraph?: unknown;
     graphId?: string | undefined;
     specId?: string | undefined;
@@ -262,6 +277,9 @@ interface KanbanToolOutput {
     boards?: KanbanBoardSummary[] | undefined;
     task?: KanbanTask | undefined;
     tasks?: KanbanSearchResult[] | undefined;
+    recoveredTasks?: KanbanTask[] | undefined;
+    events?: KanbanEvent[] | undefined;
+    queueHealth?: KanbanQueueHealth | undefined;
     children?: KanbanTask[] | undefined;
     chain?: KanbanTask[] | undefined;
     snapshot?: KanbanOrchestrationSnapshot | undefined;

package/dist/index.js

@@ -4,7 +4,7 @@ import * as path3 from 'node:path';
 import { resolve, sep, dirname, join } from 'node:path';
 import { spawn, execFileSync } from 'node:child_process';
 import * as Core from '@wrongstack/core';
-import { SKILL_LIMITS, 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, stripFrontmatter, wstackGlobalRoot, resolveWstackPaths, truncate } from '@wrongstack/core';
+import { SKILL_LIMITS, 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, stripFrontmatter, wstackGlobalRoot, resolveWstackPaths, truncate } from '@wrongstack/core';
 import * as fs2 from 'node:fs/promises';
 import * as os2 from 'node:os';
 import { toErrorMessage as toErrorMessage$3 } from '@wrongstack/core/utils';
@@ -2999,6 +2999,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.
    *
@@ -3168,9 +3186,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);
     });
   }
   /**
@@ -3182,13 +3200,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
@@ -3200,7 +3218,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;
@@ -3423,7 +3441,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)
@@ -3624,6 +3642,7 @@ var IndexStore = class {
     }
   }
   close() {
+    this.stmtCache.clear();
     try {
       this.db.close();
     } catch {
@@ -9454,6 +9473,10 @@ var kanbanTool = {
           "release_task",
           "assign_task",
           "mark_assignment",
+          "heartbeat_assignment",
+          "recover_stale",
+          "events",
+          "queue_health",
           "add_dependency",
           "add_goal_metric",
           "update_goal_metric",
@@ -9500,6 +9523,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" },
@@ -9511,6 +9540,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" },
@@ -9926,10 +9957,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.");
@@ -10081,11 +10182,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 {
@@ -10098,7 +10208,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 } : {}
   };
 }
 
@@ -13490,6 +13609,12 @@ function relatedMemoryTool(memory) {
 }
 
 // src/mode.ts
+function modeFamily(mode) {
+  if (mode.tags?.includes("lite")) return "lite";
+  if (mode.tags?.includes("deep")) return "deep";
+  if (mode.tags?.includes("balanced") || mode.id === "default") return "balanced";
+  return "custom";
+}
 function createModeTool(modeStore) {
   return {
     name: "mode",
@@ -13529,10 +13654,16 @@ function createModeTool(modeStore) {
         }
         case "list": {
           const modes = await modeStore.listModes();
-          const lines = modes.map((m) => `  ${m.id.padEnd(20)} ${m.name} \u2014 ${m.description}`).join("\n");
+          const lines = modes.map((m) => `  ${m.id.padEnd(20)} [${modeFamily(m)}] ${m.name} \u2014 ${m.description}`).join("\n");
           return {
             action: "list",
-            modes: modes.map((m) => ({ id: m.id, name: m.name, description: m.description })),
+            modes: modes.map((m) => ({
+              id: m.id,
+              name: m.name,
+              description: m.description,
+              family: modeFamily(m),
+              tags: m.tags ?? []
+            })),
             success: true,
             message: lines
           };