import { test } from "node:test";
import assert from "node:assert/strict";
import { mkdtemp, writeFile, readFile, readdir } from "node:fs/promises";
import { tmpdir } from "node:os";
import { join } from "node:path";

import { seal, unseal, compareVersions } from "../dist/primitives/index.js";
import { registerModuleTools, collectTools, resetTools, serveTools, listTools } from "../dist/tools/index.js";
import { runProvisioner, type Grant } from "../dist/provisioner/index.js";

test("seal round-trips and rejects the wrong key", () => {
  const sealed = seal("hunter2", "node-key");
  assert.equal(unseal(sealed, "node-key"), "hunter2");
  assert.notEqual(sealed, "hunter2");
  assert.throws(() => unseal(sealed, "wrong-key"));
});

test("semver orders releases", () => {
  assert.equal(compareVersions("1.2.3", "1.2.10"), -1);
  assert.equal(compareVersions("2.0.0", "1.9.9"), 1);
  assert.equal(compareVersions("v1.0.0", "1.0.0"), 0);
});

test("module tools register and collect, a thrower is skipped not fatal", () => {
  resetTools();
  registerModuleTools("umami", () => [
    { name: "umami_stats", description: "d", input: {}, run: async () => 1 },
  ]);
  registerModuleTools("broken", () => {
    throw new Error("no token");
  });
  const collected = collectTools({});
  const umami = collected.find((c) => c.module === "umami");
  const broken = collected.find((c) => c.module === "broken");
  assert.equal(umami?.tools.length, 1);
  assert.equal(broken?.tools.length, 0);
});

test("provisioner creates a sealed credential for a grant, then removes on withdrawal", async () => {
  const dir = await mkdtemp(join(tmpdir(), "prov-"));
  const created: string[] = [];
  const removed: string[] = [];

  const grant: Grant = { resource: "analytics", consumer: "webapp", node: "anchor", values: { name: "webapp" } };
  await writeFile(join(dir, "webapp.grant.json"), JSON.stringify(grant));

  const stop = runProvisioner(
    "analytics",
    {
      async create(g) {
        created.push(g.consumer);
        return { fields: { siteId: "abc", snippet: "<script>", dashboard: "https://x/webapp" } };
      },
      async remove(g) {
        removed.push(g.consumer);
      },
    },
    { grants: dir, sealKey: "k", everyMs: 20 },
  );

  await waitFor(() => created.length === 1, 2000);
  const files = await readdir(dir);
  const credFile = files.find((f) => f.endsWith(".credential"));
  assert.ok(credFile, "a credential file was written");
  const sealed = await readFile(join(dir, credFile!), "utf8");
  const body = JSON.parse(unseal(sealed, "k")) as { fields: Record<string, string> };
  assert.equal(body.fields.siteId, "abc");

  // Withdraw the grant → the harness removes via the adapter and deletes the credential.
  await (await import("node:fs/promises")).rm(join(dir, "webapp.grant.json"));
  await waitFor(() => removed.length === 1, 2000);
  stop();
});

test("modules can SERVE: a real async tool, loaded and invoked over the broker", async () => {
  resetTools();

  // Stand up a fake upstream the tool actually calls over the network — proving a tool that does
  // real work (not a pure function) serves end to end.
  const { createServer } = await import("node:http");
  const server = createServer((_req, res) => {
    res.setHeader("content-type", "application/json");
    res.end(JSON.stringify({ id: "site-123" }));
  });
  await new Promise<void>((r) => server.listen(0, r));
  const addr = server.address() as { port: number };
  const base = `http://127.0.0.1:${addr.port}`;

  // A module registers its tool exactly as umami does — the tool calls the upstream and returns a
  // result computed from it.
  registerModuleTools("demo", () => [
    {
      name: "create_site",
      description: "register a site and return its snippet",
      input: { domain: { type: "string" } },
      run: async (args) => {
        const res = await fetch(`${base}/api/websites`, { method: "POST" });
        const { id } = (await res.json()) as { id: string };
        return { domain: String(args.domain), snippet: `<script data-website-id="${id}"></script>` };
      },
    },
  ]);

  const broker = memBroker();
  const stop = await serveTools(broker, {});

  // Invoke it the way a caller (mesh-control's command API) would — over the broker, by name.
  const result = await broker.request<{ tool: string; args: Record<string, unknown> }, { snippet: string }>(
    "tools.invoke",
    { tool: "create_site", args: { domain: "my-app" } },
  );
  assert.match(result.snippet, /data-website-id="site-123"/);

  // Discovery works, and an unknown tool is refused rather than silently dropped.
  assert.deepEqual(listTools({}).map((t) => t.name), ["create_site"]);
  await assert.rejects(broker.request("tools.invoke", { tool: "nope", args: {} }));

  stop();
  server.close();
});

test("serving refuses two modules exposing one tool name", async () => {
  resetTools();
  registerModuleTools("a", () => [{ name: "dup", description: "", input: {}, run: async () => 1 }]);
  registerModuleTools("b", () => [{ name: "dup", description: "", input: {}, run: async () => 2 }]);
  await assert.rejects(serveTools(memBroker(), {}), /exposed by two modules/);
});

// A minimal in-memory broker: request routes to a registered handle. Enough to serve tools; the
// real broker binding is the mesh's, provided by the hosting runtime.
function memBroker() {
  const handlers = new Map<string, (b: unknown) => Promise<unknown>>();
  return {
    async request<Req, Res>(key: string, body: Req): Promise<Res> {
      const h = handlers.get(key);
      if (!h) throw new Error(`no handler for ${key}`);
      return (await h(body)) as Res;
    },
    async handle<Req, Res>(key: string, handler: (b: Req) => Promise<Res>): Promise<() => void> {
      handlers.set(key, handler as (b: unknown) => Promise<unknown>);
      return () => handlers.delete(key);
    },
    async publish(): Promise<void> {},
    async subscribe(): Promise<() => void> {
      return () => {};
    },
    async close(): Promise<void> {},
  };
}

async function waitFor(cond: () => boolean, ms: number): Promise<void> {
  const start = Date.now();
  while (!cond()) {
    if (Date.now() - start > ms) throw new Error("timed out waiting");
    await new Promise((r) => setTimeout(r, 10));
  }
}