import { describe, expect, mock, test } from "bun:test";

import { createMigratedTestDatabase } from "../../helpers";

let generateTextImpl: (_opts: unknown) => unknown = () => ({});

void mock.module("ai", () => ({
  createProviderRegistry: (providers: Record<string, { languageModel: (modelId: string) => unknown }>) => ({
    languageModel: (id: string) => {
      const [providerId, modelId] = id.split(":");
      const provider = providers[providerId ?? ""];
      if (!provider || !modelId) throw new Error(`No such provider: ${id}`);
      return provider.languageModel(modelId);
    },
  }),
  generateText: mock((opts: unknown) => generateTextImpl(opts)),
}));

describe("AI registry", () => {
  test("testProviderConnection rejects invalid config", async () => {
    generateTextImpl = () => {
      throw new Error("Connection failed");
    };

    const { testProviderConnection } = await import("../../../src/server/ai/registry");

    const result = await testProviderConnection({
      apiKey: "bad-key",
      baseUrl: "https://0.0.0.0:1",
      name: "Bad",
      type: "openai-compatible",
    });

    expect(result.ok).toBe(false);
    expect(result.message).toContain("连接失败");
    expect(typeof result.message).toBe("string");
  });

  test("testProviderConnection return shape is correct", async () => {
    generateTextImpl = () => ({});

    const { testProviderConnection } = await import("../../../src/server/ai/registry");

    const result = await testProviderConnection({
      apiKey: "sk-test",
      baseUrl: "https://api.openai.com/v1",
      name: "Test",
      type: "openai",
    });

    expect(result.ok).toBe(true);
    expect(result.message).toBe("连接成功");
  });

  test("buildProviderRegistry 从 DB 构建包含启用供应商的注册表", async () => {
    const handle = createMigratedTestDatabase("registry-build-test");
    const now = new Date().toISOString();

    handle.db
      .prepare(
        "INSERT INTO providers (id, name, type, base_url, api_key, enabled, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
      )
      .run("pv1", "OpenAI", "openai", "https://api.openai.com/v1", "sk-test", 1, now, now);
    handle.db
      .prepare(
        "INSERT INTO providers (id, name, type, base_url, api_key, enabled, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?, ?)",
      )
      .run("pv2", "Disabled", "anthropic", "https://api.anthropic.com", "sk-off", 0, now, now);

    const { buildProviderRegistry } = await import("../../../src/server/ai/registry");
    const registry = buildProviderRegistry(handle.db);

    expect(() => registry.languageModel("pv1:gpt-4o")).not.toThrow();
    expect(() => registry.languageModel("pv2:claude-3")).toThrow();

    handle.cleanup();
  });

  test("buildProviderRegistry 无启用供应商时返回空注册表", async () => {
    const handle = createMigratedTestDatabase("registry-empty-test");

    const { buildProviderRegistry } = await import("../../../src/server/ai/registry");
    const registry = buildProviderRegistry(handle.db);

    expect(typeof registry.languageModel).toBe("function");

    handle.cleanup();
  });
});