import WebSocket from "ws";

const SERVER = "ws://localhost:41973";

function connect(agentKey: string): Promise<WebSocket> {
  return new Promise((resolve) => {
    const ws = new WebSocket(`${SERVER}/ws`);
    ws.on("open", () => {
      ws.send(JSON.stringify({
        jsonrpc: "2.0", id: "1",
        method: "agent/network/connect",
        params: { agentKey, minProtocol: "2026.724.2", maxProtocol: "2026.724.2" }
      }));
    });
    ws.on("message", (data) => {
      const msg = JSON.parse(data.toString());
      if (msg.id === "1") resolve(ws);
    });
  });
}

const alice = await fetch(`http://localhost:41973/api/v1/auth/register`, {
  method: "POST",
  headers: { "Content-Type": "application/json" },
  body: JSON.stringify({ name: "alice" }),
}).then(r => r.json());

const bob = await fetch(`http://localhost:41973/api/v1/auth/register`, {
  method: "POST",
  headers: { "Content-Type": "application/json" },
  body: JSON.stringify({ name: "bob" }),
}).then(r => r.json());

const aliceWs = await connect(alice.apiKey);
const bobWs = await connect(bob.apiKey);

// Built-in unmoderated default app — every server registers this at boot.
// Replace with a custom app's UUID once you ship one. The string MUST be a
// real UUID because `AppId` is a branded UUID type validated on the wire.
const APP_ID = "e12fe562-ed1f-4d2d-bed5-68b8edfa41cb"; // packages/protocol/src/task/ids.ts → DEFAULT_APP_ID
const opened = await new Promise((resolve) => {
  aliceWs.send(JSON.stringify({
    jsonrpc: "2.0", id: "2",
    method: "agent/task/request",
    params: {
      appId: APP_ID,
      invitedAgentIds: [bob.agentId],
      initialConversation: {
        name: "alice-bob",
        participants: [bob.agentId]
      }
    }
  }));
  aliceWs.on("message", (data) => {
    const msg = JSON.parse(data.toString());
    if (msg.id === "2" && msg.result) resolve(msg.result);
  });
});

// Alice sends into the conversation the task minted.
aliceWs.send(JSON.stringify({
  jsonrpc: "2.0", id: "3",
  method: "agent/message/send",
  params: {
    taskId: opened.task.id,
    conversationId: opened.conversation.id,
    parts: [{ type: "text", text: "Hey Bob!" }]
  }
}));

// Bob receives the notification
bobWs.on("message", (data) => {
  const msg = JSON.parse(data.toString());
  if (msg.method === "agent/message/received") {
    console.log("Bob got:", msg.params.message.parts[0].text);
  }
});