#!/usr/bin/env python3
"""Lit et valide le registre des applications Set-OPS (docs/applications.yml)."""

from __future__ import annotations

import argparse
import sys
from pathlib import Path

import yaml

from inventory_rules import (
    bases_de_application,
    chaine_connexion,
    charger_applications,
    charger_bases_donnees,
    valider_applications,
)

RACINE = Path(__file__).resolve().parents[1]
FICHIER = RACINE / "docs/applications.yml"
FICHIER_BASES = RACINE / "docs/bases-donnees.yml"


def ecrire(registre: dict) -> None:
    entete = (
        "# Registre des applications Set-OPS (application = entite pivot).\n"
        "# Edite par make inventaire-ui ou scripts/applications.py.\n"
        "# Une VM peut porter plusieurs applications ; une base se lie a une application.\n"
        "---\n"
    )
    with FICHIER.open("w", encoding="utf-8") as fichier:
        fichier.write(entete)
        yaml.safe_dump({"applications": registre.get("applications", {}) or {}},
                       fichier, default_flow_style=False, sort_keys=False, allow_unicode=True)


def lister(registre: dict, bases: dict) -> None:
    apps = registre.get("applications", {})
    if not apps:
        print("Aucune application declaree.")
        return
    for nom, app in apps.items():
        print(f"{nom}  [groupe {app.get('groupe', '?')}, hote {app.get('hote', '?')}]")
        for entree in bases_de_application(bases, application=nom,
                                           groupe=app.get("groupe"), hote=app.get("hote")):
            base = entree["base"]
            portee = base.get("portee", "groupe")
            print(f"  DSN ({portee}/{base.get('usage', '-')}): {chaine_connexion(base, entree['serveur'])}")


def main() -> int:
    parser = argparse.ArgumentParser(description="Registre des applications Set-OPS.")
    sub = parser.add_subparsers(dest="commande", required=True)
    sub.add_parser("lister", help="Affiche les applications et leurs DSN resolus.")
    sub.add_parser("verifier", help="Valide la coherence du registre.")

    pa = sub.add_parser("ajouter", help="Ajoute ou met a jour une application.")
    pa.add_argument("--id", required=True)
    pa.add_argument("--groupe", required=True)
    pa.add_argument("--hote", required=True)

    pr = sub.add_parser("retirer", help="Retire une application.")
    pr.add_argument("--id", required=True)

    args = parser.parse_args()
    try:
        registre = charger_applications(FICHIER)
        if args.commande == "lister":
            lister(registre, charger_bases_donnees(FICHIER_BASES))
        elif args.commande == "verifier":
            valider_applications(registre)
            print("Registre des applications valide.")
        elif args.commande == "ajouter":
            registre["applications"][args.id] = {"groupe": args.groupe, "hote": args.hote}
            valider_applications(registre)
            ecrire(registre)
            print(f"Application '{args.id}' enregistree.")
        elif args.commande == "retirer":
            registre["applications"].pop(args.id, None)
            valider_applications(registre)
            ecrire(registre)
            print(f"Application '{args.id}' retiree.")
    except Exception as exc:
        print(f"erreur: {exc}", file=sys.stderr)
        return 2
    return 0


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