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/5axis/table-rotary-tilting/xyzac-trt.ini";
const PROGRAM_PATH = "linuxcnc/configs/sim/axis/vismach/5axis/table-rotary-tilting/demos/xyzac_switchkins.ngc";
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 });
assert.equal(bundle.status, "ready");
const source = bundle.files?.[PROGRAM_PATH];
assert.equal(typeof source, "string");
const core = await createModule({ wasmBinary: await readFile(new URL("../../build-wasm/linuxcnc-sim-core.wasm", import.meta.url)) });
const session = new SimSession(core, { sessionId: "xyzac-machine-sequence" });
const base = { schema_version: 1, context_id: CONTEXT_ID, session_id: session.sessionId };
let seq = 0;
const send = (kind, payload = {}) => session.accept({ ...base, kind, seq: ++seq, operation_id: `machine-${seq}`, payload });

send("HELLO");
send("SESSION_ATTACH");
assert.equal(send("LOAD_WORKSPACE", { root_path: "/workspace", configuration: toWorkerConfiguration(bundle) }).payload.status, "verified");
const loaded = send("PROGRAM_LOAD", { source, path: PROGRAM_PATH });
assert.equal(loaded.kind, "COMMAND_RESULT");
assert.equal(loaded.payload.motion_authority, "wasm");
assert.equal(loaded.payload.panel_snapshot.devices.source, "wasm_io");
const jsMotionHash = session.motion.export().state_hash;

const emergency = send("ESTOP", { active: true });
assert.equal(emergency.kind, "COMMAND_RESULT");
assert.equal(emergency.payload.task.state, "ESTOP");
assert.equal(emergency.payload.motion_authority, "wasm");
assert.equal(emergency.payload.device_io.source, "wasm_io");
assert.equal(emergency.payload.device_io.state.estop, 1);
assert.equal(emergency.payload.device_io.state.ready, 0);
assert.equal(emergency.payload.panel_snapshot.motion.estop, true);
assert.equal(emergency.payload.panel_snapshot.wasm_motion.state.queue_depth, 0);
assert.equal(session.motion.export().state_hash, jsMotionHash);

const released = send("ESTOP", { active: false });
assert.equal(released.payload.status, "verified");
assert.equal(released.payload.device_io.state.estop, 0);
assert.equal(released.payload.device_io.state.ready, 0);
assert.equal(released.payload.task.state, "STOPPED");
assert.equal(released.payload.panel_snapshot.axis_state.source_task_state, "STATE_ESTOP_RESET");

const powered = send("POWER", { active: true });
assert.equal(powered.payload.status, "verified");
assert.equal(powered.payload.task.state, "READY");
assert.equal(powered.payload.device_io.source, "wasm_io");
assert.equal(powered.payload.device_io.state.ready, 1);
assert.equal(powered.payload.panel_snapshot.axis_state.machine_state, "ON");

const homed = send("HOME", { all: true });
assert.equal(homed.payload.status, "verified");
assert.equal(homed.payload.motion_authority, "wasm");
assert.equal(homed.payload.wasm_home_limit.state.homed, true);
assert.equal(homed.payload.wasm_home_limit.state.joint_count, 5);
assert.deepEqual(homed.payload.panel_snapshot.motion.homed_axes, [true, true, true, true, true]);
assert.deepEqual(homed.payload.wasm_home_limit.state.joints, [0, 0, 0, 0, 0]);

const running = send("PROGRAM_RUN");
assert.equal(running.payload.status, "verified");
assert.equal(running.payload.task.state, "RUNNING");
assert.equal(running.payload.program.source, "linuxcnc-wasm-sim_program");
assert.equal(running.payload.motion_authority, "wasm");
assert.equal(session.wasmMotion.exportState().controls.paused, false);

const stepped = send("SIM_STEP", { cycles: 1000 });
assert.equal(stepped.payload.status, "verified");
assert.equal(stepped.payload.motion_authority, "wasm");
assert.ok(stepped.payload.program.current_line > 0);
assert.ok(stepped.payload.program_steps.length > 0);
assert.equal(stepped.payload.panel_snapshot.visual.source, "wasm_canonical_motion");
assert.deepEqual(stepped.payload.panel_snapshot.visual.pose, stepped.payload.wasm_motion.state.position);
assert.deepEqual(stepped.payload.panel_snapshot.visual.joints, stepped.payload.panel_snapshot.motion.joints);
assert.deepEqual(stepped.payload.panel_snapshot.coordinates.machine, [stepped.payload.panel_snapshot.visual.pose.x, stepped.payload.panel_snapshot.visual.pose.y, stepped.payload.panel_snapshot.visual.pose.z, stepped.payload.panel_snapshot.visual.pose.a, stepped.payload.panel_snapshot.visual.pose.c]);
assert.deepEqual(stepped.payload.panel_snapshot.coordinates.commanded, stepped.payload.panel_snapshot.coordinates.machine);
assert.ok(stepped.payload.panel_snapshot.visual.active_progress > 0);
assert.equal(stepped.payload.wasm_motion.state.active_tag, 1);
assert.equal(stepped.payload.panel_snapshot.kinematics.active_type, 1);
assert.ok(Number.isFinite(stepped.payload.panel_snapshot.visual.pose.a));
assert.ok(Number.isFinite(stepped.payload.panel_snapshot.visual.pose.c));
assert.equal(session.motion.export().state_hash, jsMotionHash);

const aborted = send("PROGRAM_ABORT");
assert.equal(aborted.payload.task.state, "STOPPED");
const restarted = send("PROGRAM_RESTART");
assert.equal(restarted.payload.status, "verified");
const fullRun = send("PROGRAM_RUN");
assert.equal(fullRun.payload.task.state, "RUNNING");
let completed = send("SIM_STEP", { cycles: 8000 });
assert.equal(completed.payload.status, "verified");
assert.equal(completed.payload.program.program_status, "completed");
// M2 closes the interpreter before the last trajectory segment drains.
let drainTicks = 0;
for (; drainTicks < 100 && completed.payload.wasm_motion.state.queue_depth > 0; drainTicks += 1) {
  assert.equal(completed.payload.panel_snapshot.gates.run.enabled, false);
  completed = send("SIM_STEP", { cycles: 1 });
  assert.equal(completed.payload.status, "verified");
}
assert.equal(completed.payload.task.state, "STOPPED");
assert.equal(completed.payload.wasm_motion.state.queue_depth, 0);
assert.equal((completed.payload.panel_snapshot.visual.toolpath ?? loaded.payload.panel_snapshot.visual.toolpath).length, 29);
assert.equal(completed.payload.panel_snapshot.kinematics.active_type, 0);

console.log(JSON.stringify({
  schema: "xyzac-machine-sequence.v1",
  status: "verified",
  case: CASE,
  sequence: ["ESTOP", "ESTOP_RESET", "POWER", "HOME_ALL", "PROGRAM_RUN", "SIM_STEP"],
  authority: { motion: "wasm", io: "wasm_io", home_limit: "wasm_home_limit", program: "linuxcnc-wasm-sim_program", visual: "wasm_canonical_motion" },
  estop_queue_depth: emergency.payload.panel_snapshot.wasm_motion.state.queue_depth,
  home_joint_count: homed.payload.wasm_home_limit.state.joint_count,
  current_line_after_step: stepped.payload.program.current_line,
  live_axis_sample: stepped.payload.panel_snapshot.visual.pose,
  live_joints: stepped.payload.panel_snapshot.visual.joints,
  active_segment_id: stepped.payload.panel_snapshot.visual.active_segment_id,
  active_progress: stepped.payload.panel_snapshot.visual.active_progress,
  rtcp_active_type: stepped.payload.panel_snapshot.kinematics.active_type,
  rtcp_joint_pose_delta: stepped.payload.panel_snapshot.kinematics.joints.map((value, index) => value - [stepped.payload.panel_snapshot.visual.pose.x, stepped.payload.panel_snapshot.visual.pose.y, stepped.payload.panel_snapshot.visual.pose.z, stepped.payload.panel_snapshot.visual.pose.a, stepped.payload.panel_snapshot.visual.pose.c][index]),
  full_run: { cycles: 8000 + drainTicks, drain_ticks: drainTicks, program_status: completed.payload.program.program_status, task_state: completed.payload.task.state, queue_depth: completed.payload.wasm_motion.state.queue_depth, final_switchkins_type: completed.payload.panel_snapshot.kinematics.active_type },
  js_motion_unchanged: session.motion.export().state_hash === jsMotionHash,
  state_hash: stepped.payload.panel_snapshot.state_hash,
}));