import psutil
import socket
import logging

_log = logging.getLogger(__name__)

class BattlefieldIntel:
    def __init__(self):
        # Known common ports to monitor for potential unauthorized access
        self.suspicious_ports = {22, 23, 3389, 4444, 4445, 8080}
        self.last_threats = []

    def scan_global_surveillance(self) -> str:
        """
        Scans network connections for active sessions and maps local routing.
        In MCU terms: "track individuals' locations globally"
        """
        active_connections = 0
        listening_ports = 0
        
        try:
            for conn in psutil.net_connections(kind='inet'):
                if conn.status == 'ESTABLISHED':
                    active_connections += 1
                elif conn.status == 'LISTEN':
                    listening_ports += 1
                    
            return f"Global surveillance active. Tracking {active_connections} established connections across {listening_ports} listening vectors."
        except Exception as e:
            _log.error(f"Global surveillance failed: {e}")
            return "Global surveillance offline. Unable to penetrate local routing shields."

    def identify_threats(self) -> str:
        """
        Scans for suspicious open ports or unrecognized high-CPU processes.
        In MCU terms: "identify threats (such as hidden missile sites)"
        """
        threats = []
        
        # Scan network threats
        try:
            for conn in psutil.net_connections(kind='inet'):
                if conn.status == 'LISTEN' and conn.laddr:
                    port = getattr(conn.laddr, 'port', None)
                    if port in self.suspicious_ports:
                        threats.append(f"Suspicious listening port detected: {port}")
        except Exception:
            pass

        # Scan process threats (High CPU usage indicating rogue process)
        try:
            for proc in psutil.process_iter(['name', 'cpu_percent']):
                if proc.info['cpu_percent'] and proc.info['cpu_percent'] > 85.0:
                    threats.append(f"Hostile process detected: '{proc.info['name']}' is consuming excessive resources.")
        except Exception:
            pass
            
        self.last_threats = threats
        
        if threats:
            return "Sir, I have identified potential threats:\n" + "\n".join([f"- {t}" for t in threats])
        else:
            return "Scanners indicate no immediate threats. The airspace is clear, sir."

intel_system = BattlefieldIntel()

def get_surveillance_report() -> str:
    return intel_system.scan_global_surveillance()
    
def get_threat_report() -> str:
    return intel_system.identify_threats()