Reduce workspace, agent, and chat sync traffic (#2028)

* feat(sync): keep live data scoped and current Only viewed chats receive live timeline rows, while directory state now uses one subscribed bootstrap followed by ordered deltas. Legacy clients and daemons retain their existing behavior through centralized compatibility gates. * fix(sync): preserve selective delivery boundaries Keep selective timeline capability app-owned, union viewed sets across shared sockets, and reconcile directory side effects from accepted state. Visibility and archive suppression now follow their existing authoritative boundaries. * test(app): align browser fixtures with runtime contracts * fix(sync): preserve mixed-client delivery guarantees * fix(sync): finish paged history before advancing * fix(sync): replay live deltas after refresh failures * test(server): model socket capability lookup * test(app): use platform shortcut for split-pane coverage * fix(app): keep hidden timelines dormant * test(app): normalize split-pane shortcut setup * fix(app): preserve sync state across retries * fix(app): preserve background sync state across races * fix directory bootstrap reconciliation edge cases * fix selective timeline compatibility races * fix superseded directory sync races * fix(sync): centralize directory replica ordering Own agent and workspace refresh transactions in HostRuntime so reconnect epochs, buffered updates, and lifecycle invalidation share one ordering boundary. Route timeline metadata and mixed-capability stream delivery through their authoritative runtime sources. * fix(app): keep timeline requests runtime-scoped * fix(app): close directory bootstrap races
2026-07-17 17:03:34 +02:00
parent 6c99efae52
commit 1977d330ed
57 changed files with 5783 additions and 1343 deletions
@@ -0,0 +1,113 @@
+import { afterEach, describe, expect, it } from "vitest";
+import type { DaemonClient } from "@getpaseo/client/internal/daemon-client";
+import { useSessionStore } from "@/stores/session-store";
+import { DirectoryRefreshSupersededError, DirectorySync } from "./index";
+
+class FakeDirectoryClient {
+  fetchAgentsCalls = 0;
+  fetchWorkspacesCalls = 0;
+
+  on(): () => void {
+    return () => undefined;
+  }
+
+  async fetchAgents(): Promise<Awaited<ReturnType<DaemonClient["fetchAgents"]>>> {
+    this.fetchAgentsCalls += 1;
+    return {
+      requestId: "agents",
+      entries: [],
+      pageInfo: { hasMore: false, nextCursor: null, prevCursor: null },
+    };
+  }
+
+  async fetchWorkspaces(): Promise<Awaited<ReturnType<DaemonClient["fetchWorkspaces"]>>> {
+    this.fetchWorkspacesCalls += 1;
+    return {
+      requestId: "workspaces",
+      entries: [],
+      emptyProjects: [],
+      pageInfo: { hasMore: false, nextCursor: null, prevCursor: null },
+    };
+  }
+}
+
+const serverIds = new Set<string>();
+
+function createDirectory(serverId: string): {
+  client: FakeDirectoryClient;
+  directory: DirectorySync;
+} {
+  serverIds.add(serverId);
+  const client = new FakeDirectoryClient();
+  const directory = new DirectorySync(serverId, {
+    drainQueuedAgentMessage: () => undefined,
+    markAgentLoading: () => undefined,
+    markAgentReady: () => undefined,
+    markAgentError: () => undefined,
+  });
+  directory.connectionChanged({
+    client: client as unknown as DaemonClient,
+    status: "online",
+    source: { clientGeneration: 1, connectionEpoch: 1 },
+  });
+  return { client, directory };
+}
+
+afterEach(() => {
+  for (const serverId of serverIds) useSessionStore.getState().clearSession(serverId);
+  serverIds.clear();
+});
+
+describe("DirectorySync session readiness", () => {
+  it("waits for workspace capability metadata before choosing the workspace protocol", async () => {
+    const serverId = "workspace-metadata";
+    const { client, directory } = createDirectory(serverId);
+
+    const refresh = directory.refreshWorkspaces({ subscribe: true });
+    await Promise.resolve();
+    expect(client.fetchWorkspacesCalls).toBe(0);
+
+    const store = useSessionStore.getState();
+    store.initializeSession(serverId, client as unknown as DaemonClient, 1);
+    await Promise.resolve();
+    expect(client.fetchWorkspacesCalls).toBe(0);
+
+    store.updateSessionServerInfo(serverId, {
+      serverId,
+      hostname: null,
+      version: "test",
+      features: { workspaceMultiplicity: true },
+    });
+    await refresh;
+
+    expect(client.fetchWorkspacesCalls).toBe(1);
+    expect(useSessionStore.getState().sessions[serverId]?.hasHydratedWorkspaces).toBe(true);
+    directory.dispose();
+  });
+
+  it("rejects a session wait on disconnect so the reconnect can refresh", async () => {
+    const serverId = "session-wait-reconnect";
+    const { client, directory } = createDirectory(serverId);
+    const staleRefresh = directory.refreshAgents();
+    await Promise.resolve();
+
+    directory.connectionChanged({
+      client: null,
+      status: "offline",
+      source: { clientGeneration: 1, connectionEpoch: 1 },
+    });
+    await expect(staleRefresh).rejects.toBeInstanceOf(DirectoryRefreshSupersededError);
+
+    directory.connectionChanged({
+      client: client as unknown as DaemonClient,
+      status: "online",
+      source: { clientGeneration: 1, connectionEpoch: 2 },
+    });
+    const currentRefresh = directory.refreshAgents();
+    useSessionStore.getState().initializeSession(serverId, client as unknown as DaemonClient, 1);
+    await currentRefresh;
+
+    expect(client.fetchAgentsCalls).toBe(1);
+    directory.dispose();
+  });
+});