Netcatty/application/state/useAgentDiscovery.ts

import { startTransition, useCallback, useEffect, useRef, useState } from 'react';
import type { DiscoveredAgent, ExternalAgentConfig } from '../../infrastructure/ai/types';
import { getExternalAgentSdkBackend } from '../../infrastructure/ai/managedAgents';

interface NetcattyBridge {
  aiDiscoverAgents(options?: { refreshShellEnv?: boolean; apiKeyPresent?: boolean }): Promise<DiscoveredAgent[]>;
}

function getBridge(): NetcattyBridge | undefined {
  return (window as unknown as { netcatty?: NetcattyBridge }).netcatty;
}

const AGENT_DISCOVERY_CACHE_TTL_MS = 60_000;
let agentDiscoveryCache: {
  agents: DiscoveredAgent[];
  apiKeyPresent: boolean;
  updatedAt: number;
} | null = null;
const agentDiscoveryPromises = new Map<string, Promise<DiscoveredAgent[]>>();
let agentDiscoveryWriteGeneration = 0;

export function useAgentDiscovery(
  externalAgents: ExternalAgentConfig[],
  setExternalAgents?: (value: ExternalAgentConfig[] | ((prev: ExternalAgentConfig[]) => ExternalAgentConfig[])) => void,
  options?: {
    enabled?: boolean;
    schedule?: (task: () => void) => () => void;
  },
) {
  const enabled = options?.enabled ?? true;
  const schedule = options?.schedule;
  const [discoveredAgents, setDiscoveredAgents] = useState<DiscoveredAgent[]>(
    () => agentDiscoveryCache?.agents ?? [],
  );
  const [isDiscovering, setIsDiscovering] = useState(false);
  const discoverSeqRef = useRef(0);
  const mountedRef = useRef(true);
  const enabledRef = useRef(enabled);

  enabledRef.current = enabled;

  useEffect(() => () => {
    mountedRef.current = false;
    discoverSeqRef.current += 1;
  }, []);

  const cursorApiKeyPresent = externalAgents.some(
    (agent) => agent.id === "discovered_cursor" && Boolean(agent.apiKey),
  );

  const discover = useCallback(async (discoverOptions?: { refreshShellEnv?: boolean }) => {
    if (!enabledRef.current) return;
    const bridge = getBridge();
    if (!bridge) return;

    const forceRefresh = discoverOptions?.refreshShellEnv === true;
    const cacheFresh =
      agentDiscoveryCache
      && agentDiscoveryCache.apiKeyPresent === cursorApiKeyPresent
      && Date.now() - agentDiscoveryCache.updatedAt < AGENT_DISCOVERY_CACHE_TTL_MS;

    if (!forceRefresh && cacheFresh) {
      startTransition(() => setDiscoveredAgents(agentDiscoveryCache?.agents ?? []));
      return;
    }

    setIsDiscovering(true);
    const discoverSeq = ++discoverSeqRef.current;
    const writeGeneration = ++agentDiscoveryWriteGeneration;
    const promiseKey = JSON.stringify({
      apiKeyPresent: cursorApiKeyPresent,
      refreshShellEnv: forceRefresh,
    });
    try {
      let discoveryPromise = agentDiscoveryPromises.get(promiseKey) ?? null;
      if (!discoveryPromise) {
        const sharedPromise = bridge.aiDiscoverAgents({
          ...discoverOptions,
          apiKeyPresent: cursorApiKeyPresent,
        }).finally(() => {
          if (agentDiscoveryPromises.get(promiseKey) === sharedPromise) {
            agentDiscoveryPromises.delete(promiseKey);
          }
        });
        agentDiscoveryPromises.set(promiseKey, sharedPromise);
        discoveryPromise = sharedPromise;
      }
      const agents = await discoveryPromise;
      if (
        !mountedRef.current
        || !enabledRef.current
        || discoverSeq !== discoverSeqRef.current
        || writeGeneration !== agentDiscoveryWriteGeneration
      ) return;
      agentDiscoveryCache = {
        agents,
        apiKeyPresent: cursorApiKeyPresent,
        updatedAt: Date.now(),
      };
      startTransition(() => setDiscoveredAgents(agents));
    } catch (err) {
      console.error('Agent discovery failed:', err);
    } finally {
      if (mountedRef.current && discoverSeq === discoverSeqRef.current) {
        setIsDiscovering(false);
      }
    }
  }, [cursorApiKeyPresent]);

  useEffect(() => {
    discoverSeqRef.current += 1;
    if (!enabled) {
      setIsDiscovering(false);
    }
  }, [cursorApiKeyPresent, enabled]);

  useEffect(() => {
    if (!enabled) return;

    let cancelled = false;
    const runDiscover = () => {
      if (!cancelled) void discover();
    };

    if (schedule) {
      const cancel = schedule(runDiscover);
      return () => {
        cancelled = true;
        cancel();
      };
    }

    if (typeof requestIdleCallback === 'function') {
      const idleId = requestIdleCallback(runDiscover);
      return () => {
        cancelled = true;
        cancelIdleCallback(idleId);
      };
    }

    const timeoutId = setTimeout(runDiscover, 0);
    return () => {
      cancelled = true;
      clearTimeout(timeoutId);
    };
  }, [discover, enabled, schedule]);

  // Auto-update args for already-configured discovered agents when
  // the canonical args from discovery change (e.g. after an app update).
  useEffect(() => {
    if (!setExternalAgents || discoveredAgents.length === 0) return;
    if (!enabled) return;

    setExternalAgents((prev) => {
      let changed = false;
      const next = prev.map((ea) => {
        // Only update agents that were auto-discovered (id starts with "discovered_")
        if (!ea.id.startsWith('discovered_')) return ea;

        const match = discoveredAgents.find(
          (da) => ea.command === da.path || ea.command === da.command,
        );
        if (!match) return ea;

        // Check if args, SDK backend, or managed SDK path env differ.
        const currentArgs = JSON.stringify(ea.args || []);
        const newArgs = JSON.stringify(match.args);
        const backend = match.sdkBackend ?? match.command;
        const backendChanged = getExternalAgentSdkBackend(ea) !== backend
          || Boolean(ea.acpCommand)
          || JSON.stringify(ea.acpArgs || []) !== JSON.stringify([]);
        const matchPath = match.binPath || match.path;
        const env = match.command === 'claude'
          ? { ...(ea.env ?? {}), CLAUDE_CODE_EXECUTABLE: matchPath }
          : match.command === 'opencode'
            ? { ...(ea.env ?? {}), OPENCODE_BIN: matchPath }
            : ea.env;
        const envChanged =
          (match.command === 'claude' && ea.env?.CLAUDE_CODE_EXECUTABLE !== matchPath)
          || (match.command === 'opencode' && ea.env?.OPENCODE_BIN !== matchPath);
        const versionChanged = Boolean(match.version) && ea.cliVersion !== match.version;
        if (currentArgs !== newArgs || backendChanged || envChanged || versionChanged) {
          changed = true;
          const { acpCommand: _legacyCommand, acpArgs: _legacyArgs, ...rest } = ea;
          return {
            ...rest,
            args: match.args,
            sdkBackend: backend,
            ...(match.version ? { cliVersion: match.version } : {}),
            ...(env ? { env } : {}),
          };
        }
        return ea;
      });
      return changed ? next : prev;
    });
  }, [discoveredAgents, enabled, setExternalAgents]);

  // Filter out agents that are already configured as external agents
  const unconfiguredAgents = discoveredAgents.filter(
    (da) => !externalAgents.some(
      (ea) => ea.command === da.command || ea.command === da.path,
    ),
  );

  // Build ExternalAgentConfig from a discovered agent
  const enableAgent = useCallback(
    (agent: DiscoveredAgent): ExternalAgentConfig => {
      const backend = agent.sdkBackend ?? agent.command;
      return {
        id: `discovered_${agent.command}`,
        name: agent.name,
        command: agent.binPath || agent.path || agent.command,
        args: agent.args,
        icon: agent.icon,
        enabled: true,
        sdkBackend: backend,
        ...(agent.version ? { cliVersion: agent.version } : {}),
        ...(agent.command === 'claude'
          ? { env: { CLAUDE_CODE_EXECUTABLE: agent.binPath || agent.path || '' } }
          : agent.command === 'opencode'
            ? { env: { OPENCODE_BIN: agent.binPath || agent.path || '' } }
          : {}),
      };
    },
    [],
  );

  return {
    discoveredAgents,
    unconfiguredAgents,
    isDiscovering,
    rediscover: () => discover({ refreshShellEnv: true }),
    enableAgent,
  };
}