feat(canvas): AI 拆图 - 完善规划、图层编辑与素材复用

web/test/canvas-image-layer-strategy.test.ts

@@ -0,0 +1,87 @@
+import { expect, test } from "bun:test";
+import { imageLayerBackgroundPatch, imageLayerExtractionTargets, imageLayerModelCalls, resolveImageLayerExtractions, validateImageLayerRegion, type ImageLayerExtraction } from "@/lib/canvas/canvas-image-layer-strategy";
+import { parseImageLayerPlan } from "@/lib/canvas/canvas-image-layer-plan";
+
+const region = { bbox: [100, 200, 600, 800] as [number, number, number, number], shape: "rect" as const };
+const methods: ImageLayerExtraction[] = [{ method: "source" }, { method: "source-region", region }];
+
+test("原图与模型混合按结构处理，原图全链路零图片调用", () => {
+    expect(resolveImageLayerExtractions(2, methods, [false, false])).toEqual(methods);
+    expect(imageLayerModelCalls(methods, true)).toBe(0);
+    expect(imageLayerModelCalls([{ method: "generate" }, methods[1], { method: "generate" }], true)).toBe(3);
+    expect(resolveImageLayerExtractions(2)).toEqual([{ method: "generate" }, { method: "generate" }]);
+});
+
+test("区域、角色、数量不合法时拒绝，显式缺失策略不能偷偷改为付费生成", () => {
+    expect(() => resolveImageLayerExtractions(2, methods, [false, true])).toThrow("不能同时移除");
+    expect(() => resolveImageLayerExtractions(2, [methods[1], methods[0]])).toThrow("第 1 层");
+    expect(() => resolveImageLayerExtractions(2, [methods[0], undefined as any])).toThrow("缺失");
+    expect(() => resolveImageLayerExtractions(2, [methods[0]])).toThrow("数量");
+    for (const bbox of [
+        [0, 0, 1001, 1000],
+        [400, 0, 300, 500],
+        [0, 0, NaN, 500],
+        [0, 0, 1, 100],
+    ])
+        expect(() => validateImageLayerRegion({ ...region, bbox: bbox as any })).toThrow("区域无效");
+    expect(() => validateImageLayerRegion({ ...region, shape: "rounded-rect", radiusRatio: 1 })).toThrow("圆角");
+});
+
+test("只有所有待移除层均有原图区域时才限制背景生成范围", () => {
+    const generated: ImageLayerExtraction[] = [{ method: "generate" }, methods[1], { method: "generate" }];
+    expect(imageLayerBackgroundPatch({ dataUrl: "original" }, generated, [false, true, false])?.regions).toEqual([region]);
+    expect(imageLayerBackgroundPatch({ dataUrl: "original" }, generated, [false, true, true])).toBeUndefined();
+    expect(imageLayerBackgroundPatch({ dataUrl: "original" }, methods, [false, false])).toBeUndefined();
+});
+
+test("编辑区域后背景提示与本地实际提取使用同一坐标", () => {
+    const targets = imageLayerExtractionTargets(["底图", "面板：全部内容，区域 <bbox>0 0 100 100</bbox>"], methods);
+    expect(targets[1]).toBe("面板：全部内容，区域 <bbox>100 200 600 800</bbox>");
+    expect(targets[1].match(/<bbox>/g)).toHaveLength(1);
+});
+
+test("规划声明结构化原图策略，像素坐标按实际参考尺寸换算", () => {
+    const plan = parseImageLayerPlan(
+        JSON.stringify({
+            canPlan: true,
+            coordinateSpace: "pixels",
+            imageSize: { width: 1000, height: 500 },
+            layers: [
+                { name: "完整底图", description: "保留全部内容", kind: "background", generation: { prompt: "保留完整原图", background: "opaque" }, extraction: { method: "source" } },
+                {
+                    name: "规则区域",
+                    description: "完整独立区域",
+                    kind: "object",
+                    generation: { prompt: "复制完整规则区域", background: "transparent" },
+                    removeFromBackground: false,
+                    bbox: [100, 100, 600, 400],
+                    extraction: { method: "source-region", shape: "rounded-rect", radiusRatio: 0.05 },
+                },
+            ],
+        }),
+        { width: 1000, height: 500 },
+    );
+    expect(plan.layers[1].extraction?.region).toEqual({ bbox: [100, 200, 600, 800], shape: "rounded-rect", radiusRatio: 0.05 });
+    expect(
+        resolveImageLayerExtractions(
+            2,
+            plan.layers.map((layer) => layer.extraction!),
+            [false, false],
+        ),
+    ).toHaveLength(2);
+});
+
+test("无效可选坐标保留语义，但显式原图策略阻断付费回退", () => {
+    const plan = parseImageLayerPlan(
+        JSON.stringify({
+            canPlan: true,
+            layers: [
+                { name: "底图", description: "完整底图", kind: "background", generation: { prompt: "保留环境并修补遮挡", background: "opaque" } },
+                { name: "图形", description: "独立图形", kind: "object", generation: { prompt: "复制独立图形", background: "transparent" }, bbox: [0, 0, 1001, 500], extraction: { method: "source-region", shape: "rect" } },
+            ],
+        }),
+    );
+    expect(plan.layers).toHaveLength(2);
+    expect(plan.layers[1].extraction?.method).toBe("source-region");
+    expect(() => resolveImageLayerExtractions(2, [{ method: "generate" }, plan.layers[1].extraction!])).toThrow("区域无效");
+});