源码快照 @ 2026-09-01 (source-only, 剔除非源码素材)

packages/agent/gui/agent-gui/agentGuiNode/controller/useAgentGUITargetConnectionState.ts

@@ -0,0 +1,65 @@
+import { useMemo } from "react";
+import type {
+  AgentGUITargetConnectionSource,
+  AgentGUITargetConnectionState
+} from "../../../types";
+import { useEngineSelector } from "../../../shared/engine/useEngineSelector";
+import {
+  AgentGUITargetConnectionController,
+  type AgentGUITargetConnectionBinding
+} from "./AgentGUITargetConnectionController";
+
+export function useAgentGUITargetConnectionState(input: {
+  agentTargetId?: string | null;
+  source?: AgentGUITargetConnectionSource | null;
+}): {
+  blocked: boolean;
+  visibleState: AgentGUITargetConnectionState | null;
+} {
+  const binding = useMemo<AgentGUITargetConnectionBinding>(
+    () => ({
+      getSnapshot: () => {
+        const agentTargetId = input.agentTargetId?.trim() ?? "";
+        return agentTargetId
+          ? (input.source?.getConnectionState(agentTargetId) ?? null)
+          : null;
+      },
+      subscribe: (listener) =>
+        input.source?.subscribe(listener) ?? (() => undefined)
+    }),
+    [input.agentTargetId, input.source]
+  );
+  const state = useEngineSelector(
+    binding,
+    identityTargetConnectionState,
+    targetConnectionStatesEqual
+  );
+  const controller = useMemo(
+    () => new AgentGUITargetConnectionController({ source: binding }),
+    [binding]
+  );
+  const visibleState = useEngineSelector(
+    controller,
+    identityTargetConnectionState,
+    targetConnectionStatesEqual
+  );
+  return {
+    blocked: state?.status === "connecting" || state?.status === "unavailable",
+    visibleState
+  };
+}
+
+function identityTargetConnectionState(
+  state: AgentGUITargetConnectionState | null
+): AgentGUITargetConnectionState | null {
+  return state;
+}
+
+function targetConnectionStatesEqual(
+  left: AgentGUITargetConnectionState | null,
+  right: AgentGUITargetConnectionState | null
+): boolean {
+  return (
+    left?.status === right?.status && left?.retryAttempt === right?.retryAttempt
+  );
+}