import { describe, expect, it } from "bun:test";
import { agentOperationCategory, agentOperationFailed, agentOperationLabel, agentOperationSegmentLabel, buildAgentFeedSegments, isAgentCarrierRecord, isAgentOperationRecord } from "@/lib/canvas/agent-operation-feed";

type Row = { id: string; role: string; title?: string; text: string; detail?: unknown; planItems?: { id: string }[]; question?: unknown };

const step = (id: string, title: string, text: string, detail: unknown = { eventType: "tool_completed" }): Row => ({ id, role: "tool", title, text, detail });
const kinds = (rows: Row[]) => buildAgentFeedSegments(rows).map((segment) => segment.kind);
const idsOf = (segment: ReturnType<typeof buildAgentFeedSegments<Row>>[number]) => (segment.kind === "operations" ? segment.items.map((item) => item.id) : [segment.item.id]);

describe("Agent operation feed", () => {
    it("folds consecutive tool records into one segment keyed by the first step", () => {
        const rows: Row[] = [
            { id: "u1", role: "user", text: "读一下画布" },
            step("t1", "canvas_get_state", "已读取当前画布"),
            step("t2", "model_list", "已获取可用模型"),
            { id: "a1", role: "assistant", text: "画布里有 1 个节点" },
            step("t3", "canvas_apply_ops", "画布内容已保存至服务端", { eventType: "canvas_updated" }),
        ];
        const segments = buildAgentFeedSegments(rows);
        expect(kinds(rows)).toEqual(["message", "operations", "message", "operations"]);
        expect(segments[1]).toMatchObject({ kind: "operations", key: "t1" });
        expect(idsOf(segments[1])).toEqual(["t1", "t2"]);
    });

    it("appends later steps to the same segment so its key and expanded state survive", () => {
        const before: Row[] = [step("t1", "canvas_get_state", "已读取当前画布"), step("t2", "model_list", "已获取可用模型")];
        const after = buildAgentFeedSegments([...before, step("t3", "task_get", "已查询任务状态")]);
        expect(after).toHaveLength(1);
        expect(after[0]).toMatchObject({ kind: "operations", key: "t1" });
        expect(idsOf(after[0])).toEqual(["t1", "t2", "t3"]);
    });

    it("keeps the approval card and unrelated roles out of the folded group", () => {
        const pending: Row = { id: "p1", role: "tool", title: "canvas_apply_ops", text: "准备更新画布内容", detail: { status: "pending" } };
        const rows: Row[] = [step("t1", "canvas_get_state", "已读取当前画布"), pending, step("t2", "model_list", "已获取可用模型")];
        expect(kinds(rows)).toEqual(["operations", "message", "operations"]);
        expect(isAgentOperationRecord(pending)).toBe(false);
        expect(isAgentOperationRecord(rows[0])).toBe(true);
    });

    it("drops plan and question carriers that carry no body of their own", () => {
        const plan: Row = { id: "plan-run1", role: "tool", text: "", planItems: [{ id: "a" }] };
        const question: Row = { id: "question-run1", role: "assistant", text: "", question: { question: "选哪个？" } };
        const answered: Row = { id: "a1", role: "assistant", text: "那就用第一个方案", question: { question: "选哪个？" } };
        expect(buildAgentFeedSegments([plan, question, answered]).map((segment) => segment.key)).toEqual(["a1"]);
        expect(isAgentCarrierRecord(plan)).toBe(true);
        expect(isAgentCarrierRecord(question)).toBe(true);
        expect(isAgentCarrierRecord(answered)).toBe(false);
    });

    it("reports the latest step, including an in-flight and a failed one", () => {
        expect(agentOperationLabel(step("t1", "canvas_get_state", "工具执行成功"))).toBe("已读取画布清单（未查看画面）");
        expect(agentOperationLabel(step("t2", "canvas_apply_ops", "", {}))).toBe("准备更新画布内容");
        expect(agentOperationLabel(step("t3", "canvas_apply_ops", "", { eventType: "tool_failed" }))).toBe("更新画布内容失败");
        expect(agentOperationFailed(step("t3", "canvas_apply_ops", "", { eventType: "tool_failed" }))).toBe(true);
        expect(agentOperationFailed(step("t1", "canvas_get_state", "工具执行成功"))).toBe(false);
    });

    it("reports the retry group instead of a stale completion", () => {
        const retrying = step("t1", "canvas_apply_ops", "缺少操作类型", { eventType: "tool_failed", retry: { groupId: "run:repair:call", attempt: 2, maxAttempts: 3, status: "retrying" } });
        expect(agentOperationLabel(retrying)).toBe("自动纠正记录 · 2/3 次尝试");
        const recovered = step("t1", "canvas_apply_ops", "缺少操作类型", { eventType: "tool_failed", retry: { groupId: "run:repair:call", attempt: 2, maxAttempts: 3, status: "recovered" } });
        expect(agentOperationLabel(recovered)).toBe("自动纠正后已恢复 · 2/3 次尝试");
    });
});

describe("Agent operation segment label", () => {
    const viewed = (id: string, title: string, extra: Record<string, unknown> = {}) => step(id, "canvas_inspect_image", "工具执行成功", { eventType: "tool_completed", result: { nodeId: id, title, ...extra } });

    it("aggregates a vision run by the images that really went out", () => {
        const items = [viewed("n1", "剧照1.png"), viewed("n2", "剧照2.png"), viewed("n3", "封面.png")];
        expect(agentOperationSegmentLabel(items)).toBe("查看了 3 张画面 · 最新《封面.png》");
        expect(agentOperationCategory(items[0])).toBe("vision");
    });

    it("counts a batch receipt once per image instead of once per call", () => {
        const items = [
            viewed("n1", "剧照1.png"),
            step("last", "canvas_inspect_image", "工具执行成功", {
                eventType: "tool_completed",
                result: {
                    nodeId: "n3",
                    title: "剧照3.png",
                    batchImages: [
                        { nodeId: "n1", title: "剧照1.png" },
                        { nodeId: "n2", title: "剧照2.png" },
                        { nodeId: "n3", title: "剧照3.png" },
                    ],
                },
            }),
        ];
        expect(agentOperationSegmentLabel(items)).toBe("查看了 3 张画面 · 最新《剧照3.png》");
    });

    it("does not count a reused observation or a read-loop receipt as a delivered picture", () => {
        const items = [viewed("n1", "剧照1.png", { repeat: true, reuseObservation: true }), viewed("n2", "剧照2.png", { repeat: true, refreshIgnored: true })];
        // 一张画面都没送出去：退回最新一步自己的摘要，绝不报"查看了 N 张画面"
        expect(agentOperationSegmentLabel(items)).toBe("本轮重复查看，只回执文字、未附图");
        expect(agentOperationSegmentLabel(items)).not.toContain("查看了");
    });

    it("keeps the newest step's own copy when the segment is not about pictures", () => {
        const items = [step("t1", "canvas_get_state", "工具执行成功"), step("t2", "canvas_apply_ops", "工具执行成功", { eventType: "canvas_updated" })];
        expect(agentOperationSegmentLabel(items)).toBe("画布内容已保存至服务端");
        expect(agentOperationCategory(items[1])).toBe("operate");
        expect(agentOperationSegmentLabel([])).toBe("");
    });
});