import assert from "node:assert/strict";
import { readFile } from "node:fs/promises";
import createModule from "../../build-wasm/linuxcnc-sim-core.js";
import { loadSimulationCase, toWorkerConfiguration } from "../workspace/sim-case-loader.mjs";
import { SimSession, CONTEXT_ID } from "./sim-session.mjs";

const CASE = "linuxcnc/configs/sim/axis/vismach/3axis-tutorial/3axis-tutorial.ini";
const fetchImpl = async (url) => {
  try {
    const body = await readFile(new URL(`../../${String(url).replace(/^\//, "")}`, import.meta.url));
    return { ok: true, text: async () => body.toString("utf8") };
  } catch {
    return { ok: false, text: async () => "" };
  }
};

const bundle = await loadSimulationCase({ path: CASE, fetchImpl });
const core = await createModule({ wasmBinary: await readFile(new URL("../../build-wasm/linuxcnc-sim-core.wasm", import.meta.url)) });
const session = new SimSession(core, { sessionId: "axis3-visible-completion" });
const base = { schema_version: 1, context_id: CONTEXT_ID, session_id: session.sessionId };
let sequence = 0;
const send = (kind, operationId, payload = {}) => session.accept({ ...base, kind, seq: ++sequence, operation_id: operationId, payload });

send("HELLO", "hello");
send("SESSION_ATTACH", "attach");
send("LOAD_WORKSPACE", "workspace", { root_path: "/workspace", configuration: toWorkerConfiguration(bundle) });
const source = "G21 G90\nG0 X0 Y0 Z0\nG1 X10 Y5 F120\nG1 Z-1 F60\nG0 X0 Y0 Z0\nM2\n";
send("PROGRAM_LOAD", "program", { source, path: "visible-completion.ngc" });
send("PROGRAM_RUN", "run");
let finish = null;
let sawWasmMotionAuthority = false;
for (let index = 0; index < 512; index += 1) {
  finish = send("SIM_STEP", `step-${index}`, { cycles: 1 });
  const wasmPosition = finish.payload.wasm_motion?.state?.position;
  const authoritativePosition = finish.payload.authoritative?.kinematics?.pose;
  if (wasmPosition && authoritativePosition && (wasmPosition.x !== authoritativePosition.x || wasmPosition.y !== authoritativePosition.y || wasmPosition.z !== authoritativePosition.z)) sawWasmMotionAuthority = true;
  if (finish.payload.program?.program_status === "completed" && finish.payload.task?.state === "STOPPED" && finish.payload.wasm_motion?.state?.queue_depth === 0) break;
}

assert.equal(finish.payload.program.program_status, "completed");
assert.equal(finish.payload.task.state, "STOPPED");
assert.equal(finish.payload.authoritative.kinematics.pose.x, 0);
assert.equal(finish.payload.authoritative.kinematics.pose.y, 0);
assert.equal(finish.payload.authoritative.kinematics.pose.z, 0);
assert.equal(finish.payload.motion.axes.x, 0);
assert.equal(finish.payload.motion.axes.y, 0);
assert.equal(finish.payload.motion.axes.z, 0);
assert.equal(finish.payload.wasm_motion.state.source, "wasm_motion_queue");
assert.equal(finish.payload.wasm_motion.state.queue_depth, 0);
assert.equal(sawWasmMotionAuthority, true);
console.log(JSON.stringify({ schema: "axis3-visible-completion.v1", status: "verified", case: CASE, final_pose: finish.payload.authoritative.kinematics.pose, task_state: finish.payload.task.state, program_status: finish.payload.program.program_status }));