import { Model } from "../../../src";
import {
  chartToImageFile,
  chartToImageUrl,
} from "../../../src/helpers/figures/charts/chart_ui_common";
import { createChart } from "../../test_helpers";
import { TEST_CHART_DATA } from "../../test_helpers/constants";
import { mockChart } from "../../test_helpers/helpers";

jest.mock("../../../src/helpers/figures/charts/chart_ui_common", () => {
  return {
    ...jest.requireActual("../../../src/helpers/figures/charts/chart_ui_common"),
  };
});

mockChart();

const figureId = "fig1";
const chartId = "chartId";
describe("chart conversion", () => {
  afterEach(() => {
    mockChart();
  });

  test("convert chart to image", async () => {
    const model = new Model();
    const sheetId = model.getters.getActiveSheetId();
    createChart(model, TEST_CHART_DATA.basicChart, chartId, undefined, { figureId });
    const figure = model.getters.getFigure(sheetId, figureId)!;
    const runtime = model.getters.getChartRuntime(chartId);
    const imageUrl = await chartToImageUrl(runtime, figure, "bar");
    // canvas is not drawn at all
    expect(imageUrl).toBe("data:image/png;base64,");

    const imageFile = await chartToImageFile(runtime, figure, "bar");
    expect(imageFile).toBeInstanceOf(Blob);
  });

  test("cannot export Chart.js chart when lib is not loaded", async () => {
    globalThis.Chart = undefined;
    const spy = jest.spyOn(console, "log").mockImplementation(() => {});

    const model = new Model();
    const sheetId = model.getters.getActiveSheetId();
    createChart(model, TEST_CHART_DATA.basicChart, chartId, undefined, { figureId });
    const figure = model.getters.getFigure(sheetId, figureId)!;
    const runtime = model.getters.getChartRuntime(chartId);

    const imageUrl = await chartToImageUrl(runtime, figure, "bar");
    expect(imageUrl).toBeUndefined();
    expect(spy).toHaveBeenCalledWith(
      "Error exporting chart to image URL: Chart.js library is not loaded"
    );

    const imageFile = await chartToImageFile(runtime, figure, "bar");
    expect(imageFile).toBeNull();
    expect(spy).toHaveBeenCalledWith(
      "Error exporting chart to image URL: Chart.js library is not loaded"
    );
  });

  test("export Chart.js chart check type from runtime config, not the given chart type", async () => {
    const model = new Model();
    const sheetId = model.getters.getActiveSheetId();

    createChart(model, TEST_CHART_DATA.combo, chartId, undefined, { figureId });
    const figure = model.getters.getFigure(sheetId, figureId)!;
    const runtime = model.getters.getChartRuntime(chartId) as any;
    expect(runtime.chartJsConfig.type).toBe("bar");

    const registrySpy = jest.spyOn(globalThis.Chart!.registry.controllers, "get");

    const imageUrl = await chartToImageUrl(runtime, figure, "combo");
    expect(imageUrl).toBe("data:image/png;base64,");
    expect(registrySpy).toHaveBeenCalledWith("bar");
    expect(registrySpy).not.toHaveBeenCalledWith("combo");

    const imageFile = await chartToImageFile(runtime, figure, "combo");
    expect(imageFile).toBeInstanceOf(Blob);
    expect(registrySpy).toHaveBeenCalledWith("bar");
    expect(registrySpy).not.toHaveBeenCalledWith("combo");
  });

  test("cannot export if the chart runtime config type is not supported", async () => {
    jest.spyOn(globalThis.Chart!.registry.controllers, "get").mockImplementation((type: string) => {
      return type === "bar" ? {} : (undefined as any);
    });
    const spy = jest.spyOn(console, "log").mockImplementation(() => {});
    const model = new Model();
    const sheetId = model.getters.getActiveSheetId();

    createChart(model, TEST_CHART_DATA.combo, chartId, undefined, { figureId });
    const figure = model.getters.getFigure(sheetId, figureId)!;
    const originalRuntime = model.getters.getChartRuntime(chartId) as any;
    const runtime = {
      ...originalRuntime,
      chartJsConfig: {
        ...originalRuntime.chartJsConfig,
        type: "combo",
      },
    };

    const imageUrl = await chartToImageUrl(runtime, figure, "combo");
    expect(imageUrl).toBeUndefined();
    expect(spy).toHaveBeenCalledWith(
      'Error exporting chart to image URL: Chart of type "combo" is not registered in Chart.js library.'
    );

    const imageFile = await chartToImageFile(runtime, figure, "combo");
    expect(imageFile).toBeNull();
    expect(spy).toHaveBeenCalledWith(
      'Error exporting chart to image URL: Chart of type "combo" is not registered in Chart.js library.'
    );
  });
});