// Kratos public-API client: typed fetch wrappers over Ory Kratos' public endpoints.
// Guards the request contracts (URLs, JSON-accept, cookie relay) and the result mapping
// (200/401/4xx, a submit's redirect, tokenized JWT). Live wiring is verified by the
// flow pages; these catch contract drift with a mock fetch.
import { test } from "node:test";
import assert from "node:assert/strict";
import { createKratosPublic, KratosError } from "./kratos-public.ts";

const BASE = "http://kratos:4433";

function res(status: number, body?: unknown, setCookie: string[] = []): Response {
  const headers = new Headers();
  if (body !== undefined) headers.set("content-type", "application/json");
  for (const c of setCookie) headers.append("set-cookie", c);
  return new Response(body === undefined ? null : JSON.stringify(body), { status, headers });
}

// Records each call so a test can assert URL/method/headers/body.
function recorder(handler: (url: string, init: RequestInit | undefined) => Response) {
  const calls: { body: string | undefined; headers: Headers; method: string; redirect: string | undefined; url: string }[] = [];
  const fetchImpl = (async (input: unknown, init?: RequestInit) => {
    calls.push({
      body: init?.body as string | undefined,
      headers: new Headers(init?.headers),
      method: init?.method ?? "GET",
      redirect: init?.redirect,
      url: String(input),
    });
    return handler(String(input), init);
  }) as typeof fetch;
  return { calls, fetchImpl };
}

test("initBrowserFlow gets /self-service/<type>/browser as JSON, relays Set-Cookie, forwards return_to", async () => {
  const flow = { id: "f1", ui: { action: `${BASE}/self-service/login?flow=f1`, method: "POST", nodes: [] } };
  const { calls, fetchImpl } = recorder(() => res(200, flow, ["csrf_token=abc; Path=/; HttpOnly"]));
  const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).initBrowserFlow("login", { returnTo: "http://app/after" });
  assert.deepEqual(out.flow, flow);
  assert.deepEqual(out.setCookie, ["csrf_token=abc; Path=/; HttpOnly"]);
  assert.match(calls[0]!.url, /\/self-service\/login\/browser\?return_to=http%3A%2F%2Fapp%2Fafter$/);
  assert.equal(calls[0]!.headers.get("accept"), "application/json");
});

test("getFlow fetches the flow by id forwarding the browser cookie", async () => {
  const flow = { id: "f2", ui: { action: "x", method: "POST", nodes: [] } };
  const { calls, fetchImpl } = recorder(() => res(200, flow));
  const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).getFlow("registration", "f2", { cookie: "csrf_token=abc" });
  assert.deepEqual(out, flow);
  assert.match(calls[0]!.url, /\/self-service\/registration\/flows\?id=f2$/);
  assert.equal(calls[0]!.headers.get("cookie"), "csrf_token=abc");
});

test("getFlow throws a KratosError carrying the status when the flow is gone (410)", async () => {
  const { fetchImpl } = recorder(() => res(410, { error: { id: "self_service_flow_expired" } }));
  await assert.rejects(
    createKratosPublic({ baseUrl: BASE, fetchImpl }).getFlow("login", "old"),
    (e: unknown) => e instanceof KratosError && e.status === 410,
  );
});

test("submitFlow POSTs the form to the flow on Kratos' own base URL, as a browser would, and relays the redirect + Set-Cookie", async () => {
  const { calls, fetchImpl } = recorder(() => new Response(null, {
    headers: new Headers([["location", "http://app/settings?flow=s1"], ["set-cookie", "plainpages_session=s; Path=/"]]),
    status: 303,
  }));
  const out = await createKratosPublic({ baseUrl: BASE, fetchImpl })
    .submitFlow("recovery", "f 1", { body: "code=123456&method=code", cookie: "csrf_token=abc" });
  assert.deepEqual(out, { location: "http://app/settings?flow=s1", setCookie: ["plainpages_session=s; Path=/"], status: 303 });
  assert.equal(calls[0]!.url, `${BASE}/self-service/recovery?flow=f+1`);
  assert.equal(calls[0]!.method, "POST");
  assert.equal(calls[0]!.redirect, "manual");
  assert.equal(calls[0]!.headers.get("accept"), "text/html");
  assert.equal(calls[0]!.headers.get("referer"), `${BASE}/`);
  assert.equal(calls[0]!.headers.get("content-type"), "application/x-www-form-urlencoded");
  assert.equal(calls[0]!.headers.get("cookie"), "csrf_token=abc");
  assert.equal(calls[0]!.body, "code=123456&method=code");
});

test("submitFlow reports a 4xx without a redirect as a null location, and throws on a Kratos 5xx", async () => {
  const k = (status: number) => createKratosPublic({ baseUrl: BASE, fetchImpl: recorder(() => res(status, { error: "x" })).fetchImpl });
  assert.deepEqual(await k(400).submitFlow("verification", "f", { body: "x=1" }), { location: null, setCookie: [], status: 400 });
  await assert.rejects(k(500).submitFlow("verification", "f", { body: "x=1" }), (e: unknown) => e instanceof KratosError && e.status === 500);
});

test("whoami returns the session on 200 (cookie forwarded) and null on 401", async () => {
  const session = { active: true, identity: { id: "u1", traits: { email: "a@b" } } };
  const { calls, fetchImpl } = recorder((url) => (url.endsWith("/sessions/whoami") ? res(200, session) : res(401)));
  const k = createKratosPublic({ baseUrl: BASE, fetchImpl });
  assert.deepEqual(await k.whoami({ cookie: "plainpages_session=s" }), session);
  assert.equal(calls[0]!.headers.get("cookie"), "plainpages_session=s");
  assert.equal(await createKratosPublic({ baseUrl: BASE, fetchImpl: (async () => res(401)) as typeof fetch }).whoami(), null);
});

test("whoami?tokenize_as mints a session JWT via the tokenizer template", async () => {
  const session = { active: true, identity: { id: "u1" }, tokenized: "header.payload.sig" };
  const { calls, fetchImpl } = recorder(() => res(200, session));
  const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).whoami({ cookie: "plainpages_session=s", tokenizeAs: "plainpages" });
  assert.equal(out?.tokenized, "header.payload.sig");
  assert.match(calls[0]!.url, /\/sessions\/whoami\?tokenize_as=plainpages$/);
});

test("whoami throws on an unexpected upstream error", async () => {
  const { fetchImpl } = recorder(() => res(500, { error: "boom" }));
  await assert.rejects(createKratosPublic({ baseUrl: BASE, fetchImpl }).whoami(), KratosError);
});

test("createLogoutFlow returns the logout URL/token on 200 (cookie forwarded) and null on 401 (no session)", async () => {
  const flow = { logout_token: "lt", logout_url: `${BASE}/self-service/logout?token=lt` };
  const { calls, fetchImpl } = recorder((url) => (url.endsWith("/self-service/logout/browser") ? res(200, flow) : res(401)));
  const out = await createKratosPublic({ baseUrl: BASE, fetchImpl }).createLogoutFlow({ cookie: "plainpages_session=s" });
  assert.deepEqual(out, { logoutToken: "lt", logoutUrl: flow.logout_url });
  assert.match(calls[0]!.url, /\/self-service\/logout\/browser$/);
  assert.equal(calls[0]!.headers.get("cookie"), "plainpages_session=s");
  assert.equal(await createKratosPublic({ baseUrl: BASE, fetchImpl: (async () => res(401)) as typeof fetch }).createLogoutFlow(), null);
});