import os
import subprocess

# Récupérer le nom du module (nom du dossier actuel)
module_name = os.path.basename(os.getcwd())

# Vérifier si terraform-docs est installé
try:
    subprocess.run(["terraform-docs", "--version"], check=True, stdout=subprocess.PIPE)
except FileNotFoundError:
    print("❌ terraform-docs n'est pas installé. Installez-le d'abord.")
    exit(1)

# Exécuter terraform-docs et récupérer la sortie
terraform_docs_output = subprocess.run(["terraform-docs", "markdown", "."], capture_output=True, text=True)

# Conditionner le comportement pour le README.md racine
if module_name == "racine":  # Changez "racine" par le nom du dossier racine
    # Pour la racine, on génère simplement un README propre sans les balises
    readme_content = f"# Projet Terraform\n\n{terraform_docs_output.stdout}"
else:
    # Pour les autres modules, on génère avec les balises
    readme_content = f"""<!-- BEGIN_TF_DOCS -->
# Module: {module_name}

{terraform_docs_output.stdout}
<!-- END_TF_DOCS -->
"""

# Écrire dans le fichier README.md
with open("README.md", "w", encoding="utf-8") as readme_file:
    readme_file.write(readme_content)

print(f"✅ README.md généré pour le module {module_name} !")