homarr/packages/custom-widgets/src/test/import.spec.ts

import { describe, expect, it } from "vitest";

import {
  CUSTOM_WIDGET_STARTER,
  getImportReview,
  parseCustomWidgetAiResponse,
  parseCustomWidgetClipboardDetailed,
} from "../core";

const response = (manifest: unknown) => `\`\`\`json\n${JSON.stringify(manifest, null, 2)}\n\`\`\``;

describe("Custom Widget imports", () => {
  it("accepts one complete JSON block and loads its template", () => {
    const result = parseCustomWidgetAiResponse(response({ ...CUSTOM_WIDGET_STARTER, template: "<Text>Ready</Text>" }));
    expect(result.success).toBe(true);
    if (result.success) expect(result.widget.template).toBe("<Text>Ready</Text>");
  });

  it("rejects the unreleased superseded schema without normalization", () => {
    const result = parseCustomWidgetAiResponse(
      response({
        ...CUSTOM_WIDGET_STARTER,
        sources: Object.values(CUSTOM_WIDGET_STARTER.sources),
        requests: [],
        optionsSchema: {},
        defaultOptions: {},
        template: "<Text>Ready</Text>",
      }),
    );
    expect(result.success).toBe(false);
  });

  it("accepts JSON copied from a code block and reports removed local state", () => {
    expect(parseCustomWidgetAiResponse(JSON.stringify(CUSTOM_WIDGET_STARTER)).success).toBe(true);
    const result = parseCustomWidgetAiResponse(
      response({ ...CUSTOM_WIDGET_STARTER, stateSchema: {}, template: "<Text>Ready</Text>" }),
    );
    expect(result.success).toBe(false);
    if (!result.success) expect(result.issues.some(({ code }) => code === "REMOVED_LOCAL_STATE")).toBe(true);
  });

  it("rejects additional code blocks", () => {
    const result = parseCustomWidgetAiResponse(
      `${response({ ...CUSTOM_WIDGET_STARTER, template: "<Text>Ready</Text>" })}\n\`\`\`jsx\n<Text>Old format</Text>\n\`\`\``,
    );
    expect(result.success).toBe(false);
    if (!result.success) expect(result.issues.some(({ code }) => code === "AI_SINGLE_JSON_BLOCK_REQUIRED")).toBe(true);
  });

  it("returns a capability review from keyed records", () => {
    const review = getImportReview(CUSTOM_WIDGET_STARTER);
    expect(review).toMatchObject({ origins: ["https://example.com"], authTypes: ["none"], hasActions: false });
  });

  it("keeps strict clipboard import compatible with canonical exports", () => {
    const result = parseCustomWidgetClipboardDetailed(JSON.stringify(CUSTOM_WIDGET_STARTER));
    expect(result.success).toBe(true);
  });
});