import * as cp from "node:child_process";
import { existsSync, promises as fs } from "node:fs";
import { tmpdir } from "node:os";
import path from "node:path";
import { promisify } from "node:util";

import glob from "glob";

const exec = promisify(cp.exec);

const newFixturesDir = "packages/ast-spec/src/unsorted-fixtures";

const oldFixtures = glob.sync("packages/shared-fixtures/fixtures/**/*.ts");

async function main(): Promise<void> {
  // Make sure the new fixtures directory exists, and has no prior contents
  if (existsSync(newFixturesDir)) {
    await fs.rm(newFixturesDir, { recursive: true });
  }

  await fs.mkdir(newFixturesDir, {
    recursive: true,
  });

  // Copy each of the old fixtures into the new fixtures directory
  for (const fixture of oldFixtures) {
    const [oldFileName, group] = fixture.split("/").reverse();
    const fixtureName = oldFileName.split(".")[0];
    const fixtureDirectory = path.join(
      newFixturesDir,
      group,
      "fixtures",
      fixtureName
    );

    await fs.mkdir(fixtureDirectory, { recursive: true });
    await fs.writeFile(
      path.join(fixtureDirectory, "fixture.ts"),
      await fs.readFile(fixture)
    );
  }

  // Run Jest on ast-spec to find new fixtures with syntax errors
  const tempFile = path.join(tmpdir(), "temp.json");
  try {
    await exec(`jest /fixtures --json > ${tempFile}`, {
      cwd: "packages/ast-spec",
    });
  } catch {
    /* expected */
  }
  const jestOutput = JSON.parse((await fs.readFile(tempFile)).toString()) as {
    testResults: {
      assertionResults: { ancestorTitles: string[]; status: string }[];
    }[];
  };
  await fs.rm(tempFile);

  // Filter & deduplicate messages from those failing tests
  const failingFixtures = Array.from(
    new Set(
      jestOutput.testResults[0]
        .filter(
          (assertion) =>
            assertion.ancestorTitles.includes("unsorted-fixtures") &&
            assertion.status !== "passed"
        )
        .map((assertion) => assertion.ancestorTitles.slice(2).join("/"))
    )
  );

  // For each failing fixture, move it to an _error_ directory
  for (const failingFixture of failingFixtures) {
    const failingFixtureDirectory = path.join(
      newFixturesDir,
      failingFixture.replace("/", "/fixtures/")
    );

    const newErrorFixtureDirectory = path.join(
      failingFixtureDirectory.replace("/fixtures/", "/fixtures/_error_/")
    );
    await fs.mkdir(newErrorFixtureDirectory, {
      recursive: true,
    });
    await fs.rename(
      path.join(failingFixtureDirectory, "fixture.ts"),
      path.join(newErrorFixtureDirectory, "fixture.ts")
    );
    await fs.rm(path.join(failingFixtureDirectory, "snapshots"), {
      recursive: true,
    });
  }

  // Re-run Jest to update snapshots a second time
  await exec(`jest /fixtures`, {
    cwd: "packages/ast-spec",
  });
}

main()
  .then(() => console.log("Done."))
  .catch((error) => {
    throw error;
  });