#!/usr/bin/env python3
from __future__ import annotations

import argparse
from pathlib import Path

from utils.update.pip import apply_updates, find_outdated_updates

from . import PROJECT_ROOT


def main() -> int:
    parser = argparse.ArgumentParser(
        description="Update exact pip pins in roles/**/*.yml to their latest release."
    )
    parser.add_argument(
        "--repo-root",
        type=Path,
        default=PROJECT_ROOT,
        help="Repository root to scan and update",
    )
    parser.add_argument(
        "--dry-run",
        action="store_true",
        help="Print planned updates without modifying files",
    )
    args = parser.parse_args()

    repo_root = args.repo_root.resolve()
    updates = find_outdated_updates(repo_root)

    if not updates:
        print("Pinned pip requirements are already up to date.")
        return 0

    print("Planned pip pin updates:")
    for update in updates:
        rel_path = update.entry.config_path.relative_to(repo_root)
        print(
            f"- {rel_path}:{update.entry.line_number} "
            f"{update.entry.package} {update.entry.version} -> {update.latest}"
        )

    if args.dry_run:
        return 0

    changed_paths = apply_updates(updates)
    if changed_paths:
        print("\nUpdated files:")
        for path in changed_paths:
            print(f"- {path.relative_to(repo_root)}")
    else:
        print("\nNo files changed.")

    return 0


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