import fs from "node:fs";
import path from "node:path";
import { test, expect } from "@playwright/test";
import { NativeUIWorkflow, observeNative } from "../fixtures/native-ui-workflow";

const root = path.resolve(import.meta.dirname, "../../..");
const effects = path.join(root, "blender-5.2.0/tests/files/sequence_editing/effects");
const mask = fs.readFileSync(path.join(effects, "mask.png"));

test("probe pythonReset before and after pack_all", async ({ page }, testInfo) => {
  test.setTimeout(180_000);
  await observeNative(page);
  await page.goto("/");
  const flow = new NativeUIWorkflow(page, testInfo.outputPath("cases"));
  await flow.command(() => page.getByTestId("blend-file-input").setInputFiles(path.join(effects, "fx_comp.blend")));
  await expect(page.getByTestId("engine-status")).toContainText("SceneIR", { timeout: 60000 });
  const out = await page.evaluate(async (maskBase64) => {
    const worker = (window as any).__nativeWorkflowEngine as Worker;
    const send = (command: any) => new Promise<any>((resolve, reject) => {
      const requestId = `rap-${crypto.randomUUID()}`;
      const receive = (event: MessageEvent) => {
        if (event.data?.requestId !== requestId || event.data?.kind !== 'result') return;
        worker.removeEventListener('message', receive);
        if (event.data.ok) resolve(event.data.result); else reject(new Error(`${event.data.error?.code}: ${event.data.error?.message}`));
      };
      worker.addEventListener('message', receive); worker.postMessage({ requestId, command });
    });
    const py = (source: string) => send({ type: 'jsonCommand', cmd: 'pythonExec', params: { source, filename: 'Rap.py', console: false } });
    const log: any = {};
    // 1) plain python edit then reset
    await py("import bpy\nbpy.context.scene['probe']='a'");
    try { await send({ type: 'jsonCommand', cmd: 'pythonReset', params: {} }); log.resetAfterPlain = 'ok'; }
    catch (e: any) { log.resetAfterPlain = String(e?.message ?? e).slice(0, 200); }
    // 2) pack then reset
    await py(["import bpy, base64", "open('/browser/mask.png','wb').write(base64.b64decode('" + maskBase64 + "'))", "print('PACK', bpy.ops.file.pack_all())"].join("\n"));
    try { await send({ type: 'jsonCommand', cmd: 'pythonReset', params: {} }); log.resetAfterPack = 'ok'; }
    catch (e: any) { log.resetAfterPack = String(e?.message ?? e).slice(0, 260); }
    return log;
  }, mask.toString("base64"));
  fs.writeFileSync("/tmp/reset-after-pack.json", JSON.stringify(out, null, 2));
});