import assert from "node:assert/strict";
import { limitedBody, synthesize, wikipediaClient } from "../src/network.js";
const signal = new AbortController().signal;
let calls: { url: string; init?: RequestInit }[] = [];
const mock = (async (url: URL | RequestInfo, init?: RequestInit) => {
  calls.push({ url: String(url), init });
  if (String(url).includes("list=search")) return Response.json({ query: { search: [{ title: "北京" }] } });
  return Response.json({ query: { pages: [{ title: "北京", extract: "北京是中国的首都。" }] } });
}) as typeof fetch;
const wiki = wikipediaClient(mock);
assert.deepEqual(await wiki.search("北京", signal), [{ title: "北京", description: "Chinese Wikipedia" }]);
const page = await wiki.read("北京", signal);
assert.equal(page.text, "北京是中国的首都。");
assert(page.attribution.includes("CC BY-SA"));
assert(page.url.startsWith("https://zh.wikipedia.org/wiki/"));
const count = calls.length;
await wiki.read("北京", signal);
assert.equal(calls.length, count, "Repeated loads use the bounded cache");
assert((calls[0].init?.headers as Record<string, string>)["user-agent"].includes("QuixosReading"));
assert(new URL(calls[0].url).searchParams.has("maxlag"));
await assert.rejects(() => wiki.read("a|b", signal), /Invalid/);
await assert.rejects(() => limitedBody(new Response("x".repeat(50)), 10), /too large/);
await assert.rejects(
  () => limitedBody(new Response("", { status: 429, headers: { "retry-after": "10" } }), 10),
  /Retry after 10/,
);
let speechRequest: RequestInit | undefined;
const speech = (async (url: URL | RequestInfo, init?: RequestInit) => {
  assert(String(url).startsWith("https://api.elevenlabs.io/v1/text-to-speech/voice/stream"));
  speechRequest = init;
  return new Response(new Uint8Array([1, 2, 3]), { headers: { "content-type": "audio/mpeg" } });
}) as typeof fetch;
assert.deepEqual(await synthesize("你好", "fixture-private-key", "voice", signal, speech), new Uint8Array([1, 2, 3]));
assert.equal((speechRequest?.headers as Record<string, string>)["xi-api-key"], "fixture-private-key");
assert.equal(JSON.parse(speechRequest?.body as string).model_id, "eleven_multilingual_v2");
await assert.rejects(() => synthesize("你好", "", "voice", signal, speech), /API key/);
await assert.rejects(() => synthesize("中".repeat(1201), "key", "voice", signal, speech), /1,200/);
await assert.rejects(
  () => synthesize("你好", "key", "voice", signal, (async () => new Response("", { status: 401 })) as typeof fetch),
  /did not accept/,
);
console.log("PASS Wikipedia bounds, identification, attribution, cache and speech headers/errors");