headroom/scripts/version-sync.py

#!/usr/bin/env python3
"""Synchronize version across all headroom packages."""

from __future__ import annotations

import argparse
import json
import re
from pathlib import Path

import tomllib


def get_version_from_pyproject(root: Path) -> str:
    """Read version from pyproject.toml."""
    pyproject_path = root / "pyproject.toml"
    with open(pyproject_path, "rb") as f:
        data = tomllib.load(f)
    return data["project"]["version"]


def bump_version(version: str, bump_type: str) -> str:
    """Bump version according to bump_type (major, minor, patch)."""
    major, minor, patch = map(int, version.split("."))
    if bump_type == "major":
        major += 1
        minor = 0
        patch = 0
    elif bump_type == "minor":
        minor += 1
        patch = 0
    elif bump_type == "patch":
        patch += 1
    return f"{major}.{minor}.{patch}"


def update_version_py(root: Path, version: str) -> None:
    """Update headroom/_version.py with new version."""
    version_py_path = root / "headroom" / "_version.py"
    content = version_py_path.read_text(encoding="utf-8")
    updated = re.sub(
        r'__version__ = "[^"]+"',
        f'__version__ = "{version}"',
        content,
    )
    version_py_path.write_text(updated, encoding="utf-8")


def update_package_json(file_path: Path, version: str) -> None:
    """Update a package.json version field."""
    with open(file_path, encoding="utf-8") as f:
        data = json.load(f)
    data["version"] = version
    with open(file_path, "w", encoding="utf-8") as f:
        json.dump(data, f, indent=2)
        f.write("\n")


def update_pyproject_version(root: Path, version: str) -> None:
    """Update pyproject.toml version."""
    pyproject_path = root / "pyproject.toml"
    content = pyproject_path.read_text(encoding="utf-8")
    updated = re.sub(
        r'^version = "[^"]+"',
        f'version = "{version}"',
        content,
        flags=re.MULTILINE,
    )
    pyproject_path.write_text(updated, encoding="utf-8")


def write_release_metadata(root: Path, version: str) -> None:
    """Write .releaseetadata JSON file."""
    metadata = {
        "version": version,
        "packages": {
            "pypi": version,
            "npm-sdk": version,
            "npm-openclaw": version,
        },
    }
    metadata_path = root / ".releaseetadata"
    with open(metadata_path, "w", encoding="utf-8") as f:
        json.dump(metadata, f, indent=2)
        f.write("\n")


def main() -> None:
    parser = argparse.ArgumentParser(description="Synchronize version across headroom packages")
    parser.add_argument(
        "--root",
        type=Path,
        default=Path(__file__).parent.parent,
        help="Root directory of the project",
    )
    group = parser.add_mutually_exclusive_group()
    group.add_argument("--version", help="Explicit version to set (e.g., 0.6.0)")
    group.add_argument(
        "--bump",
        choices=["major", "minor", "patch"],
        help="Bump version from pyproject.toml",
    )
    args = parser.parse_args()

    if args.version:
        version = args.version
    elif args.bump:
        base_version = get_version_from_pyproject(args.root)
        version = bump_version(base_version, args.bump)
    else:
        version = get_version_from_pyproject(args.root)

    # Update all versioned files
    update_pyproject_version(args.root, version)
    update_version_py(args.root, version)
    update_package_json(args.root / "plugins" / "openclaw" / "package.json", version)
    update_package_json(args.root / "sdk" / "typescript" / "package.json", version)
    write_release_metadata(args.root, version)

    print(f"Version synchronized to {version}")


if __name__ == "__main__":
    main()