import assert from "node:assert/strict";
import test from "node:test";

import { createModelAgentContentPreparer } from "../src/ui/lib/modelImageReferences.ts";

test("replaces only the submitted URL for named SVG references", async () => {
  const calls = { fetch: 0, rasterize: 0, upload: 0 };
  const prepare = createModelAgentContentPreparer({
    fetchImage: async () => {
      calls.fetch += 1;
      return new Response('<svg viewBox="0 0 10 10"></svg>', { headers: { "content-type": "image/svg+xml" } });
    },
    rasterizeSvg: async (_source, fileName) => {
      calls.rasterize += 1;
      assert.equal(fileName, "drawing (DXF).svg");
      return { name: "drawing (DXF).model.webp", type: "image/webp" };
    },
    uploadImage: async (file) => {
      calls.upload += 1;
      assert.equal(file.type, "image/webp");
      return "https://assets.example.com/model/drawing.webp";
    }
  });
  const contents = [
    { type: "image", imageUrl: "https://assets.example.com/source/opaque-id", imageName: "drawing (DXF).svg" },
    { type: "text", text: "Use this plan", textSource: "input" }
  ];

  const prepared = await prepare(contents);

  assert.equal(prepared[0].imageUrl, "https://assets.example.com/model/drawing.webp");
  assert.equal(prepared[0].imageName, "drawing (DXF).svg");
  assert.equal(contents[0].imageUrl, "https://assets.example.com/source/opaque-id");
  assert.deepEqual(calls, { fetch: 1, rasterize: 1, upload: 1 });
});

test("keeps supported raster references unchanged without downloading them", async () => {
  let called = false;
  const prepare = createModelAgentContentPreparer({
    fetchImage: async () => {
      called = true;
      throw new Error("unexpected fetch");
    },
    rasterizeSvg: async () => {
      throw new Error("unexpected rasterization");
    },
    uploadImage: async () => {
      throw new Error("unexpected upload");
    }
  });
  const contents = [{ type: "image", imageUrl: "https://assets.example.com/reference.webp", imageName: "reference.webp" }];

  const prepared = await prepare(contents);

  assert.strictEqual(prepared[0], contents[0]);
  assert.equal(called, false);
});

test("detects SVG references from response MIME when names have no extension", async () => {
  let rasterized = 0;
  const prepare = createModelAgentContentPreparer({
    fetchImage: async () => new Response("<svg></svg>", { headers: { "content-type": "image/svg+xml; charset=utf-8" } }),
    rasterizeSvg: async () => {
      rasterized += 1;
      return { name: "reference.model.webp", type: "image/webp" };
    },
    uploadImage: async () => "https://assets.example.com/reference.model.webp"
  });

  const prepared = await prepare([{ type: "image", imageUrl: "https://assets.example.com/opaque-id" }]);

  assert.equal(prepared[0].imageUrl, "https://assets.example.com/reference.model.webp");
  assert.equal(rasterized, 1);
});

test("caches uploaded WebP derivatives by source URL", async () => {
  let uploads = 0;
  const prepare = createModelAgentContentPreparer({
    fetchImage: async () => new Response("<svg></svg>", { headers: { "content-type": "image/svg+xml" } }),
    rasterizeSvg: async () => ({ name: "reference.model.webp", type: "image/webp" }),
    uploadImage: async () => {
      uploads += 1;
      return "https://assets.example.com/reference.model.webp";
    }
  });
  const contents = [{ type: "image", imageUrl: "https://assets.example.com/reference.svg", imageName: "reference.svg" }];

  await Promise.all([prepare(contents), prepare(contents)]);

  assert.equal(uploads, 1);
});