refactor(app): move agent tab visibility policy (#851)

2026-05-09 23:28:35 +08:00
parent bb5ac5b1d0
commit 263886fc1c
5 changed files with 78 additions and 23 deletions
@@ -0,0 +1,113 @@
+import type { Agent } from "@/stores/session-store";
+import type { WorkspaceTabSnapshot } from "@/stores/workspace-layout-actions";
+import { shouldAutoOpenAgentTab } from "@/subagents/policies";
+import { normalizeWorkspacePath } from "@/utils/workspace-identity";
+
+function normalizeWorkspaceId(value: string | null | undefined): string {
+  return normalizeWorkspacePath(value) ?? "";
+}
+
+export interface WorkspaceAgentVisibility {
+  activeAgentIds: Set<string>;
+  autoOpenAgentIds: Set<string>;
+  knownAgentIds: Set<string>;
+}
+
+export function deriveWorkspaceAgentVisibility(input: {
+  sessionAgents: Map<string, Agent> | undefined;
+  agentDetails?: Map<string, Agent> | undefined;
+  workspaceDirectory: string | null | undefined;
+}): WorkspaceAgentVisibility {
+  const { sessionAgents, agentDetails, workspaceDirectory } = input;
+  const normalizedWorkspaceDirectory = normalizeWorkspaceId(workspaceDirectory);
+  if ((!sessionAgents && !agentDetails) || !normalizedWorkspaceDirectory) {
+    return {
+      activeAgentIds: new Set<string>(),
+      autoOpenAgentIds: new Set<string>(),
+      knownAgentIds: new Set<string>(),
+    };
+  }
+
+  const activeAgentIds = new Set<string>();
+  const autoOpenAgentIds = new Set<string>();
+  const knownAgentIds = new Set<string>();
+  for (const agent of sessionAgents?.values() ?? []) {
+    if (normalizeWorkspaceId(agent.cwd) !== normalizedWorkspaceDirectory) {
+      continue;
+    }
+    knownAgentIds.add(agent.id);
+    if (!agent.archivedAt) {
+      activeAgentIds.add(agent.id);
+      if (shouldAutoOpenAgentTab(agent)) {
+        autoOpenAgentIds.add(agent.id);
+      }
+    }
+  }
+  for (const agent of agentDetails?.values() ?? []) {
+    if (normalizeWorkspaceId(agent.cwd) !== normalizedWorkspaceDirectory) {
+      continue;
+    }
+    knownAgentIds.add(agent.id);
+  }
+
+  return { activeAgentIds, autoOpenAgentIds, knownAgentIds };
+}
+
+export function buildWorkspaceTabSnapshot(input: {
+  agentVisibility: WorkspaceAgentVisibility;
+  agentsHydrated: boolean;
+  terminalsHydrated: boolean;
+  knownTerminalIds: Iterable<string>;
+  standaloneTerminalIds: Iterable<string>;
+  hasActivePendingDraftCreate: boolean;
+}): WorkspaceTabSnapshot {
+  return {
+    agentsHydrated: input.agentsHydrated,
+    terminalsHydrated: input.terminalsHydrated,
+    activeAgentIds: input.agentVisibility.activeAgentIds,
+    autoOpenAgentIds: input.agentVisibility.autoOpenAgentIds,
+    knownAgentIds: input.agentVisibility.knownAgentIds,
+    knownTerminalIds: input.knownTerminalIds,
+    standaloneTerminalIds: input.standaloneTerminalIds,
+    hasActivePendingDraftCreate: input.hasActivePendingDraftCreate,
+  };
+}
+
+export function workspaceAgentVisibilityEqual(
+  a: WorkspaceAgentVisibility,
+  b: WorkspaceAgentVisibility,
+): boolean {
+  return (
+    setsEqual(a.activeAgentIds, b.activeAgentIds) &&
+    setsEqual(a.autoOpenAgentIds, b.autoOpenAgentIds) &&
+    setsEqual(a.knownAgentIds, b.knownAgentIds)
+  );
+}
+
+function setsEqual(a: Set<string>, b: Set<string>): boolean {
+  if (a.size !== b.size) {
+    return false;
+  }
+  for (const item of a) {
+    if (!b.has(item)) {
+      return false;
+    }
+  }
+  return true;
+}
+
+// Prune agent tabs that are no longer active once agents are hydrated.
+// Archived agents get pruned so that archiving on one client closes the tab on all clients.
+export function shouldPruneWorkspaceAgentTab(input: {
+  agentId: string;
+  agentsHydrated: boolean;
+  activeAgentIds: Set<string>;
+}): boolean {
+  if (!input.agentId.trim()) {
+    return false;
+  }
+  if (!input.agentsHydrated) {
+    return false;
+  }
+  return !input.activeAgentIds.has(input.agentId);
+}