/// <reference path="../node_modules/@types/jest/index.d.ts" />
// @ts-check

const { start } = require("../utils");
const lg = require("../utils/litegraph");

beforeEach(() => {
	lg.setup(global);
});

afterEach(() => {
	lg.teardown(global);
	jest.resetModules();
});

test("converted widget works after reload", async () => {
	const { graph, ez } = await start();
	let { $: n } = ez.CheckpointLoaderSimple();

	const inputCount = n.inputs.length;

	// Convert ckpt name to an input
	n.widgets.ckpt_name.convertToInput();
	expect(n.widgets.ckpt_name.isConvertedToInput).toBeTruthy();
	expect(n.inputs.ckpt_name).toBeTruthy();
	expect(n.inputs.length).toEqual(inputCount + 1);

	// Convert back to widget and ensure input is removed
	n.widgets.ckpt_name.convertToWidget();
	expect(n.widgets.ckpt_name.isConvertedToInput).toBeFalsy();
	expect(n.inputs.ckpt_name).toBeFalsy();
	expect(n.inputs.length).toEqual(inputCount);

	// Convert again and reload the graph to ensure it maintains state
	n.widgets.ckpt_name.convertToInput();
	expect(n.inputs.length).toEqual(inputCount + 1);

	let { $: primitive } = ez.PrimitiveNode();
	primitive.outputs[0].connectTo(n.inputs.ckpt_name);

	await graph.reload();

	// Find the reloaded nodes in the graph
	n = graph.find(n);
	primitive = graph.find(primitive);

	// Ensure widget is converted
	expect(n.widgets.ckpt_name.isConvertedToInput).toBeTruthy();
	expect(n.inputs.ckpt_name).toBeTruthy();
	expect(n.inputs.length).toEqual(inputCount + 1);

	// Ensure primitive is connected
	let { connections } = primitive.outputs[0];
	expect(connections).toHaveLength(1);
	expect(connections[0].targetNode.id).toBe(n.node.id);

	// Disconnect & reconnect
	connections[0].disconnect();
	({ connections } = primitive.outputs[0]);
	expect(connections).toHaveLength(0);
	primitive.outputs[0].connectTo(n.inputs.ckpt_name);

	({ connections } = primitive.outputs[0]);
	expect(connections).toHaveLength(1);
	expect(connections[0].targetNode.id).toBe(n.node.id);

	// Convert back to widget and ensure input is removed
	n.widgets.ckpt_name.convertToWidget();
	expect(n.widgets.ckpt_name.isConvertedToInput).toBeFalsy();
	expect(n.inputs.ckpt_name).toBeFalsy();
	expect(n.inputs.length).toEqual(inputCount);
});

test("converted widget works on clone", async () => {
	const { graph, ez } = await start();
	let { $: n } = ez.CheckpointLoaderSimple();

	// Convert the widget to an input
	n.widgets.ckpt_name.convertToInput();
	expect(n.widgets.ckpt_name.isConvertedToInput).toBeTruthy();

	// Clone the node
	n.menu["Clone"].call();
	expect(graph.nodes).toHaveLength(2);
	const clone = graph.nodes[1];
	expect(clone.id).not.toEqual(n.id);

	// Ensure the clone has an input
	expect(clone.widgets.ckpt_name.isConvertedToInput).toBeTruthy();
	expect(clone.inputs.ckpt_name).toBeTruthy();

	// Ensure primitive connects to both nodes
	let { $: primitive } = ez.PrimitiveNode();
	primitive.outputs[0].connectTo(n.inputs.ckpt_name);
	primitive.outputs[0].connectTo(clone.inputs.ckpt_name);
	expect(primitive.outputs[0].connections).toHaveLength(2);

	// Convert back to widget and ensure input is removed
	clone.widgets.ckpt_name.convertToWidget();
	expect(clone.widgets.ckpt_name.isConvertedToInput).toBeFalsy();
	expect(clone.inputs.ckpt_name).toBeFalsy();
});