feat(tasks): aggregated task progress FAB + bottom sheet

- Add TaskAggregator (server/stores) to unify TaskCreate/TaskUpdate/TodoWrite - Broadcast task-state snapshots via new WS event on tool events + reconnect - TaskFab: SVG progress ring with fade-out on completion, reappears on new tasks - TaskBottomSheet: full task list with dependencies, activeForm, expandable description - Remove inline TodoWrite rendering (TaskProgress), filter task tools from chat flow - Rebuild task state from JSONL history on server restart/reconnect Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-03-29 07:11:26 +08:00
parent fa81cb175c
commit a1ada37cba
14 changed files with 588 additions and 88 deletions
@@ -0,0 +1,124 @@
+/**
+ * Pure aggregator: processes tool_use blocks (TaskCreate, TaskUpdate, TodoWrite)
+ * and produces a unified AggregatedTask[] snapshot.
+ *
+ * Stateful — call processToolUse() for each relevant tool call in order.
+ * Call getSnapshot() to read current state (cached, invalidated on changes).
+ */
+
+export interface AggregatedTask {
+  id: string;
+  subject: string;
+  description?: string;
+  activeForm?: string;
+  status: 'pending' | 'in_progress' | 'completed';
+  blockedBy?: string[];
+  blocks?: string[];
+  source: 'task-api' | 'todo-write';
+}
+
+export interface TaskSnapshot {
+  tasks: AggregatedTask[];
+  completed: number;
+  total: number;
+}
+
+/** Tool names that produce task state changes. */
+export const TASK_TOOL_NAMES = new Set(['TaskCreate', 'TaskUpdate', 'TodoWrite']);
+/** Subset handled on tool-start (TaskCreate needs result text, so it's deferred to tool-done). */
+export const TASK_TOOLS_ON_START = new Set(['TaskUpdate', 'TodoWrite']);
+
+export class TaskAggregator {
+  private taskApiTasks = new Map<string, AggregatedTask>();
+  private todoWriteTasks = new Map<string, AggregatedTask>();
+  private cachedSnapshot: TaskSnapshot | null = null;
+
+  /** Process a single tool_use block. Returns true if state changed. */
+  processToolUse(toolName: string, input: Record<string, unknown>, resultText?: string): boolean {
+    switch (toolName) {
+      case 'TaskCreate': {
+        const match = resultText?.match(/Task #(\d+)/);
+        const id = match?.[1] || String(this.taskApiTasks.size + 1);
+        this.taskApiTasks.set(id, {
+          id,
+          subject: (input.subject as string) || '',
+          description: (input.description as string) || undefined,
+          activeForm: (input.activeForm as string) || undefined,
+          status: 'pending',
+          source: 'task-api',
+        });
+        this.cachedSnapshot = null;
+        return true;
+      }
+
+      case 'TaskUpdate': {
+        const taskId = input.taskId as string;
+        if (!taskId) return false;
+        const existing = this.taskApiTasks.get(taskId);
+        if (!existing) return false;
+        if (input.status === 'deleted') {
+          this.taskApiTasks.delete(taskId);
+          this.cachedSnapshot = null;
+          return true;
+        }
+        const updated: AggregatedTask = { ...existing };
+        if (input.status) updated.status = input.status as AggregatedTask['status'];
+        if (input.subject) updated.subject = input.subject as string;
+        if (input.description) updated.description = input.description as string;
+        if (input.activeForm) updated.activeForm = input.activeForm as string;
+        if (input.addBlockedBy) {
+          updated.blockedBy = [...(updated.blockedBy || []), ...(input.addBlockedBy as string[])];
+        }
+        if (input.addBlocks) {
+          updated.blocks = [...(updated.blocks || []), ...(input.addBlocks as string[])];
+        }
+        this.taskApiTasks.set(taskId, updated);
+        this.cachedSnapshot = null;
+        return true;
+      }
+
+      case 'TodoWrite': {
+        this.todoWriteTasks.clear();
+        const tasks = (input.tasks || input.todos || []) as Array<{
+          id: string;
+          content: string;
+          status: string;
+        }>;
+        for (const t of tasks) {
+          this.todoWriteTasks.set(t.id, {
+            id: t.id,
+            subject: t.content || '',
+            status: (t.status as AggregatedTask['status']) || 'pending',
+            source: 'todo-write',
+          });
+        }
+        this.cachedSnapshot = null;
+        return true;
+      }
+
+      default:
+        return false;
+    }
+  }
+
+  getSnapshot(): TaskSnapshot {
+    if (this.cachedSnapshot) return this.cachedSnapshot;
+    const tasks: AggregatedTask[] = [
+      ...this.taskApiTasks.values(),
+      ...this.todoWriteTasks.values(),
+    ];
+    const completed = tasks.filter(t => t.status === 'completed').length;
+    this.cachedSnapshot = { tasks, completed, total: tasks.length };
+    return this.cachedSnapshot;
+  }
+
+  get hasTasks(): boolean {
+    return this.taskApiTasks.size > 0 || this.todoWriteTasks.size > 0;
+  }
+
+  clear(): void {
+    this.taskApiTasks.clear();
+    this.todoWriteTasks.clear();
+    this.cachedSnapshot = null;
+  }
+}