@csdwd/ai-teams-server 0.3.1 → 0.3.2

package/dist/index.js

@@ -13,6 +13,7 @@ import websocket from "@fastify/websocket";
 import fastifyStatic from "@fastify/static";
 
 // ../../packages/shared/dist/index.js
+var TERMINAL_STATUSES = /* @__PURE__ */ new Set(["completed", "failed", "cancelled", "timeout"]);
 var ProtocolError = class extends Error {
   constructor(message) {
     super(message);
@@ -578,6 +579,16 @@ async function hydrateState(db, state, defaultTimeoutSec, maxLogChunksPerTask) {
         queue.push(task.id);
         state.taskQueues.set(task.employeeId, queue);
       }
+    } else if (task.employeeId && !TERMINAL_STATUSES.has(task.status)) {
+      task.status = "queued";
+      persistTask(db, task);
+      if (task.targetMode === "queue") {
+        state.sharedTaskQueue.push(task.id);
+      } else {
+        const queue = state.mainTaskQueues.get(task.employeeId) ?? [];
+        queue.push(task.id);
+        state.mainTaskQueues.set(task.employeeId, queue);
+      }
     }
   }
   const logRows = await db.all(
@@ -1222,7 +1233,6 @@ var scheduleListResponseSchema = {
 // src/dispatch.ts
 import { createHmac, randomUUID } from "node:crypto";
 import WebSocket from "ws";
-var TERMINAL_STATUSES = /* @__PURE__ */ new Set(["completed", "failed", "cancelled", "timeout"]);
 function nowIso() {
   return (/* @__PURE__ */ new Date()).toISOString();
 }
@@ -1959,11 +1969,52 @@ function createDispatch(ctx) {
         break;
     }
   }
+  function startDisconnectRecovery(employeeId) {
+    const employee = state.employees.get(employeeId);
+    if (!employee) return;
+    const taskIds = [];
+    if (employee.mainTaskId) taskIds.push(employee.mainTaskId);
+    if (employee.queueTaskId) taskIds.push(employee.queueTaskId);
+    if (taskIds.length === 0) return;
+    const timer = setTimeout(() => {
+      state.disconnectTimers.delete(employeeId);
+      if (state.agentSockets.has(employeeId)) return;
+      for (const taskId of taskIds) {
+        const task = state.tasks.get(taskId);
+        if (!task || TERMINAL_STATUSES.has(task.status)) continue;
+        clearTaskTimeout(taskId);
+        task.status = "queued";
+        task.employeeId = null;
+        upsertTask(task);
+        log.info({ taskId, employeeId }, "Task re-queued after disconnect grace period");
+        if (task.targetMode === "queue") {
+          enqueueSharedTask(taskId);
+        } else {
+          const queue = state.mainTaskQueues.get(employeeId) ?? [];
+          if (!queue.includes(taskId)) queue.push(taskId);
+          state.mainTaskQueues.set(employeeId, queue);
+        }
+      }
+      const emp = state.employees.get(employeeId);
+      if (emp) {
+        if (taskIds.includes(emp.mainTaskId ?? "")) setMainTask(employeeId, null, null);
+        if (taskIds.includes(emp.queueTaskId ?? "")) setQueueTask(employeeId, null, null);
+        broadcastToLeaders({ type: "employee.upsert", employee: emp });
+      }
+      for (const taskId of taskIds) {
+        const task = state.tasks.get(taskId);
+        if (task) broadcastToLeaders({ type: "task.upsert", task });
+      }
+      dispatchSharedQueuedTasks();
+    }, ctx.disconnectGraceMs);
+    state.disconnectTimers.set(employeeId, timer);
+  }
   return {
     dispatchLeaderCommand,
     handleAgentMessage,
     handleLeaderMessage,
-    cancelTaskById
+    cancelTaskById,
+    startDisconnectRecovery
   };
 }
 
@@ -2157,7 +2208,7 @@ async function createAiTeamsServer(options) {
     disconnectGraceMs,
     encryptor: createEncryptor(process.env.AI_TEAMS_ENCRYPTION_KEY)
   };
-  const { dispatchLeaderCommand, handleAgentMessage, handleLeaderMessage, cancelTaskById } = createDispatch(dispatchCtx);
+  const { dispatchLeaderCommand, handleAgentMessage, handleLeaderMessage, cancelTaskById, startDisconnectRecovery } = createDispatch(dispatchCtx);
   const scheduleDispatchFn = (message, webhookUrl, cliConfig, priority, requiredLabels) => {
     return dispatchLeaderCommand(message, webhookUrl, cliConfig, priority, requiredLabels);
   };
@@ -2786,6 +2837,7 @@ async function createAiTeamsServer(options) {
         for (const leaderSocket of state.leaderSockets) {
           sendJson(leaderSocket, { type: "employee.upsert", employee }, dispatchCtx.encryptor);
         }
+        startDisconnectRecovery(employeeId);
       }
       app.log.info({ employeeId }, "Agent disconnected");
     });
@@ -2891,7 +2943,7 @@ if (isCli) {
   getArgValue2 = getArgValue, resolveDataDir2 = resolveDataDir, resolvePidFile2 = resolvePidFile, resolveLogDir2 = resolveLogDir, applyCliArgsToEnv2 = applyCliArgsToEnv;
   const args = process.argv.slice(2);
   if (args.includes("--version") || args.includes("-v")) {
-    console.log("0.3.1");
+    console.log("0.3.2");
     process.exit(0);
   }
   if (args.includes("--help") || args.includes("-h")) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@csdwd/ai-teams-server",
-  "version": "0.3.1",
+  "version": "0.3.2",
   "description": "AI Teams central server — Fastify HTTP + WebSocket server for task dispatch, employee management, and web UI",
   "type": "module",
   "bin": {