@wrongstack/webui 0.281.3 → 0.282.0

package/dist/server/index.js

@@ -5724,7 +5724,16 @@ function setupEvents(deps2) {
     });
   });
   events.on("provider.response", (e) => {
-    broadcast2(clients, { type: "provider.response", payload: sessionPayload2({ sessionId: e.sessionId, usage: e.usage, stopReason: e.stopReason, messageId: "current" }) });
+    broadcast2(clients, {
+      type: "provider.response",
+      payload: sessionPayload2({
+        sessionId: e.sessionId,
+        content: e.content,
+        usage: e.usage,
+        stopReason: e.stopReason,
+        messageId: "current"
+      })
+    });
   });
   events.on("ctx.pct", (e) => {
     broadcast2(clients, {
@@ -8706,12 +8715,13 @@ function createConnectionHandler(opts) {
     void opts.sessionStartPayload().then(async (payload) => {
       const enriched = { ...payload };
       try {
-        const live = opts.context.messages ?? [];
+        const replay = await opts.loadReplay?.();
+        const live = replay?.messages ?? opts.context.messages ?? [];
         const slice = live.length > REPLAY_MESSAGE_CAP ? live.slice(-REPLAY_MESSAGE_CAP) : live;
         if (slice.length > 0) {
           enriched.replayMessages = slice;
         }
-        const total = opts.tokenCounter.total();
+        const total = replay?.usage ?? opts.tokenCounter.total();
         if (total.input + total.output + (total.cacheRead ?? 0) + (total.cacheWrite ?? 0) > 0) {
           enriched.replayUsage = {
             input: total.input,
@@ -8838,8 +8848,8 @@ async function handleGoalGet(projectRoot, broadcast2) {
 }
 
 // src/server/handlers/worklist-handlers.ts
-function sendResult2(ws, ctx, ok, message) {
-  ctx.send(ws, { type: ok ? "ok" : "error", message });
+function sendResult2(ws, ctx, ok2, message) {
+  ctx.send(ws, { type: ok2 ? "ok" : "error", message });
 }
 function sessionPayload(ctx, payload) {
   const provided = payload["sessionId"];
@@ -9060,6 +9070,295 @@ async function handleWorklistMessage(ctx, ws, msg) {
   }
 }
 
+// src/server/kanban-routes.ts
+import {
+  addColumn,
+  addTask,
+  assignTask,
+  copyTaskToBoard,
+  createBoard,
+  duplicateBoard,
+  generateBoardFromDescription,
+  getBoard,
+  getTask,
+  listBoards,
+  moveTask,
+  parseLinesIntoTasks,
+  removeBoard,
+  removeColumn,
+  removeTask,
+  transferTaskToBoard,
+  updateBoard,
+  updateTask
+} from "@wrongstack/core";
+function ok(ws, type, data) {
+  send(ws, { type, payload: { success: true, data: data ?? null } });
+}
+function fail(ws, type, message) {
+  send(ws, { type, payload: { success: false, error: message } });
+}
+async function handleKanbanRoute(ws, msg, ctx) {
+  if (!msg.type.startsWith("kanban.")) return false;
+  const payload = msg.payload;
+  const type = msg.type;
+  try {
+    switch (type) {
+      case "kanban.list":
+        ok(ws, type, await listBoards(ctx.projectRoot));
+        return true;
+      case "kanban.get": {
+        const boardId = payload?.boardId;
+        if (!boardId) {
+          fail(ws, type, "boardId required");
+          return true;
+        }
+        const board = await getBoard(ctx.projectRoot, boardId);
+        board ? ok(ws, type, board) : fail(ws, type, `Board not found: ${boardId}`);
+        return true;
+      }
+      case "kanban.create": {
+        const title = payload?.title;
+        if (!title) {
+          fail(ws, type, "title required");
+          return true;
+        }
+        ok(
+          ws,
+          type,
+          await createBoard(ctx.projectRoot, {
+            title,
+            ...payload?.description ? { description: payload.description } : {},
+            ...payload?.tags ? { tags: payload.tags } : {},
+            ...payload?.columns ? { columns: payload.columns } : {}
+          })
+        );
+        return true;
+      }
+      case "kanban.update": {
+        const boardId = payload?.boardId;
+        if (!boardId) {
+          fail(ws, type, "boardId required");
+          return true;
+        }
+        const board = await updateBoard(ctx.projectRoot, boardId, {
+          ...payload?.title ? { title: payload.title } : {},
+          ...payload?.description ? { description: payload.description } : {},
+          ...payload?.tags ? { tags: payload.tags } : {},
+          ...payload?.columns ? { columns: payload.columns } : {}
+        });
+        board ? ok(ws, type, board) : fail(ws, type, `Board not found: ${boardId}`);
+        return true;
+      }
+      case "kanban.duplicate": {
+        const boardId = payload?.boardId;
+        if (!boardId) {
+          fail(ws, type, "boardId required");
+          return true;
+        }
+        const board = await duplicateBoard(ctx.projectRoot, boardId, {
+          ...payload?.title ? { title: payload.title } : {},
+          ...typeof payload?.includeTasks === "boolean" ? { includeTasks: payload.includeTasks } : {},
+          ...typeof payload?.includeCompletedTasks === "boolean" ? { includeCompletedTasks: payload.includeCompletedTasks } : {},
+          ...typeof payload?.preserveAssignment === "boolean" ? { preserveAssignment: payload.preserveAssignment } : {}
+        });
+        board ? ok(ws, type, board) : fail(ws, type, `Board not found: ${boardId}`);
+        return true;
+      }
+      case "kanban.delete": {
+        const boardId = payload?.boardId;
+        if (!boardId) {
+          fail(ws, type, "boardId required");
+          return true;
+        }
+        ok(ws, type, { removed: await removeBoard(ctx.projectRoot, boardId) });
+        return true;
+      }
+      case "kanban.generate": {
+        const description = payload?.description;
+        if (!description) {
+          fail(ws, type, "description required");
+          return true;
+        }
+        const board = await createBoard(
+          ctx.projectRoot,
+          generateBoardFromDescription({
+            description,
+            ...payload?.title ? { title: payload.title } : {},
+            ...payload?.context ? { context: payload.context } : {}
+          })
+        );
+        for (const taskInput of parseLinesIntoTasks(
+          description,
+          board.columns[0]?.id ?? "backlog"
+        )) {
+          await addTask(ctx.projectRoot, board.id, taskInput);
+        }
+        ok(ws, type, await getBoard(ctx.projectRoot, board.id) ?? board);
+        return true;
+      }
+      case "kanban.task.add": {
+        const boardId = payload?.boardId;
+        const title = payload?.title;
+        if (!boardId || !title) {
+          fail(ws, type, "boardId and title required");
+          return true;
+        }
+        const result = await addTask(ctx.projectRoot, boardId, {
+          title,
+          columnId: payload?.columnId ?? "backlog",
+          ...payload?.description ? { description: payload.description } : {},
+          ...payload?.priority ? { priority: payload.priority } : {},
+          ...payload?.assignedAgent ? { assignedAgent: payload.assignedAgent } : {},
+          ...payload?.labels ? { labels: payload.labels } : {}
+        });
+        result ? ok(ws, type, result.task) : fail(ws, type, `Board not found: ${boardId}`);
+        return true;
+      }
+      case "kanban.task.update": {
+        const boardId = payload?.boardId;
+        const taskId = payload?.taskId;
+        if (!boardId || !taskId) {
+          fail(ws, type, "boardId and taskId required");
+          return true;
+        }
+        const board = await updateTask(ctx.projectRoot, boardId, taskId, {
+          ...payload?.title ? { title: payload.title } : {},
+          ...payload?.description ? { description: payload.description } : {},
+          ...payload?.columnId ? { columnId: payload.columnId } : {},
+          ...payload?.priority ? { priority: payload.priority } : {},
+          ...payload?.status ? { status: payload.status } : {},
+          ...payload?.assignedAgent ? { assignedAgent: payload.assignedAgent } : {},
+          ...payload?.labels ? { labels: payload.labels } : {}
+        });
+        board ? ok(ws, type, findTask(board.tasks, taskId)) : fail(ws, type, "Board or task not found");
+        return true;
+      }
+      case "kanban.task.move": {
+        const boardId = payload?.boardId;
+        const taskId = payload?.taskId;
+        const columnId = payload?.columnId;
+        if (!boardId || !taskId || !columnId) {
+          fail(ws, type, "boardId, taskId, columnId required");
+          return true;
+        }
+        const board = await moveTask(
+          ctx.projectRoot,
+          boardId,
+          taskId,
+          columnId,
+          payload?.order
+        );
+        board ? ok(ws, type, findTask(board.tasks, taskId)) : fail(ws, type, "Move failed");
+        return true;
+      }
+      case "kanban.task.copy":
+      case "kanban.task.transfer": {
+        const boardId = payload?.boardId;
+        const taskId = payload?.taskId;
+        const targetBoardId = payload?.targetBoardId;
+        if (!boardId || !taskId || !targetBoardId) {
+          fail(ws, type, "boardId, taskId, targetBoardId required");
+          return true;
+        }
+        const result = type === "kanban.task.copy" ? await copyTaskToBoard(ctx.projectRoot, boardId, taskId, targetBoardId, {
+          ...payload?.targetColumnId ? { targetColumnId: payload.targetColumnId } : {},
+          ...typeof payload?.preserveAssignment === "boolean" ? { preserveAssignment: payload.preserveAssignment } : {},
+          ...typeof payload?.preserveDependencies === "boolean" ? { preserveDependencies: payload.preserveDependencies } : {}
+        }) : await transferTaskToBoard(ctx.projectRoot, boardId, taskId, targetBoardId, {
+          ...payload?.targetColumnId ? { targetColumnId: payload.targetColumnId } : {},
+          ...typeof payload?.preserveAssignment === "boolean" ? { preserveAssignment: payload.preserveAssignment } : {},
+          ...typeof payload?.preserveDependencies === "boolean" ? { preserveDependencies: payload.preserveDependencies } : {}
+        });
+        result ? ok(ws, type, result.targetBoard) : fail(ws, type, "Board or task not found");
+        return true;
+      }
+      case "kanban.task.assign": {
+        const boardId = payload?.boardId;
+        const taskId = payload?.taskId;
+        if (!boardId || !taskId) {
+          fail(ws, type, "boardId and taskId required");
+          return true;
+        }
+        const board = await assignTask(ctx.projectRoot, boardId, taskId, {
+          ...payload?.agentId ? { agentId: payload.agentId } : {},
+          ...payload?.name ? { name: payload.name } : {},
+          ...payload?.role ? { role: payload.role } : {},
+          ...payload?.provider ? { provider: payload.provider } : {},
+          ...payload?.model ? { model: payload.model } : {},
+          ...payload?.fallbackProfile ? { fallbackProfile: payload.fallbackProfile } : {},
+          ...payload?.fallbackModels ? { fallbackModels: payload.fallbackModels } : {},
+          ...payload?.tools ? { tools: payload.tools } : {},
+          ...payload?.allowedCapabilities ? { allowedCapabilities: payload.allowedCapabilities } : {}
+        });
+        board ? ok(ws, type, findTask(board.tasks, taskId)) : fail(ws, type, "Board or task not found");
+        return true;
+      }
+      case "kanban.task.dispatch":
+        fail(
+          ws,
+          type,
+          "Kanban agent dispatch is only available from the CLI-hosted WebUI runtime."
+        );
+        return true;
+      case "kanban.task.remove": {
+        const boardId = payload?.boardId;
+        const taskId = payload?.taskId;
+        if (!boardId || !taskId) {
+          fail(ws, type, "boardId and taskId required");
+          return true;
+        }
+        const board = await removeTask(ctx.projectRoot, boardId, taskId);
+        board ? ok(ws, type, { removed: true }) : fail(ws, type, "Board or task not found");
+        return true;
+      }
+      case "kanban.task.get": {
+        const boardId = payload?.boardId;
+        const taskId = payload?.taskId;
+        if (!boardId || !taskId) {
+          fail(ws, type, "boardId and taskId required");
+          return true;
+        }
+        const task = await getTask(ctx.projectRoot, boardId, taskId);
+        task ? ok(ws, type, task) : fail(ws, type, "Task not found");
+        return true;
+      }
+      case "kanban.column.add": {
+        const boardId = payload?.boardId;
+        const title = payload?.title;
+        if (!boardId || !title) {
+          fail(ws, type, "boardId and title required");
+          return true;
+        }
+        const result = await addColumn(ctx.projectRoot, boardId, { title });
+        result ? ok(ws, type, result.board.columns) : fail(ws, type, `Board not found: ${boardId}`);
+        return true;
+      }
+      case "kanban.column.remove": {
+        const boardId = payload?.boardId;
+        const columnId = payload?.columnId;
+        if (!boardId || !columnId) {
+          fail(ws, type, "boardId and columnId required");
+          return true;
+        }
+        const board = await removeColumn(ctx.projectRoot, boardId, columnId, {
+          moveTasksToColumnId: payload?.moveTasksToColumnId
+        });
+        board ? ok(ws, type, { removed: true }) : fail(ws, type, `Column not found: ${columnId}`);
+        return true;
+      }
+      default:
+        fail(ws, type, `Unknown kanban message type: ${type}`);
+        return true;
+    }
+  } catch (err) {
+    fail(ws, type, err instanceof Error ? err.message : String(err));
+    return true;
+  }
+}
+function findTask(tasks, taskId) {
+  return tasks.find((task) => task.id === taskId || task.id.startsWith(taskId));
+}
+
 // src/server/mailbox-routes.ts
 async function handleMailboxRoute(ws, msg, handlers) {
   switch (msg.type) {
@@ -9544,6 +9843,7 @@ function createMessageDispatcher(opts) {
     if (await handleSpecsRoute(ws, msg, routes.specsRoutes)) return;
     if (await handleSddBoardRoute(ws, msg, routes.sddBoardRoutes)) return;
     if (await handleSddWizardRoute(ws, msg, routes.sddWizardRoutes)) return;
+    if (await handleKanbanRoute(ws, msg, { projectRoot: state.getProjectRoot() })) return;
     if (msg.type.startsWith("worktree.") && await deps2.worktreeHandler.handleMessage(
       msg
     ))
@@ -9558,7 +9858,10 @@ function createMessageDispatcher(opts) {
       case "collab.resume":
       case "collab.grant_control":
       case "collab.inject_tool": {
-        deps2.collabHandler.handleMessage(ws, msg);
+        deps2.collabHandler.handleMessage(
+          ws,
+          msg
+        );
         return;
       }
       // Integrated terminal — interactive pty transport, bypasses the agent loop.
@@ -9586,7 +9889,10 @@ function createMessageDispatcher(opts) {
         runLock.set(thisRun);
         try {
           const maxIt = typeof deps2.context.meta["maxIterations"] === "number" ? deps2.context.meta["maxIterations"] : void 0;
-          const result = await deps2.agent.run(content, { signal: thisRun.signal, maxIterations: maxIt });
+          const result = await deps2.agent.run(content, {
+            signal: thisRun.signal,
+            maxIterations: maxIt
+          });
           send(ws, {
             type: "run.result",
             payload: sessionPayload2({
@@ -9726,22 +10032,43 @@ function createMessageDispatcher(opts) {
         break;
       // Design Studio — shared handlers (design-handlers.ts).
       case "design.list":
-        await handleDesignList(ws, { projectRoot: state.getProjectRoot(), agentMeta: deps2.context });
+        await handleDesignList(ws, {
+          projectRoot: state.getProjectRoot(),
+          agentMeta: deps2.context
+        });
         break;
       case "design.use":
-        await handleDesignUse(ws, { projectRoot: state.getProjectRoot(), agentMeta: deps2.context }, msg);
+        await handleDesignUse(
+          ws,
+          { projectRoot: state.getProjectRoot(), agentMeta: deps2.context },
+          msg
+        );
         break;
       case "design.state":
-        await handleDesignState(ws, { projectRoot: state.getProjectRoot(), agentMeta: deps2.context });
+        await handleDesignState(ws, {
+          projectRoot: state.getProjectRoot(),
+          agentMeta: deps2.context
+        });
         break;
       case "design.set":
-        await handleDesignSet(ws, { projectRoot: state.getProjectRoot(), agentMeta: deps2.context }, msg);
+        await handleDesignSet(
+          ws,
+          { projectRoot: state.getProjectRoot(), agentMeta: deps2.context },
+          msg
+        );
         break;
       case "design.materialize":
-        await handleDesignMaterialize(ws, { projectRoot: state.getProjectRoot(), agentMeta: deps2.context }, msg);
+        await handleDesignMaterialize(
+          ws,
+          { projectRoot: state.getProjectRoot(), agentMeta: deps2.context },
+          msg
+        );
         break;
       case "design.verify":
-        await handleDesignVerify(ws, { projectRoot: state.getProjectRoot(), agentMeta: deps2.context });
+        await handleDesignVerify(ws, {
+          projectRoot: state.getProjectRoot(),
+          agentMeta: deps2.context
+        });
         break;
       case "diag.get": {
         if (!ensureCurrentSession(ws, msg, "diag.get")) return;
@@ -11934,6 +12261,11 @@ async function startWebUI(opts = {}) {
     sessionStartPayload,
     tokenCounter,
     context,
+    loadReplay: async () => {
+      await session.flush().catch(() => void 0);
+      const data = await sessionStore.load(session.id);
+      return { messages: data.messages, usage: data.usage };
+    },
     clients,
     pendingConfirms,
     autoPhaseHandler,