/**
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 * @format
 */

import {
  argv,
  cleanTask,
  copyTask,
  eslintTask,
  logger,
  jestTask,
  option,
  parallel,
  series,
  spawn,
  task,
  tscTask,
} from "just-scripts";

import glob from "glob";
import path from "path";
import which from "which";

const node = process.execPath;

option("fix");

task("clean", cleanTask({ paths: ["build", "dist"] }));

task(
  "prepare-for-build",
  parallel(
    babelTransformTask({ paths: ["src_js"], dest: "dist" }),
    copyTask({ paths: ["src_js/**/*.d.ts"], dest: "dist" }),
    emcmakeGenerateTask()
  )
);

function defineFlavor(flavor: string, env: NodeJS.ProcessEnv) {
  task(`cmake-build:${flavor}`, cmakeBuildTask({ targets: [flavor] }));
  task(
    `jest:${flavor}`,
    jestTask({ config: path.join(__dirname, "jest.config.ts"), env })
  );
  task(
    `test:${flavor}`,
    series("prepare-for-build", `cmake-build:${flavor}`, `jest:${flavor}`)
  );
}

defineFlavor("asmjs-async", { WASM: "0", SYNC: "0" });
defineFlavor("asmjs-sync", { WASM: "0", SYNC: "1" });
defineFlavor("wasm-async", { WASM: "1", SYNC: "0" });
defineFlavor("wasm-sync", { WASM: "1", SYNC: "1" });

task("cmake-build:all", cmakeBuildTask());
task(
  "cmake-build:async",
  cmakeBuildTask({ targets: ["asmjs-async", "wasm-async"] })
);
task(
  "cmake-build:sync",
  cmakeBuildTask({ targets: ["asmjs-sync", "wasm-sync"] })
);

task("build", series("prepare-for-build", "cmake-build:all"));

task(
  "test",
  series(
    "prepare-for-build",
    series("cmake-build:asmjs-async", "jest:asmjs-async"),
    series("cmake-build:asmjs-sync", "jest:asmjs-sync"),
    series("cmake-build:wasm-async", "jest:wasm-async"),
    series("cmake-build:wasm-sync", "jest:wasm-sync")
  )
);

task(
  "benchmark",
  series("prepare-for-build", "cmake-build:sync", runBenchTask())
);

task(
  "lint",
  parallel(
    tscTask(),
    series(
      eslintTask({ fix: argv().fix }),
      clangFormatTask({ fix: argv().fix })
    )
  )
);

function babelTransformTask(opts: {
  paths: ReadonlyArray<string>;
  dest: string;
}) {
  return () => {
    const args = [...opts.paths, "--source-maps", "--out-dir", opts.dest];
    logger.info(`Transforming [${opts.paths.join(",")}] to '${opts.dest}'`);

    return spawn(node, [require.resolve("@babel/cli/bin/babel"), ...args]);
  };
}

function runBenchTask() {
  return () => {
    const files = glob.sync("./tests/Benchmarks/**/*");
    const args = ["./tests/bin/run-bench.ts", ...files];
    logger.info(args.join(" "));

    return spawn(node, args, {
      stdio: "inherit",
      env: { NODE_OPTIONS: "-r ts-node/register" },
    });
  };
}

function emcmakeGenerateTask() {
  return () => {
    const emcmake = which.sync("emcmake");
    const ninja = which.sync("ninja", { nothrow: true });
    const args = [
      "cmake",
      "-S",
      ".",
      "-B",
      "build",
      ...(ninja ? ["-G", "Ninja"] : []),
    ];
    logger.info(["emcmake", ...args].join(" "));

    return spawn(emcmake, args);
  };
}

function cmakeBuildTask(opts?: { targets?: ReadonlyArray<string> }) {
  return () => {
    const cmake = which.sync("cmake");
    const args = [
      "--build",
      "build",
      ...(opts?.targets ? ["--target", ...opts.targets] : []),
    ];
    logger.info(["cmake", ...args].join(" "));

    return spawn(cmake, args, { stdio: "inherit" });
  };
}

function clangFormatTask(opts?: { fix?: boolean }) {
  return () => {
    const args = [
      ...(opts?.fix ? ["-i"] : ["--dry-run", "--Werror"]),
      ...glob.sync("**/*.{h,hh,hpp,c,cpp,cc,m,mm}"),
    ];
    logger.info(["clang-format", ...args].join(" "));

    return spawn(node, [require.resolve("clang-format"), ...args], {
      stdio: "inherit",
    });
  };
}