import { existsSync, readFileSync, rmSync } from "fs";
import { join } from "path";
import os from "os";
import proc from "child_process";
import { collectPackageLicenses } from "./licenses";
import chalk from "chalk";
import { getDesktopPackageInfo } from "./package-info";
import { bundleCopy } from "./build-desktop";

const __dirname = import.meta.dirname; // polyfill __dirname

const scripts = {
  Windows_NT: "build:win",
  Linux: "build:linux",
  Darwin: "build:mac",
};

async function installDependencies() {
  if (process.argv.includes("--skip-deps")) {
    console.log("Skipping dependency installation as per --skip-deps flag.");
    return;
  }
  console.log("🚚 Installing dependencies");
  proc.execSync("pnpm install", {
    shell: true,
    stdio: "inherit",
  });
}

async function runPnpmScript(script) {
  console.log(`Running pnpm script: ${script}`);
  proc.execSync(`pnpm ${script}`, {
    shell: true,
    stdio: "inherit",
  });
}

async function removeArtifacts(version) {
  console.log(chalk.gray("Checking for previous artifacts..."));
  const artifactDir = join(
    process.cwd(),
    `packages/app-desktop/release/${version}/`,
  );
  const artifactExists = existsSync(artifactDir);

  if (artifactExists) {
    console.log(
      chalk.gray(
        "Found older artifacts for this version. Removing for rebuild...",
      ),
    );
    if (os.type() == "Windows_NT") {
      console.log("Killing all Darkwrite.exe processes");
      try {
        proc.execSync("taskkill /f /im Darkwrite.exe");
      } catch {
        /*empty*/
      }
    }
    console.log(chalk.gray("Removing old artifacts..."));
    rmSync(artifactDir, { recursive: true });
  } else
    console.log("There are no previous artifacts. Proceeding with build...");
}

async function main() {
  console.log(chalk.cyanBright.bold("📦 Darkwrite Builder ---"));
  console.log(`\u001b[30mRunning on ${os.type()} ${os.release()}`);

  const { version } = await getDesktopPackageInfo();
  console.log(chalk.blue(`Target Darkwrite version: ${version}`));

  await installDependencies();
  await removeArtifacts(version);

  console.log("\u001b[1;36m🔨 Starting new build...\u001b[0m");

  const startTimestamp = Date.now();

  const rootDir = process.cwd();

  await runPnpmScript("build:common");
  await runPnpmScript("build:i18n");

  await installDependencies(); // Reinstall after building common to ensure all dependencies are correctly linked
  await runPnpmScript("build:frontend");
  await runPnpmScript("bundle:desktop");

  await collectPackageLicenses();
  
  await bundleCopy();
  process.chdir(join(rootDir, "packages/app-desktop"));
  const buildProcess = proc.spawn(`pnpm`, [scripts[os.type()]], {
    stdio: "inherit",
    shell: true,
  });

  buildProcess.on("error", (err) => {
    console.error("\u001b[31;1m✘ Build failed: \u001b[0m" + err.message);
    if (err.stack) console.error(err.stack);
  });

  buildProcess.on("exit", (code) => {
    const endTimestamp = Date.now();
    if (code == 0) {
      console.log(
        `\u001b[32;1m ✔ Built Darkwrite ${version} for ${os.type()} \u001b[30;22mtook ${(endTimestamp - startTimestamp) / 1000}s\u001b[0m`,
      );
    } else
      console.log(
        `\u001b[31;1m✘ Build failed with exit code ${code}: \u001b[0m`,
      );
  });

  process.on("SIGINT", () => {
    console.log("\nBuild interrupted by user. Exiting...");
    buildProcess.kill("SIGINT");
    process.exit(1);
  });
}

main();