#!/usr/bin/env tsx
/**
 * Set the version across all publishable packages in the monorepo.
 *
 * Usage:
 *   bun run set-version 0.1.1
 *   bun run set-version 0.1.1 --tag   # also creates a git commit and tag
 *
 * All packages share a single version number (fixed versioning).
 */

import { readFileSync, writeFileSync } from "fs";
import { join } from "path";
import { execSync } from "child_process";

const PACKAGES = [
  "packages/core",
  "packages/engine",
  "packages/producer",
  "packages/studio",
  "packages/cli",
];

const ROOT = join(import.meta.dirname, "..");

function main() {
  const args = process.argv.slice(2);
  const version = args.find((a) => !a.startsWith("--"));
  const shouldTag = args.includes("--tag");

  if (!version) {
    console.error("Usage: bun run set-version <version> [--tag]");
    console.error("Example: bun run set-version 0.1.1 --tag");
    process.exit(1);
  }

  if (!/^\d+\.\d+\.\d+(-[\w.]+)?$/.test(version)) {
    console.error(`Invalid semver: ${version}`);
    process.exit(1);
  }

  // Update each package.json
  for (const pkg of PACKAGES) {
    const pkgPath = join(ROOT, pkg, "package.json");
    const content = JSON.parse(readFileSync(pkgPath, "utf-8"));
    const oldVersion = content.version;
    content.version = version;
    writeFileSync(pkgPath, JSON.stringify(content, null, 2) + "\n");
    console.log(`  ${content.name}: ${oldVersion} -> ${version}`);
  }

  console.log(`\nSet ${PACKAGES.length} packages to v${version}`);

  if (shouldTag) {
    // Verify working tree is clean (aside from the version bumps we just made)
    const status = execSync("git status --porcelain", {
      cwd: ROOT,
      encoding: "utf-8",
    }).trim();
    const unexpected = status
      .split("\n")
      .filter((line) => line && !PACKAGES.some((pkg) => line.includes(pkg)));
    if (unexpected.length > 0) {
      console.error("\nUnexpected uncommitted changes:");
      unexpected.forEach((line) => console.error(`  ${line}`));
      console.error("Commit or stash these before tagging.");
      process.exit(1);
    }

    execSync(`git add ${PACKAGES.map((p) => join(p, "package.json")).join(" ")}`, {
      cwd: ROOT,
      stdio: "inherit",
    });
    execSync(`git commit -m "chore: release v${version}"`, { cwd: ROOT, stdio: "inherit" });
    execSync(`git tag v${version}`, { cwd: ROOT, stdio: "inherit" });
    console.log(`\nCreated commit and tag v${version}`);
    console.log(`Run 'git push origin main --tags' to trigger the publish workflow.`);
  }
}

main();