import os
import re
import argparse

def process_file(file_path):
    """
    Lit un fichier .md, remplace les séquences de mots/phrases séparées par '/' qui commencent par 'Univers étendue'
    par le dernier mot/phrase, et écrit le fichier modifié.
    """
    with open(file_path, 'r', encoding='utf-8') as f:
        content = f.read()
    
    # Pattern : commence par 'Univers étendue' suivi d'au moins un '/', où chaque partie peut contenir des espaces mais pas de '/'
    # Ex: "Univers étendue/Personnages/Secondaires/Namisse"
    pattern = r'Univers étendue(?:/[^/]+)+'
    
    def replace_chain(match):
        # Prend le dernier élément après le dernier '/'
        last_part = match.group(0).rsplit('/', 1)[-1].strip()
        return last_part
    
    new_content = re.sub(pattern, replace_chain, content)
    
    with open(file_path, 'w', encoding='utf-8') as f:
        f.write(new_content)
    
    print(f"Séquences remplacées dans : {file_path}")

def main():
    parser = argparse.ArgumentParser(description="Parcourt un dossier et ses sous-dossiers, remplace les chaînes 'Univers étendue/mot1/mot2/mot3' par 'mot3' dans tous les fichiers .md.")
    parser.add_argument('dossier', help="Chemin du dossier à parcourir (ex: '/chemin/vers/votre/vault')")
    args = parser.parse_args()
    
    dossier = args.dossier
    
    if not os.path.exists(dossier):
        print(f"Erreur : Le dossier '{dossier}' n'existe pas.")
        return
    
    # Parcours récursif des fichiers .md
    processed_count = 0
    for root, dirs, files in os.walk(dossier):
        for file in files:
            if file.endswith('.md'):
                file_path = os.path.join(root, file)
                try:
                    process_file(file_path)
                    processed_count += 1
                except Exception as e:
                    print(f"Erreur lors du traitement de {file_path} : {e}")
    
    print(f"\nTerminé ! {processed_count} fichiers .md traités.")

if __name__ == "__main__":
    main()