repoburg 1.3.202 → 1.3.209

package/agent-pane/frontend/src/store/wsSlice.ts

@@ -1,368 +0,0 @@
-import type { StateCreator } from 'zustand';
-import type { SessionDetail, Item, TokenUsage, ContextUsage } from '../types/index.ts';
-import type { StoreState } from './store.ts';
-
-export type BackgroundTaskInfo = {
-  task_id: string;
-  task_type: string;
-  description: string;
-};
-
-export type WsSlice = {
-  wsConnected: boolean;
-  setWsConnected: (connected: boolean) => void;
-  onTurnStarted: (sessionId: string, turnId: string) => void;
-  onItemStarted: (sessionId: string, turnId: string, item: Item) => void;
-  onItemUpdated: (sessionId: string, turnId: string, item: Item) => void;
-  onItemCompleted: (sessionId: string, turnId: string, item: Item) => void;
-  onItemDelta: (sessionId: string, turnId: string, itemId: string, delta: string) => void;
-  onTurnCompleted: (sessionId: string, turnId: string, usage?: TokenUsage | null, hasUnreviewed?: boolean, response?: string) => void;
-  onTokenUsageUpdated: (sessionId: string, turnId: string, usage: TokenUsage) => void;
-  onContextUsageUpdated: (sessionId: string, ctx: ContextUsage) => void;
-  onTurnFailed: (sessionId: string, turnId: string, error: string) => void;
-  onTurnCancelled: (sessionId: string, turnId: string) => void;
-  onSessionTitle: (sessionId: string, title: string) => void;
-  onBackgroundTasksChanged: (sessionId: string, tasks: BackgroundTaskInfo[]) => void;
-};
-
-export const createWsSlice: StateCreator<StoreState, [], [], WsSlice> = (set, get) => ({
-  wsConnected: false,
-  setWsConnected: (connected) => set({ wsConnected: connected }),
-
-  onTurnStarted: (sessionId, turnId) => {
-    get().addLogEntry('DEBUG', `turn:started — ${turnId.slice(0, 8)}`);
-
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: {
-            ...detail,
-            is_running: 1,
-            status: 'running',
-            turns: detail.turns.map((t) => (t.id === turnId ? { ...t, status: 'running' } : t)),
-          },
-        },
-        sessions: state.sessions.map((s) => (s.id === sessionId ? { ...s, is_running: 1, status: 'running' } : s)),
-        activeTurnIndex:
-          state.activeSessionId === sessionId ? detail.turns.findIndex((t) => t.id === turnId) : state.activeTurnIndex,
-      };
-    });
-  },
-
-  onItemStarted: (sessionId, turnId, item) => {
-    get().addLogEntry('DEBUG', `item:started — ${item.type}`);
-
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: upsertItem(detail, turnId, item),
-        },
-      };
-    });
-  },
-
-  onItemUpdated: (sessionId, turnId, item) => {
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: upsertItem(detail, turnId, item),
-        },
-      };
-    });
-  },
-
-  onItemCompleted: (sessionId, turnId, item) => {
-    get().addLogEntry('DEBUG', `item:completed — ${item.type}`);
-
-    // Toast on background_task completion — user needs to know the monitor/subagent finished
-    if (item.type === 'background_task' && (item.status === 'completed' || item.status === 'failed')) {
-      const data = item.data || {};
-      const taskType = (data.task_type as string) || 'unknown';
-      const desc = (data.description as string) || 'background task';
-      const isSubagent = taskType === 'local_agent';
-      const taskLabel = isSubagent ? 'Subagent' : taskType === 'local_bash' ? 'Monitor' : 'Task';
-      if (item.status === 'completed') {
-        get().addToast('success', `${taskLabel} completed: ${desc}`);
-      } else {
-        get().addToast('error', `${taskLabel} failed: ${desc}`);
-      }
-    }
-
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      const updatedDetail = upsertItem(detail, turnId, item);
-
-      let turns = updatedDetail.turns;
-      if (item.type === 'agent_message' && item.data?.text) {
-        turns = turns.map((t) => (t.id === turnId ? { ...t, agent_response: item.data!.text as string } : t));
-      }
-      if (item.type === 'reasoning' && item.data?.text) {
-        turns = turns.map((t) => {
-          if (t.id !== turnId) return t;
-          const thoughts = t.thoughts ? t.thoughts + '\n' + (item.data!.text as string) : (item.data!.text as string);
-          return { ...t, thoughts };
-        });
-      }
-
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: { ...updatedDetail, turns },
-        },
-      };
-    });
-  },
-
-  onItemDelta: (sessionId, turnId, itemId, delta) => {
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: applyDelta(detail, turnId, itemId, delta),
-        },
-      };
-    });
-  },
-
-  onTurnCompleted: (sessionId, turnId, usage, hasUnreviewed, response) => {
-    const inTok = usage?.input_tokens ?? 0;
-    const outTok = usage?.output_tokens ?? 0;
-    get().addLogEntry('INFO', `turn:completed — ${inTok} in / ${outTok} out`);
-
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: {
-            ...detail,
-            is_running: 0,
-            status: 'active',
-            has_unreviewed: hasUnreviewed ? 1 : detail.has_unreviewed,
-            turns: detail.turns.map((t) =>
-              t.id === turnId
-                ? {
-                    ...t,
-                    status: 'completed',
-                    token_usage: usage ?? t.token_usage ?? null,
-                    // Final text for turns that produce no agent_message item
-                    // (repoburg final-action rounds) — keep any streamed text.
-                    ...(response && !t.agent_response ? { agent_response: response } : {}),
-                  }
-                : t,
-            ),
-          },
-        },
-        sessions: state.sessions.map((s) =>
-          s.id === sessionId
-            ? { ...s, is_running: 0, status: 'active', has_unreviewed: hasUnreviewed ? 1 : s.has_unreviewed }
-            : s,
-        ),
-      };
-    });
-  },
-
-  onTokenUsageUpdated: (sessionId, turnId, usage) => {
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: {
-            ...detail,
-            turns: detail.turns.map((t) => (t.id === turnId ? { ...t, token_usage: usage } : t)),
-          },
-        },
-      };
-    });
-  },
-
-  onContextUsageUpdated: (sessionId, ctx) => {
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: { ...detail, contextUsage: ctx },
-        },
-      };
-    });
-  },
-
-  onTurnFailed: (sessionId, turnId, error) => {
-    get().addLogEntry('ERROR', `turn:failed — ${error}`);
-    get().addToast('error', `Turn failed — ${error}`);
-
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      const isAbort = error.includes('cancelled') || error.includes('aborted');
-      const status = isAbort ? 'aborted' : 'failed';
-
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: {
-            ...detail,
-            is_running: 0,
-            status: isAbort ? 'active' : 'failed',
-            turns: detail.turns.map((t) => {
-              if (t.id !== turnId) return t;
-              // Mark all in_progress items as failed — the turn failed so they'll never complete.
-              const items = t.items.map((item) =>
-                item.status === 'in_progress' ? { ...item, status: 'failed' as const } : item,
-              );
-              return { ...t, status, error, items };
-            }),
-          },
-        },
-        sessions: state.sessions.map((s) =>
-          s.id === sessionId ? { ...s, is_running: 0, status: isAbort ? 'active' : 'failed' } : s,
-        ),
-      };
-    });
-  },
-
-  onTurnCancelled: (sessionId, turnId) => {
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-
-      const cancelledTurn = detail.turns.find((t) => t.id === turnId);
-      const restoredPrompt = cancelledTurn?.user_prompt ?? null;
-
-      const newTurns = detail.turns.filter((t) => t.id !== turnId);
-
-      return {
-        restoredPrompt,
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: {
-            ...detail,
-            is_running: 0,
-            status: 'active',
-            turns: newTurns,
-          },
-        },
-        sessions: state.sessions.map((s) => (s.id === sessionId ? { ...s, is_running: 0, status: 'active' } : s)),
-        activeTurnIndex: newTurns.length > 0 ? Math.min(state.activeTurnIndex, newTurns.length - 1) : 0,
-      };
-    });
-  },
-
-  onSessionTitle: (sessionId, title) => {
-    set((state) => ({
-      sessions: state.sessions.map((s) => (s.id === sessionId ? { ...s, title } : s)),
-      sessionDetails: state.sessionDetails[sessionId]
-        ? { ...state.sessionDetails, [sessionId]: { ...state.sessionDetails[sessionId], title } }
-        : state.sessionDetails,
-    }));
-  },
-
-  onBackgroundTasksChanged: (sessionId, tasks) => {
-    set((state) => {
-      const detail = state.sessionDetails[sessionId];
-      if (!detail) return {};
-      return {
-        sessionDetails: {
-          ...state.sessionDetails,
-          [sessionId]: { ...detail, backgroundTasks: tasks },
-        },
-      };
-    });
-  },
-});
-
-/** Insert or replace an item in the correct turn within a SessionDetail. */
-function upsertItem(detail: SessionDetail, turnId: string, item: Item): SessionDetail {
-  return {
-    ...detail,
-    turns: detail.turns.map((turn) => {
-      if (turn.id !== turnId) return turn;
-
-      const existingIdx = turn.items.findIndex((i) => i.external_item_id === item.external_item_id);
-
-      if (existingIdx >= 0) {
-        // Replace with the WS event's item — use the real DB id, not the delta-* placeholder.
-        const newItems = [...turn.items];
-        newItems[existingIdx] = item;
-        return { ...turn, items: newItems };
-      }
-
-      // New item not found by external_item_id — clean up any delta placeholder
-      // agent_message items (created by applyDelta with a fallback stream-text ID
-      // when streamingMessageIds wasn't set). Without this, the placeholder and the
-      // completed item coexist, causing duplicate rendering in thread cards.
-      if (item.type === 'agent_message') {
-        const cleanedItems = turn.items.filter(
-          (i) => !(i.type === 'agent_message' && i.id.startsWith('delta-')),
-        );
-        return { ...turn, items: [...cleanedItems, item] };
-      }
-
-      return { ...turn, items: [...turn.items, item] };
-    }),
-  };
-}
-
-
-
-/** Append a text delta to an existing agent_message item, or create one if not found. */
-function applyDelta(detail: SessionDetail, turnId: string, itemId: string, delta: string): SessionDetail {
-  return {
-    ...detail,
-    turns: detail.turns.map((turn) => {
-      if (turn.id !== turnId) return turn;
-
-      const existingIdx = turn.items.findIndex((i) => i.external_item_id === itemId);
-
-      if (existingIdx >= 0) {
-        const existing = turn.items[existingIdx];
-        const currentText = (existing.data?.text as string) ?? '';
-        const newText = currentText + delta;
-        const newItems = [...turn.items];
-        newItems[existingIdx] = {
-          ...existing,
-          data: { ...existing.data, text: newText },
-        };
-        return { ...turn, items: newItems, agent_response: newText };
-      }
-
-      // Item not found — create in-progress agent_message with delta text
-      const newItem: Item = {
-        id: `delta-${itemId}`,
-        turn_id: turnId,
-        external_item_id: itemId,
-        type: 'agent_message',
-        data: { text: delta },
-        status: 'in_progress',
-        created_at: new Date().toISOString(),
-        updated_at: new Date().toISOString(),
-      };
-      return { ...turn, items: [...turn.items, newItem], agent_response: delta };
-    }),
-  };
-}