#!/usr/bin/env python3
"""Validate release versions before starting native builds."""

import argparse
import os
import re
import tomllib
from pathlib import Path


def msix_version(match: re.Match) -> str:
    """Microsoft Store package version, ordered like the Android version codes.

    The Store requires Major.Minor.Build.0 with a nonzero first part and parts
    up to 65535, so a 0.x version cannot keep its own numbers. Major is offset
    by one, Minor holds minor and patch, and Build orders the channels like
    scripts/android-version-code.py: alpha.N < beta.N < rc.N < the release.
    0.1.0-alpha.8 becomes 1.100.8.0 and 0.1.0 becomes 1.100.9999.0.
    """
    major, minor, patch = map(int, match.group(1, 2, 3))
    stage, number = match.group(4, 5)
    if major > 20 or minor > 99 or patch > 99 or (number and int(number) > 999):
        raise ValueError("Store versions support major 0–20, minor/patch 0–99, and prerelease 1–999")
    build = {"alpha": 0, "beta": 1000, "rc": 2000}.get(stage, 9999) + int(number or 0)
    return f"{major + 1}.{minor * 100 + patch}.{build}.0"


def metadata(version: str, tag: str | None = None) -> dict[str, str]:
    # Keep native package versions predictable; add more channels deliberately.
    match = re.fullmatch(
        r"(0|[1-9][0-9]*)\.(0|[1-9][0-9]*)\.(0|[1-9][0-9]*)"
        r"(?:-(alpha|beta|rc)\.([1-9][0-9]*))?",
        version,
    )
    if not match:
        raise ValueError("Use X.Y.Z or X.Y.Z-{alpha,beta,rc}.N in Cargo.toml")
    if tag is not None and tag != f"v{version}":
        raise ValueError(f"Tag {tag!r} must match Cargo.toml version: v{version}")
    return {
        "version": version,
        "native_version": ".".join(match.group(1, 2, 3)),
        "msix_version": msix_version(match),
        "prerelease": str(match.group(4) is not None).lower(),
    }


def main() -> None:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--tag")
    args = parser.parse_args()
    with Path("Cargo.toml").open("rb") as manifest:
        version = tomllib.load(manifest)["package"]["version"]
    for path in (Path("Cargo.lock"), Path("platform/android/Cargo.lock")):
        with path.open("rb") as lockfile:
            packages = tomllib.load(lockfile)["package"]
        if not any(p["name"] == "flectar-mail" and p["version"] == version for p in packages):
            raise SystemExit(f"{path} does not match the root Cargo.toml; refresh its workspace lockfile")
    try:
        values = metadata(version, args.tag)
    except ValueError as error:
        raise SystemExit(str(error)) from error
    output = "".join(f"{key}={value}\n" for key, value in values.items())
    print(output, end="")
    if output_path := os.environ.get("GITHUB_OUTPUT"):
        with open(output_path, "a", encoding="utf-8") as stream:
            stream.write(output)


if __name__ == "__main__":
    main()