from __future__ import annotations

import argparse
import logging
import sys


def build_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(
        description="Maintainer commands for Aman release and packaging workflows."
    )
    subparsers = parser.add_subparsers(dest="command")
    subparsers.required = True

    sync_model_parser = subparsers.add_parser(
        "sync-default-model",
        help="sync managed editor model constants with benchmark winner report",
    )
    sync_model_parser.add_argument(
        "--report",
        default="benchmarks/results/latest.json",
        help="path to winner report JSON",
    )
    sync_model_parser.add_argument(
        "--artifacts",
        default="benchmarks/model_artifacts.json",
        help="path to model artifact registry JSON",
    )
    sync_model_parser.add_argument(
        "--constants",
        default="src/constants.py",
        help="path to constants module to update/check",
    )
    sync_model_parser.add_argument(
        "--check",
        action="store_true",
        help="check only; exit non-zero if constants do not match winner",
    )
    sync_model_parser.add_argument(
        "--json",
        action="store_true",
        help="print JSON summary output",
    )
    return parser


def parse_args(argv: list[str]) -> argparse.Namespace:
    return build_parser().parse_args(argv)


def _configure_logging() -> None:
    logging.basicConfig(
        stream=sys.stderr,
        level=logging.INFO,
        format="aman: %(asctime)s %(levelname)s %(message)s",
    )


def main(argv: list[str] | None = None) -> int:
    args = parse_args(list(argv) if argv is not None else sys.argv[1:])
    _configure_logging()
    if args.command == "sync-default-model":
        from aman_model_sync import sync_default_model_command

        return sync_default_model_command(args)
    raise RuntimeError(f"unsupported maintainer command: {args.command}")


if __name__ == "__main__":
    raise SystemExit(main())