feat: implement daemon connectivity updates and relay protocol versioning

2026-02-20 19:27:56 +07:00
parent 6dad3212da
commit 0350f00840
49 changed files with 4146 additions and 1351 deletions
@@ -0,0 +1,60 @@
+import type { FetchAgentsEntry } from "@server/client/daemon-client";
+import { useSessionStore, type Agent } from "@/stores/session-store";
+import {
+  derivePendingPermissionKey,
+  normalizeAgentSnapshot,
+} from "@/utils/agent-snapshots";
+import { resolveProjectPlacement } from "@/utils/project-placement";
+
+type PendingPermissionEntry = {
+  key: string;
+  agentId: string;
+  request: Agent["pendingPermissions"][number];
+};
+
+export function buildAgentDirectoryState(input: {
+  serverId: string;
+  entries: FetchAgentsEntry[];
+}): {
+  agents: Map<string, Agent>;
+  pendingPermissions: Map<string, PendingPermissionEntry>;
+} {
+  const agents = new Map<string, Agent>();
+  const pendingPermissions = new Map<string, PendingPermissionEntry>();
+
+  for (const entry of input.entries) {
+    const normalized = normalizeAgentSnapshot(entry.agent, input.serverId);
+    const projectPlacement = resolveProjectPlacement({
+      projectPlacement: entry.project,
+      cwd: normalized.cwd,
+    });
+    const agent: Agent = {
+      ...normalized,
+      projectPlacement,
+    };
+    agents.set(agent.id, agent);
+
+    for (const request of agent.pendingPermissions) {
+      const key = derivePendingPermissionKey(agent.id, request);
+      pendingPermissions.set(key, { key, agentId: agent.id, request });
+    }
+  }
+
+  return { agents, pendingPermissions };
+}
+
+export function applyFetchedAgentDirectory(input: {
+  serverId: string;
+  entries: FetchAgentsEntry[];
+}): { agents: Map<string, Agent> } {
+  const { agents, pendingPermissions } = buildAgentDirectoryState(input);
+  const store = useSessionStore.getState();
+  store.setAgents(input.serverId, agents);
+  for (const agent of agents.values()) {
+    store.setAgentLastActivity(agent.id, agent.lastActivityAt);
+  }
+  store.setPendingPermissions(input.serverId, pendingPermissions);
+  store.setInitializingAgents(input.serverId, new Map());
+  store.setHasHydratedAgents(input.serverId, true);
+  return { agents };
+}