import { spawnSync } from "node:child_process";
import { resolve as resolvePath } from "node:path";
import { fileURLToPath } from "node:url";

const DEFAULT_ATTEMPTS = 3;
const DEFAULT_BACKOFF_MS = 20_000;

function runNpm(args) {
  const command = process.platform === "win32" ? "npm.cmd" : "npm";
  const result = spawnSync(command, args, {
    shell: process.platform === "win32",
    stdio: "inherit",
  });

  if (result.error) {
    console.error(`Failed to start npm: ${result.error.message}`);
    return 1;
  }

  return result.status ?? 1;
}

function wait(delayMs) {
  return new Promise((resolve) => setTimeout(resolve, delayMs));
}

export async function runWithRetry(
  args,
  { attempts = DEFAULT_ATTEMPTS, backoffMs = DEFAULT_BACKOFF_MS, run = runNpm, sleep = wait } = {},
) {
  let exitCode = 1;

  for (let attempt = 1; attempt <= attempts; attempt += 1) {
    exitCode = await run(args);
    if (exitCode === 0) {
      return 0;
    }

    if (attempt < attempts) {
      const delayMs = attempt * backoffMs;
      console.warn(
        `npm failed with exit code ${exitCode}; retrying in ${delayMs / 1000}s (${attempt + 1}/${attempts})`,
      );
      await sleep(delayMs);
    }
  }

  return exitCode;
}

if (process.argv[1] && fileURLToPath(import.meta.url) === resolvePath(process.argv[1])) {
  const args = process.argv.slice(2);
  if (args.length === 0) {
    console.error("Usage: node scripts/npm-retry.mjs <npm arguments...>");
    process.exitCode = 2;
  } else {
    process.exitCode = await runWithRetry(args);
  }
}