import uuid
import time
import random
from typing import Optional, Dict, Any
from .schemas import MNB, EvidencePack
from .omega import OmegaGate
from .ledger import Ledger

class TruthEngine:
    """O Coração do MatVerse: Processamento em 19 estágios com governança"""
    
    STAGES = [
        "Ingestão", "Extração de Intenção", "Análise Semântica", 
        "Normalização M=PSNOL", "Verificação de Coerência (ψ)",
        "Cálculo de Risco (CVaR)", "Ancoragem PoLE", "Simulação GHDL",
        "Avaliação Ω-GATE", "Geração de Código", "Refinamento Iterativo",
        "Auditoria On-chain", "Validação de Reuso", "Teste de Estresse",
        "Sincronização de Ledger", "Consolidação de Blocos",
        "Geração de Evidence Pack", "Replay de Segurança", "Finalização"
    ]

    def __init__(self):
        self.ledger = Ledger()
        self.omega_gate = OmegaGate()
        self.active_mnbs: Dict[str, MNB] = {}

    def ingest(self, intent: str) -> MNB:
        mnb_id = str(uuid.uuid4())[:8]
        mnb = MNB(id=mnb_id, intent=intent)
        self.active_mnbs[mnb_id] = mnb
        self.ledger.append({"action": "INGEST", "mnb_id": mnb_id, "intent": intent})
        return mnb

    def advance(self, mnb_id: str) -> Dict[str, Any]:
        mnb = self.active_mnbs.get(mnb_id)
        if not mnb or mnb.current_stage >= len(self.STAGES):
            return {"status": "FINISHED", "stage": len(self.STAGES)}

        stage_name = self.STAGES[mnb.current_stage]
        
        # Simulação de processamento por estágio
        metrics = {
            "psi": round(random.uniform(0.85, 0.99), 4),
            "theta_hat": round(random.uniform(0.80, 0.98), 4),
            "cvar": round(random.uniform(0.01, 0.05), 4),
            "pole": 1.0 if random.random() > 0.2 else 0.0
        }
        
        omega = self.omega_gate.calculate(metrics)
        status = self.omega_gate.evaluate(omega)
        
        result = {
            "stage_index": mnb.current_stage,
            "stage_name": stage_name,
            "metrics": metrics,
            "omega": omega,
            "status": status,
            "timestamp": time.time()
        }
        
        mnb.stages_history.append(result)
        mnb.current_stage += 1
        mnb.status = status
        
        self.ledger.append({"action": "ADVANCE", "mnb_id": mnb_id, "stage": stage_name, "omega": omega})
        
        return result

    def generate_evidence_pack(self, mnb_id: str) -> Optional[EvidencePack]:
        mnb = self.active_mnbs.get(mnb_id)
        if not mnb: return None
        
        blocks = {
            "Resumo Trivial": f"O MNB {mnb.id} processou a intenção '{mnb.intent}' com sucesso através de 19 estágios.",
            "Matemática Robusta": f"Ω final atingido: {mnb.stages_history[-1]['omega'] if mnb.stages_history else 'N/A'}",
            "Governança": f"Status de maturidade: {mnb.status}",
            "Auditabilidade": f"Ledger Hash: {self.ledger.get_last_hash()}"
        }
        
        pack = EvidencePack(mnb_id=mnb_id, blocks=blocks, ledger_hash=self.ledger.get_last_hash())
        self.ledger.append({"action": "GENERATE_PACK", "mnb_id": mnb_id, "pack_hash": pack.pack_hash})
        return pack