make_release.py

import os
import re

import typer
from git import Optional, Repo

app = typer.Typer()


@app.command()
def make_release(version_number: Optional[str] = None) -> None:
    repo = Repo(os.getcwd())
    if repo.is_dirty():
        raise Exception("Local git repository is not clean")
    if repo.active_branch.name != "main":
        raise Exception("Can only make a release from main")
    remote_branch = repo.references[f"origin/{repo.active_branch.name}"]
    local_commit = repo.active_branch.commit
    remote_commit = remote_branch.commit
    if local_commit != remote_commit:
        raise Exception(
            f"Your local branch {repo.active_branch.name} is not up to date with origin."
        )

    if version_number is None:
        version_number = guess_next_version(repo)
    version_pattern = r"^\d+\.\d+\.\d+$"
    if not re.match(version_pattern, version_number):
        raise Exception(f"Invalid version number: {version_number}")

    create_and_push_release_tag(repo=repo, version_number=version_number)
    create_and_push_release_branch(repo=repo, version_number=version_number)


def guess_next_version(repo: Repo) -> str:
    tags = sorted(repo.tags, key=lambda t: t.commit.committed_datetime, reverse=True)
    for tag in tags:
        if repo.is_ancestor(tag.commit, repo.heads.main.commit):
            last_tag = tag.name
            break
    if last_tag is None:
        raise Exception("No previous tags found")

    major, minor, patch = map(int, last_tag.split("."))
    minor += 1
    return f"{major}.{minor}.{patch}"


def create_and_push_release_tag(repo: Repo, version_number: str) -> None:
    release_tag = repo.create_tag(version_number)
    push_info = repo.remote("origin").push(release_tag)
    if push_info[0].flags & push_info[0].ERROR:
        raise Exception(f"Failed to push tags: {push_info[0].summary}")
    print(f"Release tag pushed: {version_number}")


def create_and_push_release_branch(repo: Repo, version_number: str) -> None:
    branch_name = f"release/v{version_number}"
    new_branch = repo.create_head(branch_name, repo.heads["main"])
    new_branch.checkout()
    push_info = repo.remote("origin").push(new_branch)
    if push_info[0].flags & push_info[0].ERROR:
        raise Exception(f"Failed to push branch: {push_info[0].summary}")
    print(f"Release branch pushed: {branch_name}")


if __name__ == "__main__":
    app()