fix(code): remove archived tasks from command center

Author: adboio

apps/code/src/renderer/features/command-center/stores/commandCenterStore.ts

@@ -29,6 +29,7 @@ interface CommandCenterStoreActions {
   setLayout: (preset: LayoutPreset) => void;
   assignTask: (cellIndex: number, taskId: string) => void;
   removeTask: (cellIndex: number) => void;
+  removeTaskById: (taskId: string) => void;
   clearAll: () => void;
   setZoom: (zoom: number) => void;
   zoomIn: () => void;
@@ -86,6 +87,15 @@ export const useCommandCenterStore = create<CommandCenterStore>()(
           return { cells };
         }),
 
+      removeTaskById: (taskId) =>
+        set((state) => {
+          const index = state.cells.indexOf(taskId);
+          if (index === -1) return state;
+          const cells = [...state.cells];
+          cells[index] = null;
+          return { cells };
+        }),
+
       clearAll: () =>
         set((state) => ({
           cells: state.cells.map(() => null),

apps/code/src/renderer/features/tasks/hooks/useArchiveTask.ts

@@ -1,3 +1,4 @@
+import { useCommandCenterStore } from "@features/command-center/stores/commandCenterStore";
 import { getSessionService } from "@features/sessions/service/service";
 import { pinnedTasksApi } from "@features/sidebar/hooks/usePinnedTasks";
 import { useTerminalStore } from "@features/terminal/stores/terminalStore";
@@ -33,6 +34,7 @@ export function useArchiveTask() {
 
     pinnedTasksApi.unpin(taskId);
     useTerminalStore.getState().clearTerminalStatesForTask(taskId);
+    useCommandCenterStore.getState().removeTaskById(taskId);
 
     queryClient.setQueryData<string[]>(
       trpc.archive.archivedTaskIds.queryKey(),