release

#!/usr/bin/env python3
"""Release script: ./release <version>  (e.g. ./release 0.0.5)"""

import re
import shutil
import subprocess
import sys
from pathlib import Path

ROOT = Path(__file__).parent
DATA = ROOT / "ai_prompt_auto_commit" / "data"
CLAUDE_SETTINGS = ROOT / ".claude" / "settings.json"
ASSISTANT_GUIDELINES = ROOT / ".github" / "assistant-guidelines.md"

def die(msg: str) -> None:
    print(f"error: {msg}", file=sys.stderr)
    sys.exit(1)


def replace_in_file(path: Path, pattern: str, replacement: str) -> None:
    text = path.read_text(encoding="utf-8")
    new_text, count = re.subn(pattern, replacement, text)
    if count == 0:
        die(f"pattern {pattern!r} not found in {path}")
    path.write_text(new_text, encoding="utf-8")
    print(f"  {path.relative_to(ROOT)}: {count} replacement(s)")


def main() -> None:
    if len(sys.argv) != 2:
        die(f"usage: {sys.argv[0]} <version>")

    version = sys.argv[1]

    if version.startswith("v"):
        die(f"version must not start with 'v', got: {version!r}")

    # ensure we're on main
    branch = subprocess.run(
        ["git", "rev-parse", "--abbrev-ref", "HEAD"],
        capture_output=True,
        text=True,
        check=True,
        cwd=ROOT,
    ).stdout.strip()
    if branch != "main":
        die(f"must be on main branch, currently on: {branch!r}")

    print(f"releasing {version}")

    # pyproject.toml
    replace_in_file(
        ROOT / "pyproject.toml",
        r'(?m)^version = ".*?"',
        f'version = "{version}"',
    )

    # README.md  — rev: v<old>
    replace_in_file(
        ROOT / "README.md",
        r"rev: v[\d.]+",
        f"rev: v{version}",
    )

    # data files
    # claude_settings.json - "version": "0.0.8"
    replace_in_file(
        CLAUDE_SETTINGS,
        r'"version"\s*:\s*"[\d.]+"',
        f"\"version\": \"{version}\"",
    )
    # assistant-guidelines.md - version: "0.0.8"
    replace_in_file(
        ASSISTANT_GUIDELINES,
        r"version: \"[\d.]+\"",
        f"version: \"{version}\"",
    )

    # CHANGES.md — insert new version heading only if not already present
    changes = ROOT / "CHANGES.md"
    text = changes.read_text(encoding="utf-8")
    if f"## v{version}" not in text:
        new_section = f"## v{version}\n\n- (fill in)\n\n"
        new_text = re.sub(r"(## v)", new_section + r"\1", text, count=1)
        if new_text == text:
            die("could not find a version heading in CHANGES.md to insert before")
        changes.write_text(new_text, encoding="utf-8")
        print(f"  CHANGES.md: inserted v{version} section")
    else:
        print(f"  CHANGES.md: v{version} already present, skipping")

    # copy data files
    print("  Updating bundled data files...")
    shutil.copyfile(CLAUDE_SETTINGS, DATA / "claude_settings.json")
    shutil.copyfile(ASSISTANT_GUIDELINES, DATA / "assistant-guidelines.md")

    # tag and push only when the working tree is clean
    status = subprocess.run(
        ["git", "status", "--porcelain"],
        capture_output=True,
        text=True,
        check=True,
        cwd=ROOT,
    )
    if status.stdout.strip():
        print("\nworking tree has changes — commit them, then re-run to tag and push")
    else:
        tag = f"v{version}"
        subprocess.run(["git", "push", "origin", "main"], check=True, cwd=ROOT)
        subprocess.run(["git", "tag", "-a", tag, "-m", f"Release {tag}"], check=True, cwd=ROOT)
        print(f"  tagged {tag}")
        subprocess.run(["git", "push", "origin", tag], check=True, cwd=ROOT)
        print(f"  pushed {tag}")


if __name__ == "__main__":
    main()