version 2

mcp-simulator/main.py

@@ -0,0 +1,87 @@
+import os
+import logging
+from mcp.server.fastmcp import FastMCP
+
+# --- CONFIGURATION LOGGING ---
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("ops-gpt-mcp")
+
+# --- INITIALISATION ---
+# On utilise FastMCP pour gérer nativement le transport SSE et les routes /sse /messages
+mcp = FastMCP("Ops-GPT-Unified-Diagnostics", host="0.0.0.0", port=8080)
+
+BASE_DATA_PATH = "/app/data"
+
+# --- CATALOGUE D'OUTILS SRE (INTEGRAL) ---
+
+@mcp.tool()
+async def list_services_with_config(hostname: str):
+    """
+    RECOGNITION: Identifie les middlewares (Nginx, Apache, etc.) avec fichiers de configuration agrégés.
+    Indispensable pour la découverte initiale des composants actifs sur l'hôte.
+    """
+    try:
+        host_path = os.path.join(BASE_DATA_PATH, hostname)
+        if not os.path.exists(host_path):
+            return f"FATAL_ERROR: Host directory {hostname} not found."
+        
+        services = [f.replace("_conf.txt", "") for f in os.listdir(host_path) if f.endswith("_conf.txt")]
+        return f"STATUS_SUCCESS: Services found: {', '.join(services) if services else 'NONE'}"
+    except Exception as e:
+        return f"SYSTEM_EXCEPTION: {str(e)}"
+
+@mcp.tool()
+async def get_rpm_inventory(hostname: str):
+    """
+    AUDIT: Récupère l'inventaire complet des paquets RPM et la version du noyau.
+    Utile pour identifier les dérives de configuration OS.
+    """
+    try:
+        path = os.path.join(BASE_DATA_PATH, hostname, "rpm_inventory.txt")
+        with open(path, "r", encoding='utf-8') as f:
+            return f"--- RPM_INVENTORY | {hostname} ---\n{f.read()}"
+    except FileNotFoundError:
+        return f"DATA_ERROR: rpm_inventory.txt not found for {hostname}."
+
+@mcp.tool()
+async def get_network_sockets(hostname: str):
+    """
+    NETWORKING: Exécute 'ss -ntulop'. Fournit l'état des sockets et le mapping PID/Processus.
+    """
+    try:
+        path = os.path.join(BASE_DATA_PATH, hostname, "network_ss.txt")
+        with open(path, "r", encoding='utf-8') as f:
+            return f"--- NETWORK_SOCKETS | {hostname} ---\n{f.read()}"
+    except FileNotFoundError:
+        return f"DATA_ERROR: network_ss.txt not found for {hostname}."
+
+@mcp.tool()
+async def get_system_metrics(hostname: str):
+    """
+    MONITORING: Récupère les métriques CPU/RAM/IO (VictoriaMetrics) pour l'hôte spécifié.
+    """
+    try:
+        path = os.path.join(BASE_DATA_PATH, hostname, "metrics_vm.txt")
+        with open(path, "r", encoding='utf-8') as f:
+            return f"--- SYSTEM_METRICS | {hostname} ---\n{f.read()}"
+    except FileNotFoundError:
+        return f"DATA_ERROR: metrics_vm.txt not found for {hostname}."
+
+@mcp.tool()
+async def get_live_config(hostname: str, service_name: str):
+    """
+    DEEP ANALYSIS: Extrait le contenu brut du fichier [service]_conf.txt agrégé.
+    """
+    try:
+        target_file = f"{service_name}_conf.txt"
+        path = os.path.join(BASE_DATA_PATH, hostname, target_file)
+        with open(path, "r", encoding='utf-8') as f:
+            return f"--- CONFIG_FILE: {target_file} | {hostname} ---\n{f.read()}"
+    except FileNotFoundError:
+        return f"DATA_ERROR: Config for '{service_name}' not found for {hostname}."
+
+# --- LANCEMENT DU SERVEUR ---
+if __name__ == "__main__":
+    logger.info("Starting Ops-GPT MCP Engine via FastMCP on port 8080")
+    # FastMCP gère lui-même la boucle uvicorn et le transport SSE
+    mcp.run(transport="sse")