#!/usr/bin/env node

import assert from "node:assert/strict";
import fs from "node:fs";
import {
    activeSceneFromOutput,
    applySceneReroll,
    resumeSelection,
} from "../web/h3_chain_cancel_reroll_core.mjs";

const plan = {
    shots: [
        {id: "one", prompt: ["One."]},
        {id: "two", prompt: ["Two."], seed: "2"},
        {id: "three", prompt: ["Three."]},
    ],
};
applySceneReroll(plan, 2, "18446744073709551615");
assert.equal(plan.shots[1].seed, "18446744073709551615");
assert.throws(() => applySceneReroll(plan, 4, "1"), /does not exist/);
assert.throws(
    () => applySceneReroll(plan, 1, "18446744073709551616"),
    /unsigned 64-bit/,
);

assert.deepEqual(resumeSelection(2, 3, 3), {
    startClip: 2,
    sceneRange: "",
});
assert.deepEqual(resumeSelection(3, 5, 8), {
    startClip: 3,
    sceneRange: "3:5",
});
assert.deepEqual(resumeSelection(5, 5, 8), {
    startClip: 5,
    sceneRange: "5",
});
assert.throws(() => resumeSelection(6, 5, 8), /Invalid reroll range/);

assert.deepEqual(activeSceneFromOutput({
    h3_chain_active_scene: [{
        run_name: "project",
        clip_index: 2,
        clip_count: 8,
        end_clip: 5,
        shot_id: "hallway",
        seed: "9007199254740993",
    }],
}), {
    runName: "project",
    clipIndex: 2,
    clipCount: 8,
    endClip: 5,
    shotId: "hallway",
    seed: "9007199254740993",
});
assert.equal(activeSceneFromOutput({h3_chain_active_scene: []}), null);

const source = fs.readFileSync(
    new URL("../web/h3_chain_cancel_reroll.js", import.meta.url),
    "utf8",
);
assert.match(source, /refreshRestoredPlanEditors\(planNode\)/,
    "reroll seed changes refresh every Plan companion");
assert.match(source, /\/api\/jobs\/\$\{encodeURIComponent\(record\.promptId\)\}\/cancel/);
assert.match(source, /execution_interrupted/);
assert.match(source, /await waiter\.promise/);
assert.match(source, /requireVisibleWorkflow\(record\);[\s\S]*verifyPredecessorCheckpoint/);
assert.match(source, /function activeWorkflowIdentity\(\)/);
assert.match(source, /workflowIdentity !== record\.workflowIdentity/);
assert.match(source, /widgetByName\(planNode, "run_name"\)/);
assert.match(source, /record\.currentNode = currentNode/);
assert.doesNotMatch(source, /currentNode !== record\.currentNode/);
assert.match(source, /active = null;[\s\S]*Waiting for ComfyUI to finish interrupting/);
assert.match(source, /applySceneReroll/);
assert.match(source, /resumeSelection/);
assert.match(source, /await app\.queuePrompt\(0, 1\)/);
assert.match(source, /checkpoint \$\{predecessor\} is not ready/);
assert.match(source, /queue the workflow manually/);
assert.match(source, /MiniMaxH3ContexLoop\.cancelRerollControl/);
assert.match(source, /Show floating Cancel & reroll control/);
assert.match(source, /defaultValue:\s*true/);
assert.match(source, /onChange\(value\)[\s\S]*setControlAllowed\(value\)/);
assert.match(source, /if \(!controlAllowed \|\| !active \|\| busy\) return/);
assert.match(source, /root\.hidden = !controlAllowed \|\| !wantsVisible/);
assert.match(source, /\.h3cr-status[\s\S]*max-width:100%[\s\S]*overflow-wrap:anywhere/);
assert.doesNotMatch(source, /fetchApi\("\/interrupt"/);

console.log("H3 cancel-and-reroll helpers: ok");