import test from "node:test";
import assert from "node:assert/strict";
import os from "node:os";
import path from "node:path";
import { mkdtemp, rm } from "node:fs/promises";
import { NextRequest } from "next/server";

let runtimeRoot = "";
let validateAiAccountConnection: (typeof import("../src/lib/boss-master-agent"))["validateAiAccountConnection"];
let readState: (typeof import("../src/lib/boss-data"))["readState"];
let deviceHeartbeatRoute: (typeof import("../src/app/api/device-heartbeat/route"))["POST"];
let saveAiAccount: (typeof import("../src/lib/boss-data"))["saveAiAccount"];

async function setup() {
  if (runtimeRoot) return;

  runtimeRoot = await mkdtemp(path.join(os.tmpdir(), "boss-ai-account-"));
  process.env.BOSS_RUNTIME_ROOT = runtimeRoot;
  process.env.BOSS_STATE_FILE = path.join(runtimeRoot, "boss-state.json");

  const [masterAgent, data, heartbeatModule] = await Promise.all([
    import("../src/lib/boss-master-agent.ts"),
    import("../src/lib/boss-data.ts"),
    import("../src/app/api/device-heartbeat/route.ts"),
  ]);

  validateAiAccountConnection = masterAgent.validateAiAccountConnection;
  readState = data.readState;
  saveAiAccount = data.saveAiAccount;
  deviceHeartbeatRoute = heartbeatModule.POST;
}

test.after(async () => {
  if (runtimeRoot) {
    await rm(runtimeRoot, { recursive: true, force: true });
  }
});

test("validateAiAccountConnection explains master node login happens on the bound device", async () => {
  await setup();

  const result = await validateAiAccountConnection("master-codex-primary");

  assert.equal(result.ok, true);
  assert.equal(result.status, "ready");
  assert.match(result.message, /不在手机里直接登录/);
  assert.match(result.message, /Mac Studio|本机 Codex/);
});

test("validateAiAccountConnection reports degraded when the bound master node device is offline", async () => {
  await setup();
  const state = await readState();
  const device = state.devices.find((item) => item.id === "mac-studio");
  assert.ok(device, "expected default mac-studio seed device");

  const heartbeatResponse = await deviceHeartbeatRoute(
    new NextRequest("http://127.0.0.1:3000/api/device-heartbeat", {
      method: "POST",
      headers: { "content-type": "application/json" },
      body: JSON.stringify({
        deviceId: device!.id,
        token: device!.token,
        name: device!.name,
        avatar: device!.avatar,
        account: device!.account,
        status: "offline",
        quota5h: device!.quota5h,
        quota7d: device!.quota7d,
        projects: device!.projects,
        endpoint: device!.endpoint,
      }),
    }),
  );
  assert.equal(heartbeatResponse.status, 200);

  const result = await validateAiAccountConnection("master-codex-primary");

  assert.equal(result.ok, false);
  assert.equal(result.status, "degraded");
  assert.match(result.message, /当前不在线|不在线/);
});

test("validateAiAccountConnection probes aliyun qwen backup accounts through the compatible endpoint", async () => {
  await setup();

  await saveAiAccount({
    accountId: "aliyun-qwen-backup",
    label: "阿里备用",
    role: "backup",
    provider: "aliyun_qwen_api",
    displayName: "阿里百炼备用账号",
    accountIdentifier: "dashscope-demo",
    model: "qwen3.5-plus",
    apiKey: "sk-aliyun-demo-123456",
    enabled: true,
  });

  const originalFetch = globalThis.fetch;
  globalThis.fetch = (async (input) => {
    if (typeof input === "string" && input === "https://dashscope.aliyuncs.com/compatible-mode/v1/responses") {
      return new Response(JSON.stringify({ output_text: "连接正常" }), {
        status: 200,
        headers: {
          "content-type": "application/json",
          "x-request-id": "req-aliyun-validate",
        },
      });
    }
    throw new Error(`unexpected fetch: ${String(input)}`);
  }) as typeof fetch;

  try {
    const result = await validateAiAccountConnection("aliyun-qwen-backup");
    assert.equal(result.ok, true);
    assert.equal(result.status, "ready");
    assert.match(result.message, /连接正常/);
  } finally {
    globalThis.fetch = originalFetch;
  }
});