import { expect, test } from "bun:test";
import { groupModelsForPicker, modelChannelLabel } from "../src/lib/model-picker-groups";
import { modelCompatibilityError, resolveCompatibleModel } from "../src/lib/model-selection";
import { modelQuoteRequest, priceTiersForCurrentSelection, priceTierSummaryLabel } from "../src/lib/model-pricing";
import { systemChannelModelChannels } from "../src/lib/user-session";
import type { PublicChannelCatalog, PublicChannelModel } from "../src/services/api/logical-models";
import { defaultConfig, normalizeConfigSnapshot, resolveModelRequestConfig, selectableModelsByCapability } from "../src/stores/use-config-store";
import { defaultModelCapabilityConfig } from "../src/lib/model-capabilities";

function model(label = "", price = 300000): PublicChannelModel {
    return {
        id: label || "test-model", modelKey: "seedance-2.0", displayName: "Seedance 2.0", channelLabel: label,
        description: label ? `${label}的使用说明` : "",
        icon: "ByteDance", capability: "video", protocol: "seedance", available: true,
        pricingMode: "provider", priceLabel: "",
        capabilityConfig: defaultModelCapabilityConfig("seedance", "seedance-2.0"),
        priceTiers: [{ id: "tier", selector: {}, resolution: "*", videoSeconds: 0, billingMode: "per_second", unitPriceMicrocredits: price, inputTokenPriceMicrocredits: 0, outputTokenPriceMicrocredits: 0, cachedTokenPriceMicrocredits: 0 }],
    };
}

function fixture() {
    const channels: PublicChannelCatalog[] = [
        { id: "a", name: "正常渠道", displayName: "正常渠道", models: [model()] },
        { id: "b", name: "渠道 B", displayName: "渠道 B", models: [model("优惠渠道-993")] },
        { id: "c", name: "渠道 C", displayName: "渠道 C", models: [model("特惠渠道-730", 200000)] },
    ];
    return normalizeConfigSnapshot({ config: { ...defaultConfig, channels: systemChannelModelChannels(channels), model: "b::seedance-2.0", videoModel: "b::seedance-2.0" } }).config;
}

test("one product exposes every channel and preserves prices through catalog normalization", () => {
    const config = fixture();
    const groups = groupModelsForPicker(config, selectableModelsByCapability(config, "video"));
    expect(groups).toHaveLength(1);
    expect(config.channels.map((channel) => channel.modelCosts![0].description)).toEqual(["", "优惠渠道-993的使用说明", "特惠渠道-730的使用说明"]);
    expect(groups[0]).toMatchObject({ label: "Seedance 2.0", icon: "ByteDance", kind: "product" });
    expect(groups[0].models.map((item) => [item.label, item.models])).toEqual([
        ["正常渠道", ["a::seedance-2.0"]], ["优惠渠道-993", ["b::seedance-2.0"]], ["特惠渠道-730", ["c::seedance-2.0"]],
    ]);
    expect(config.channels.map((channel) => priceTierSummaryLabel(priceTiersForCurrentSelection(channel.modelCosts![0].logicalPriceTiers!, "video", config)))).toEqual(["0.3 积分/秒", "0.3 积分/秒", "0.2 积分/秒"]);
});

test("selection and quote keep the chosen channel even when another channel is cheaper", () => {
    const config = fixture();
    const value = "b::seedance-2.0";
    expect(resolveCompatibleModel(config, value, { capability: "video" })).toBe(value);
    expect(resolveModelRequestConfig(config, value)).toMatchObject({ channelId: "b", model: "seedance-2.0" });
    expect(modelQuoteRequest(config, value, "video")).toMatchObject({ channelId: "b", modelKey: "seedance-2.0" });
    config.channels[1].modelCosts![0].channelLabel = "新展示名";
    expect(modelChannelLabel(config, value)).toBe("新展示名");
    expect(resolveModelRequestConfig(config, value)).toMatchObject({ channelId: "b", model: "seedance-2.0" });
});

test("product keys and capabilities stay distinct despite equal display names", () => {
    const config = fixture();
    config.channels[1].models = ["seedance-2.0-fast"];
    config.channels[1].modelCosts![0].model = "seedance-2.0-fast";
    config.channels[2].modelCosts![0].capability = "image";
    const groups = groupModelsForPicker(config, ["a::seedance-2.0", "b::seedance-2.0-fast", "c::seedance-2.0"]);
    expect(groups).toHaveLength(3);
});

test("group metadata is stable across channel reorder and current selection", () => {
    const config = fixture();
    config.channels[1].modelCosts![0].displayName = "不同展示名";
    config.channels[1].modelCosts![0].icon = "Jimeng";
    const options = selectableModelsByCapability(config, "video");
    const first = groupModelsForPicker(config, options)[0];
    const reordered = groupModelsForPicker({ ...config, channels: [...config.channels].reverse(), model: "c::seedance-2.0" }, options)[0];
    expect([reordered.label, reordered.icon]).toEqual([first.label, first.icon]);
    expect(reordered.models[0].models).toEqual(["c::seedance-2.0"]);
});

test("each channel keeps its own capability restrictions", () => {
    const config = fixture();
    config.channels[0].modelCosts![0].capabilityConfig!.video!.references.maxImages = 0;
    config.channels[1].modelCosts![0].capabilityConfig!.video!.references.maxImages = 2;
    config.channels[1].modelCosts![0].capabilityConfig!.video!.operations = ["image_to_video"];
    const requirements = { capability: "video" as const, input: { textCount: 1, imageCount: 1, videoCount: 0, audioCount: 0, characterCount: 0 } };
    expect(modelCompatibilityError(config, "a::seedance-2.0", requirements)).not.toBe("");
    expect(modelCompatibilityError(config, "b::seedance-2.0", requirements)).toBe("");
    expect(groupModelsForPicker(config, selectableModelsByCapability(config, "video"))[0].models).toHaveLength(3);
});

test("managed logical models and personal channels never become system product routes", () => {
    const config = fixture();
    config.channels[1].id = "managed";
    config.channels[1].modelCosts![0].logicalModelId = "logical-test";
    config.channels[2].scope = "user";
    const groups = groupModelsForPicker(config, ["a::seedance-2.0", "managed::seedance-2.0", "c::seedance-2.0"]);
    expect(groups.map((item) => item.kind)).toEqual(["product", "channel", "channel"]);
    expect(groups[0].models).toHaveLength(1);
    expect(groupModelsForPicker(config, [])).toEqual([]);
});