import { spawn } from "bun";
import { expect, it } from "bun:test";
import { bunEnv } from "bunEnv";
import { bunExe } from "bunExe";
import { readFileSync, renameSync, rmSync, unlinkSync, writeFileSync } from "fs";

it("should hot reload when file is overwritten", async () => {
  const root = import.meta.dir + "/hot-runner.js";
  const runner = spawn({
    cmd: [bunExe(), "--hot", "run", root],
    env: bunEnv,
    stdout: "pipe",
    stderr: "inherit",
    stdin: "ignore",
  });

  var reloadCounter = 0;

  async function onReload() {
    writeFileSync(root, readFileSync(root, "utf-8"));
  }

  for await (const line of runner.stdout!) {
    var str = new TextDecoder().decode(line);
    var any = false;
    for (let line of str.split("\n")) {
      if (!line.includes("[#!root]")) continue;
      reloadCounter++;

      if (reloadCounter === 3) {
        runner.unref();
        runner.kill();
        break;
      }

      expect(line).toContain(`[#!root] Reloaded: ${reloadCounter}`);
      any = true;
    }

    if (any) await onReload();
  }

  expect(reloadCounter).toBe(3);
});

it("should recover from errors", async () => {
  const root = import.meta.dir + "/hot-runner.js";
  const runner = spawn({
    cmd: [bunExe(), "--hot", "run", root],
    env: bunEnv,
    stdout: "pipe",
    stderr: "pipe",
    stdin: "ignore",
  });

  let reloadCounter = 0;
  const input = readFileSync(root, "utf-8");
  function onReloadGood() {
    writeFileSync(root, input);
  }

  function onReloadError() {
    writeFileSync(root, "throw new Error('error');\n");
  }

  var queue = [onReloadError, onReloadGood, onReloadError, onReloadGood];
  var errors: string[] = [];
  var onError;
  (async () => {
    for await (let line of runner.stderr!) {
      var str = new TextDecoder().decode(line);
      errors.push(str);
      onError && onError(str);
    }
  })();

  for await (const line of runner.stdout!) {
    var str = new TextDecoder().decode(line);
    var any = false;
    for (let line of str.split("\n")) {
      if (!line.includes("[#!root]")) continue;
      reloadCounter++;

      if (reloadCounter === 3) {
        runner.unref();
        runner.kill();
        break;
      }

      expect(line).toContain(`[#!root] Reloaded: ${reloadCounter}`);
      any = true;
    }

    if (any) {
      queue.shift()!();
      await new Promise<void>((resolve, reject) => {
        if (errors.length > 0) {
          errors.length = 0;
          resolve();
          return;
        }

        onError = resolve;
      });

      queue.shift()!();
    }
  }

  expect(reloadCounter).toBe(3);
});

it("should not hot reload when a random file is written", async () => {
  const root = import.meta.dir + "/hot-runner.js";
  const runner = spawn({
    cmd: [bunExe(), "--hot", "run", root],
    env: bunEnv,
    stdout: "pipe",
    stderr: "inherit",
    stdin: "ignore",
  });

  let reloadCounter = 0;
  const code = readFileSync(root, "utf-8");
  async function onReload() {
    writeFileSync(root + ".another.yet.js", code);
    unlinkSync(root + ".another.yet.js");
  }
  var waiter = new Promise<void>((resolve, reject) => {
    setTimeout(async () => {
      resolve();
    }, 50);
  });
  var finished = false;
  await Promise.race([
    waiter,
    (async () => {
      if (finished) {
        return;
      }
      for await (const line of runner.stdout!) {
        if (finished) {
          return;
        }

        var str = new TextDecoder().decode(line);
        for (let line of str.split("\n")) {
          if (!line.includes("[#!root]")) continue;
          if (finished) {
            return;
          }
          await onReload();

          reloadCounter++;
          expect(line).toContain(`[#!root] Reloaded: ${reloadCounter}`);
        }
      }
    })(),
  ]);
  finished = true;
  runner.kill(0);
  runner.unref();

  expect(reloadCounter).toBe(1);
});

it("should hot reload when a file is deleted and rewritten", async () => {
  const root = import.meta.dir + "/hot-runner.js";
  const runner = spawn({
    cmd: [bunExe(), "--hot", "run", root],
    env: bunEnv,
    stdout: "pipe",
    stderr: "inherit",
    stdin: "ignore",
  });

  var reloadCounter = 0;

  async function onReload() {
    const contents = readFileSync(root, "utf-8");
    unlinkSync(root);
    writeFileSync(root, contents);
  }

  for await (const line of runner.stdout!) {
    var str = new TextDecoder().decode(line);
    var any = false;
    for (let line of str.split("\n")) {
      if (!line.includes("[#!root]")) continue;
      reloadCounter++;

      if (reloadCounter === 3) {
        runner.unref();
        runner.kill();
        break;
      }

      expect(line).toContain(`[#!root] Reloaded: ${reloadCounter}`);
      any = true;
    }

    if (any) await onReload();
  }

  expect(reloadCounter).toBe(3);
});

it("should hot reload when a file is renamed() into place", async () => {
  const root = import.meta.dir + "/hot-runner.js";
  const runner = spawn({
    cmd: [bunExe(), "--hot", "run", root],
    env: bunEnv,
    stdout: "pipe",
    stderr: "inherit",
    stdin: "ignore",
  });

  var reloadCounter = 0;

  async function onReload() {
    const contents = readFileSync(root, "utf-8");
    rmSync(root + ".tmpfile", { force: true });
    await 1;
    writeFileSync(root + ".tmpfile", contents);
    await 1;
    renameSync(root + ".tmpfile", root);
    await 1;
  }

  for await (const line of runner.stdout!) {
    var str = new TextDecoder().decode(line);
    var any = false;
    for (let line of str.split("\n")) {
      if (!line.includes("[#!root]")) continue;
      reloadCounter++;

      if (reloadCounter === 3) {
        runner.unref();
        runner.kill();
        break;
      }

      expect(line).toContain(`[#!root] Reloaded: ${reloadCounter}`);
      any = true;
    }

    if (any) await onReload();
  }

  expect(reloadCounter).toBe(3);
});