import json
import random

class AICore:
    """
    Saldırı ve Savunma Yollarını Dinamik Olarak Belirleyen Çekirdek.
    """
    def __init__(self, modules, mode):
        self.modules = modules
        self.mode = mode
        # Gerçekte, bu, tensör akışı veya PyTorch üzerinde eğitilmiş bir model olurdu.
        self.success_rates = {
            "web": {"RECON_PASSIVE": 0.9, "EXPLOIT_WEB": 0.7, "LATERAL_MOVE": 0.5},
            "network": {"RECON_PASSIVE": 0.9, "AUTH_BREAKER": 0.6, "PRIVILEGE_ESCALATION": 0.8},
        }

    async def get_initial_plan(self, target_type):
        """
        Hedef tespiti sonrası ilk adımı belirler (her zaman keşif).
        """
        if self.mode == 'hacker':
             # Saldırgan modu için ilk adım
            return {"step": "RECON_PASSIVE", "risk_level": "LOW"}
        else:
            # Savunma modu için ilk adım
            return {"step": "SCAN_DEFENSE_CONFIG", "risk_level": "N/A"}
    
    async def decide_next_step(self, current_state):
        """
        Mevcut duruma (toplanan veriler, erişim seviyesi) göre sonraki en iyi adımı seçer.
        """
        target = current_state["target"]
        access = current_state["access"]
        
        # 1. Mevcut Veriyi Analiz Et
        data_keys = current_state["data"].keys()
        
        next_step = "COMPLETE" # Varsayılan: İşlem bitti.

        # 2. KARAR AĞACI SİMÜLASYONU
        if self.mode == 'hacker':
            if "RECON_PASSIVE" in data_keys and not access:
                # Keşif yapıldı, henüz erişim yok. Saldırı vektörünü seç.
                # Gerçekte burada bir ML modeli hangi modülün en yüksek P(Başarı) sağlayacağını hesaplar.
                print("[AICore] Erişim yok. Exploit vektörü aranıyor...")
                next_step = "EXPLOIT_WEB" # Web hedefi varsayımıyla

            elif access and current_state["privilege"] == "user":
                # Erişime sahip, ancak yetki düşük. Yetki yükseltme modülünü çalıştır.
                print("[AICore] Erişim var, Yetki Yükseltiliyor...")
                next_step = "PRIVILEGE_ESCALATION"

            elif access and current_state["privilege"] == "root":
                # Hem erişim hem yetki var. Gizlenme ve kalıcılık modülüne geç.
                print("[AICore] Tam Kontrol. Gizlenme ve Kalıcılık sağlanıyor...")
                next_step = "PERSISTENCE_STEALTH"
                
        elif self.mode == 'defender':
            # Basit bir savunma mantığı
            if "SCAN_DEFENSE_CONFIG" in data_keys:
                print("[AICore] Savunma Zayıflıkları Bulundu. Yamalama planı hazırlanıyor...")
                next_step = "AUTO_PATCH_VULNERABILITIES"
            
        current_state["privilege"] = "root" if next_step == "PERSISTENCE_STEALTH" else current_state["privilege"]
        
        # Olası bir döngüyü önlemek için rastgele bir 'tamamlandı' koşulu
        if random.random() < 0.05 and next_step != "COMPLETE":
             next_step = "COMPLETE"
             print("[AICore] İşlem tamamlandı (Rastgele sonlandırma).")
        
        return next_step, current_state