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

import {
    buildImageResolutionOptions,
    formatImageResolutionSize,
    imageResolutionChoices,
    imageResolutionOption,
    imageSizeForResolution,
    supportsImageResolutionPresets,
} from "../src/lib/image-resolution-tiers";
import { defaultImageCapabilityConfig, normalizeImageValue } from "../src/lib/model-capabilities";
import { resolveImageRequestSize } from "../src/services/api/image-validation";

const sizes = [
    "1024x1024", "1360x1024", "1024x1360", "1536x1024", "1024x1536", "1024x1280", "1280x1024", "2048x878", "1824x1024", "1024x1824",
    "2048x2048", "2304x1728", "1728x2304", "2496x1664", "1664x2496", "1792x2240", "2240x1792", "3136x1344", "2752x1536", "1536x2752",
    "2880x2880", "3264x2448", "2448x3264", "3504x2336", "2336x3504", "2560x3200", "3200x2560", "3808x1632", "3840x2160", "2160x3840",
];

describe("image resolution tiers", () => {
    test("精确像素预设保持请求原值，比例协议发送比例", () => {
        const profile = defaultImageCapabilityConfig("openai-image", "test");
        profile.size = { parameter: "size", values: ["1920x1080", "3840x2160", "2160x3840", "1824x1024"], default: "1920x1080", allowCustom: false };
        expect(imageResolutionChoices(profile.size.values)).toEqual(["1k", "2k", "4k"]);
        expect(imageSizeForResolution(buildImageResolutionOptions(profile.size.values), "1k", "16:9")).toBe("1824x1024");
        expect(resolveImageRequestSize(profile, undefined, "1920x1080")).toEqual({ parameter: "size", value: "1920x1080" });
        profile.size = { parameter: "aspect_ratio", values: ["16:9"], default: "16:9", allowCustom: false };
        expect(resolveImageRequestSize(profile, undefined, "16:9")).toEqual({ parameter: "aspect_ratio", value: "16:9" });
        profile.size = { parameter: "size", values: [], default: "auto", allowCustom: true };
        expect(() => resolveImageRequestSize(profile, undefined, "1920x1080")).toThrow("16 的倍数");
    });
    test("将 Xiaobaishu 的精确尺寸整理为 1K、2K、4K 各十种比例", () => {
        const options = buildImageResolutionOptions(sizes);

        expect(options.filter((item) => item.tier === "1k")).toHaveLength(10);
        expect(options.filter((item) => item.tier === "2k")).toHaveLength(10);
        expect(options.filter((item) => item.tier === "4k")).toHaveLength(10);
        expect(imageResolutionOption(options, "1536x2752")).toMatchObject({ tier: "2k", ratio: "9:16" });
    });

    test("切换档位时按相同比例还原精确像素并显示比例与档位", () => {
        const options = buildImageResolutionOptions(sizes);

        expect(imageSizeForResolution(options, "4k", "9:16")).toBe("2160x3840");
        expect(formatImageResolutionSize("1536x2752", options)).toBe("9:16 · 2K");
    });

    test("保留自动尺寸并在摘要中显示中文标签", () => {
        const values = ["auto", ...sizes];

        expect(imageResolutionChoices(values)).toEqual(["auto", "1k", "2k", "4k"]);
        expect(buildImageResolutionOptions(values)).toHaveLength(30);
        expect(formatImageResolutionSize("auto", [])).toBe("自动");
    });

    test("允许自定义尺寸时仍保留自动与分辨率预设", () => {
        expect(supportsImageResolutionPresets({ parameter: "size", values: ["auto", ...sizes], allowCustom: true })).toBe(true);
    });

    test("默认图片能力为 OpenAI 兼容模型提供完整 1K、2K、4K 档位", () => {
        const profile = defaultImageCapabilityConfig("openai-image", "gpt-image-2-4K");

        expect(supportsImageResolutionPresets(profile.size)).toBe(true);
        expect(imageResolutionChoices(profile.size.values)).toEqual(["auto", "1k", "2k", "4k"]);
        expect(buildImageResolutionOptions(profile.size.values)).toHaveLength(30);
    });

    // Agnes 图像的 size 是官方必填项，但取分辨率档位，画面比例走独立的 ratio 字段，
    // 所以界面必须同时保留比例和档位两个选择器，不能退化成 OpenAI 的像素尺寸。
    test("Agnes 图像按比例加档位选择尺寸，并关闭上游不支持的输出开关", () => {
        const profile = defaultImageCapabilityConfig("agnes-image", "agnes-image-2.1-flash");

        expect(profile.size.parameter).toBe("aspect_ratio");
        expect(profile.size.values).toEqual(["1:1", "3:4", "4:3", "16:9", "9:16", "2:3", "3:2", "21:9"]);
        expect(profile.quality.values).toEqual(["1k", "2k", "4k"]);
        expect(normalizeImageValue(profile, { size: "3:4", quality: "4k", count: "4" })).toMatchObject({ size: "3:4", quality: "4k", count: "1" });

        // 顶层 response_format 是官方明确列出的错误写法，output_format 会被上游直接拒绝。
        expect(profile.responseFormat.supported).toBe(false);
        expect(profile.outputFormat.supported).toBe(false);
        expect(profile.transparentBackground.supported).toBe(false);
        expect(profile.references.maskSupported).toBe(false);
    });

    test("识别适合全景图的 2:1 自定义尺寸", () => {
        const options = buildImageResolutionOptions(["1440x720", "2048x1024", "3840x1920"]);

        expect(options.map((item) => item.tier)).toEqual(["1k", "2k", "4k"]);
        expect(imageResolutionOption(options, "3840x1920")).toMatchObject({ tier: "4k", ratio: "2:1" });
    });
});