#!/usr/bin/env node

import assert from "node:assert/strict";
import {readFile} from "node:fs/promises";
import {
    LORA_ROUTE_LETTERS,
    availableLoRARoutes,
    connectedLoRARoutes,
    loraInputRoute,
    loraRouteLabel,
    nextLoRARoute,
} from "../web/h3_lora_scheduler_core.mjs";

const plan = {id: 1, type: "MiniMaxH3ChainPlan", inputs: []};
const current = {
    id: 2,
    type: "MiniMaxH3ChainCurrentShot",
    inputs: [{name: "state", link: 11}],
};
const scheduler = {
    id: 3,
    type: "MiniMaxH3ChainLoRAScheduler",
    inputs: [
        {name: "state", link: 12},
        {name: "base_model", link: 13},
        {name: "lora_a", link: 14},
        {name: "lora_b", link: null},
        {name: "lora_d", link: 15},
    ],
};
const unrelatedPlan = {id: 4, type: "MiniMaxH3ChainPlan", inputs: []};
const nodes = [plan, current, scheduler, unrelatedPlan];
const graph = {
    _nodes: nodes,
    links: {
        11: {origin_id: 1},
        12: {origin_id: 2},
        13: {origin_id: 20},
        14: {origin_id: 21},
        15: {origin_id: 22},
    },
    getNodeById(id) { return nodes.find((node) => node.id === id) ?? null; },
};
for (const node of nodes) node.graph = graph;

assert.equal(LORA_ROUTE_LETTERS.length, 26);
assert.equal(loraInputRoute("lora_z"), "z");
assert.equal(loraInputRoute("base_model"), null);
assert.deepEqual(connectedLoRARoutes(scheduler), ["a", "d"]);
assert.equal(nextLoRARoute(scheduler), "b");
assert.equal(loraRouteLabel("base"), "Base model");
assert.equal(loraRouteLabel("z"), "LoRA Z");
assert.deepEqual(availableLoRARoutes(graph, [plan]), ["base", "a", "d"]);
assert.deepEqual(
    availableLoRARoutes(graph, [plan], "z"),
    ["base", "a", "d", "z"],
);
assert.deepEqual(
    availableLoRARoutes(graph, [unrelatedPlan]),
    ["base"],
);

const frontend = await readFile(
    new URL("../web/h3_chain_lora_scheduler.js", import.meta.url), "utf8",
);
assert.match(frontend, /function stabilizeRouteInputs\(node\)/);
assert.match(frontend, /node\.removeInput\(index\)/);
assert.match(frontend, /node\.addInput\(`lora_\$\{next\}`, "MODEL"\)/);
assert.match(frontend, /Connect next LoRA route/);
assert.match(frontend, /h3-lora-routes-changed/);
assert.match(frontend, /onConnectionsChange/);
assert.match(frontend, /onGraphConfigured/);

console.log("H3 dynamic LoRA scheduler: progressive A-Z inputs and Plan route discovery pass");