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

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
  );
}