Admin Socket.IO

import type { Server as HTTPServer } from "node:http";
import { Server, type Socket } from "socket.io";
import { createAdminGateway } from "@agentium/admin";
import type { StorageDriver } from "@agentium/core";
import { CalculatorToolkit } from "@agentium/core/toolkits";

export async function attachOperatorGateway(
  httpServer: HTTPServer,
  storage: StorageDriver,
  verifyOperator: (token: unknown) => Promise<boolean>,
) {
  const io = new Server(httpServer);
  const { hydrate } = createAdminGateway({
    io,
    storage,
    toolkits: [new CalculatorToolkit()],
    namespace: "/agentium-admin",
    authMiddleware: (socket: Socket, next: (error?: Error) => void) => {
      void verifyOperator(socket.handshake.auth?.token).then(
        (allowed) => next(allowed ? undefined : new Error("Operator access required")),
        () => next(new Error("Operator access required")),
      );
    },
  });
  try {
    const restored = await hydrate();
    return { io, restored };
  } catch (error) {
    await new Promise<void>((resolve) => io.close(() => resolve()));
    throw error;
  }
}

import { io } from "socket.io-client";

const url = process.env.ADMIN_URL ?? "http://127.0.0.1:3000/agentium-admin";
const token = process.env.ADMIN_TOKEN;
if (!token) throw new Error("Set ADMIN_TOKEN");

const socket = io(url, { auth: { token }, reconnection: false, timeout: 5_000 });
try {
  await new Promise<void>((resolve, reject) => {
    socket.once("connect", resolve);
    socket.once("connect_error", reject);
  });
  const response: unknown = await socket.timeout(5_000).emitWithAck("admin.agent.list", {});
  if (typeof response !== "object" || response === null || !("ok" in response) || response.ok !== true) {
    throw new Error("Admin listing failed; inspect the acknowledgement error");
  }
  console.log(JSON.stringify(response, null, 2));
} finally {
  socket.disconnect();
}