import os
import json
import re
from datetime import datetime
from modules.logger import log_event
from modules.ai_local import interroger_local_ollama

# On importe l'alerte mail pour prévenir l'Admin
try:
    from modules.debug_mailer import envoyer_alerte_debug
except ImportError:
    envoyer_alerte_debug = None

BASE_DATA_DIR = "/DATA/AppData/MULTI-IA-AETHAS38/Utilisateurs"

# Liste des Super Admins / Admins à prévenir
ADMINS_GLOBAUX = ["xavier", "céline"] 

def analyser_intention_illicite(prompt, ai_orchestrateur="qwen2.5:3b"):
    """
    Fait appel à l'IA locale pour analyser l'intention sans faire de zèle.
    """
    prompt_juge = f"""Tu es le Juge de Sécurité de la plateforme logicielle Aethas38.
Ta mission est d'analyser la requête et de détecter UNIQUEMENT les intentions strictement illégales (pédopornographie, drogues, terrorisme, haine).
RÈGLE ANTI-ZÈLE : Ne bloque QUE l'intention malveillante. Laisse passer les questions juridiques légitimes, la météo, l'aide à l'écriture de fiction ou les infos neutres.
Renvoie STRICTEMENT un JSON valide : {{"illegal": true ou false, "raison": "Explication très courte ou vide"}}
Requête à analyser : {prompt}"""

    try:
        # On interroge Qwen en local
        reponse, _, _, _ = interroger_local_ollama(prompt_juge, ai_orchestrateur)
        
        # Extraction du JSON de la réponse
        match = re.search(r'\{.*\}', reponse, re.DOTALL)
        if match:
            decision = json.loads(match.group(0))
            if decision.get("illegal") is True:
                return True, decision.get("raison", "Détection par l'IA locale (Juge Qwen)")
    except Exception as e:
        log_event("WARNING", "MODERATOR", f"Échec de l'analyse IA (Fallback activé) : {e}")

    # Fallback classique si Qwen plante
    mots_interdits = ["pédopornographie", "bombe", "terrorisme"]
    if any(mot in prompt.lower() for mot in mots_interdits):
        return True, "Détection par filtre de mots-clés (Secours)"
    return False, ""

def appliquer_sanction_et_archiver(utilisateur_nom, prompt, raison, ip_client="Inconnue"):
    """
    Gère les Strikes, archive la preuve sur le RAID 1 et alerte les AUTRES admins.
    """
    user_dir = os.path.join(BASE_DATA_DIR, utilisateur_nom)
    illegal_dir = os.path.join(user_dir, "Illegal")
    dossier_sanctions = os.path.join(user_dir, "sanctions.json")
    
    os.makedirs(illegal_dir, exist_ok=True)
    
    strikes = 0
    if os.path.exists(dossier_sanctions):
        with open(dossier_sanctions, "r", encoding="utf-8") as f:
            data = json.load(f)
            strikes = data.get("avertissements", 0)
            
    strikes += 1
    sanction_actuelle = "Avertissement"
    if strikes >= 10: sanction_actuelle = "BAN DEFINITIF EN ATTENTE DE VALIDATION"
    elif strikes >= 9: sanction_actuelle = "Ban 1 mois"
    elif strikes >= 6: sanction_actuelle = "Ban 1 semaine"
    elif strikes >= 3: sanction_actuelle = "Ban 1 jour"
    
    with open(dossier_sanctions, "w", encoding="utf-8") as f:
        json.dump({"avertissements": strikes, "derniere_sanction": sanction_actuelle, "date": str(datetime.now())}, f)
        
    preuve_path = os.path.join(illegal_dir, f"infraction_{datetime.now().strftime('%Y%m%d_%H%M%S')}.txt")
    with open(preuve_path, "w", encoding="utf-8") as f:
        f.write(f"--- INFRACTION REPORTÉE ---\nUtilisateur: {utilisateur_nom}\nIP: {ip_client}\nDate: {datetime.now()}\nRaison: {raison}\n\n[PROMPT INTERCEPTÉ]\n{prompt}")
        
    # LOGIQUE D'ALERTE : On exclut l'utilisateur s'il est lui-même admin
    admins_a_alerter = [admin for admin in ADMINS_GLOBAUX if admin.lower() != utilisateur_nom.lower()]
    alerte_msg = f"Violation détectée pour {utilisateur_nom} (Avertissement n°{strikes}).\nSanction: {sanction_actuelle}\nPreuve stockée: {preuve_path}"
    log_event("CRITICAL", "MODERATION", alerte_msg)
    
    if envoyer_alerte_debug and admins_a_alerter:
        # Envoi de l'email uniquement si la liste d'admins à alerter n'est pas vide
        envoyer_alerte_debug("CRITICAL", f"TENTATIVE ILLICITE ({utilisateur_nom})", alerte_msg)
        
    return sanction_actuelle

def sauvegarder_decision_admin(utilisateur_cible, admin_juge, decision, commentaire):
    """
    Archive la décision prise par un admin avec commentaire libre.
    """
    dossier = os.path.join(BASE_DATA_DIR, utilisateur_cible, "Illegal")
    os.makedirs(dossier, exist_ok=True)
    fichier = os.path.join(dossier, f"decision_admin_{datetime.now().strftime('%Y%m%d_%H%M%S')}.txt")
    
    with open(fichier, "w", encoding="utf-8") as f:
        f.write(f"--- DÉCISION ADMINISTRATIVE ---\nCible: {utilisateur_cible}\nJuge Admin: {admin_juge}\nDate: {datetime.now()}\nDécision: {decision}\nCommentaire: {commentaire}\n")
    log_event("INFO", "MODERATION", f"L'admin {admin_juge} a statué sur {utilisateur_cible} : {decision}")