import { discover } from '@agentcommunity/aid';

// A simplified interface based on the A2A AgentCard specification
interface AgentCard {
  protocolVersion: string;
  name: string;
  description: string;
  url: string; // The actual interaction endpoint
  preferredTransport?: string;
  skills: { id: string; name: string; description: string }[];
  securitySchemes?: { [scheme: string]: any };
}

async function connectToA2aAgent(domain: string) {
  try {
    // 1. Discover the AgentCard's location using AID
    console.log(`Discovering A2A AgentCard for ${domain}...`);
    const { record } = await discover(domain);

    // 2. Validate that it's an A2A agent
    if (record.proto !== 'a2a') {
      throw new Error(`Expected 'a2a' protocol, but found '${record.proto}'`);
    }
    console.log(`Found A2A AgentCard at: ${record.uri}`);

    // 3. Fetch the AgentCard from the discovered URI
    const res = await fetch(record.uri);
    if (!res.ok) throw new Error(`HTTP ${res.status}`);
    const agentCard = (await res.json()) as AgentCard;

    console.log(`Successfully fetched card for agent: "${agentCard.name}"`);
    console.log(`Description: ${agentCard.description}`);
    console.log(`Skills:`, agentCard.skills.map((s) => s.name).join(', '));

    // 4. Use the card's data to determine how to communicate
    const interactionEndpoint = agentCard.url; // This is the agent's actual endpoint
    const transport = agentCard.preferredTransport || 'jsonrpc';

    console.log(`Ready to initiate A2A task at ${interactionEndpoint} via ${transport}`);
    // Now, use your A2A library to send a task to the 'interactionEndpoint'...
  } catch (error) {
    const msg = error instanceof Error ? error.message : String(error);
    console.error(`Failed to connect to ${domain}:`, msg);
  }
}

// Replace with a real domain publishing an A2A AID record
connectToA2aAgent('a2a.agentcommunity.org'); // Fictional example domain