V1.0 de GameurBot (Metrics surveillance des perfs)

bot.py

@@ -0,0 +1,327 @@
+"""
+bot.py — Bot Discord de monitoring serveur.
+Lance simultanément :
+  • Un serveur HTTP (aiohttp) qui reçoit les métriques de l'agent
+  • Le bot Discord qui met à jour les messages épinglés dans les guildes configurées
+
+Lancement : python bot.py
+"""
+
+import asyncio
+import json
+import os
+import logging
+from datetime import datetime, timezone
+from pathlib import Path
+
+import discord
+from discord.ext import commands, tasks
+from aiohttp import web
+
+from status_engine import StatusEngine, AnalysisResult, STATUS_OK, STATUS_WARNING, STATUS_CRITICAL
+
+# ─── Logging ──────────────────────────────────────────────────────────────────
+
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s [%(levelname)s] %(message)s",
+    datefmt="%H:%M:%S",
+)
+log = logging.getLogger("bot")
+
+# ─── Config ───────────────────────────────────────────────────────────────────
+
+CONFIG_PATH = Path("config.json")
+
+def load_config() -> dict:
+    with open(CONFIG_PATH, encoding="utf-8") as f:
+        return json.load(f)
+
+def save_config(cfg: dict):
+    with open(CONFIG_PATH, "w", encoding="utf-8") as f:
+        json.dump(cfg, f, indent=2, ensure_ascii=False)
+
+# ─── Bot ──────────────────────────────────────────────────────────────────────
+
+class StatusBot(commands.Bot):
+
+    def __init__(self, cfg: dict):
+        intents = discord.Intents.default()
+        intents.guilds = True
+        super().__init__(command_prefix="!", intents=intents)
+
+        self.cfg     = cfg
+        self.engine  = StatusEngine()
+
+        # guild_id (str) → {"channel_id": int, "message_id": int | None}
+        self.targets: dict[str, dict] = cfg.get("guilds", {})
+
+        self._api_runner: web.AppRunner | None = None
+
+    # ── Lifecycle ─────────────────────────────────────────────────────────────
+
+    async def setup_hook(self):
+        # On lance l'API et la boucle AVANT la synchro des commandes (qui peut être longue)
+        asyncio.create_task(self._start_api_server())
+        self.update_loop.start()
+        log.info("Services de monitoring démarrés.")
+
+        await self.tree.sync()
+        log.info("Slash commands synced.")
+
+    async def on_ready(self):
+        log.info(f"Connecté en tant que {self.user} (ID {self.user.id})")
+        await self.change_presence(
+            activity=discord.Activity(type=discord.ActivityType.watching, name="l'état du serveur")
+        )
+
+    # ── API HTTP (réception métriques) ────────────────────────────────────────
+
+    async def _start_api_server(self):
+        app = web.Application()
+        app.router.add_post("/metrics", self._handle_metrics)
+        runner = web.AppRunner(app)
+        await runner.setup()
+        site = web.TCPSite(runner, "0.0.0.0", self.cfg.get("api_port", 8765))
+        await site.start()
+        self._api_runner = runner
+        log.info(f"API HTTP en écoute sur le port {self.cfg.get('api_port', 8765)}")
+
+    async def _handle_metrics(self, request: web.Request) -> web.Response:
+        try:
+            data = await request.json()
+        except Exception:
+            return web.Response(status=400, text="JSON invalide")
+
+        if data.get("api_key") != self.cfg.get("api_secret"):
+            return web.Response(status=403, text="Clé API invalide")
+
+        log.info(f"Métriques reçues de l'agent (CPU: {data.get('system', {}).get('cpu_percent') or '?'}%)")
+        self.engine.push(data)
+        return web.Response(status=200, text="ok")
+
+    # ── Boucle de mise à jour ─────────────────────────────────────────────────
+
+    @tasks.loop(seconds=20)
+    async def update_loop(self):
+        if not self.engine.has_data:
+            return
+
+        result = self.engine.analyse()
+        if result is None:
+            return
+
+        embed = self._build_embed(result)
+
+        for guild_id_str, target in self.targets.items():
+            if not isinstance(target, dict):
+                continue
+
+            channel_id = target.get("channel_id")
+            if not channel_id:
+                continue
+
+            channel = self.get_channel(channel_id)
+            if not channel:
+                try:
+                    channel = await self.fetch_channel(channel_id)
+                except Exception as exc:
+                    log.warning(f"Impossible de récupérer le salon {channel_id}: {exc}")
+                    continue
+
+            msg_id = target.get("message_id")
+
+            try:
+                if msg_id:
+                    # Mise à jour du message existant
+                    msg = await channel.fetch_message(msg_id)
+                    await msg.edit(embed=embed)
+                else:
+                    # Premier envoi : on envoie et on épingle
+                    msg = await channel.send(embed=embed)
+                    await msg.pin()
+                    target["message_id"] = msg.id
+                    save_config(self.cfg)
+                    log.info(f"Message épinglé dans #{channel.name} ({guild_id_str})")
+
+            except discord.NotFound:
+                # Le message a été supprimé — on recrée
+                target["message_id"] = None
+                save_config(self.cfg)
+            except discord.Forbidden:
+                log.warning(f"Permissions insuffisantes dans #{channel.name}")
+            except Exception as exc:
+                log.error(f"Erreur update guilde {guild_id_str}: {exc}")
+
+    @update_loop.before_loop
+    async def before_loop(self):
+        await self.wait_until_ready()
+
+    # ── Construction de l'embed ───────────────────────────────────────────────
+
+    def _build_embed(self, r: AnalysisResult) -> discord.Embed:
+        COLOR = {
+            STATUS_OK:       0x2ECC71,   # vert
+            STATUS_WARNING:  0xF39C12,   # orange
+            STATUS_CRITICAL: 0xE74C3C,   # rouge
+        }
+
+        TITLE = {
+            STATUS_OK:       "🟢 Serveur — Tout est normal",
+            STATUS_WARNING:  "🟡 Serveur — Charge élevée",
+            STATUS_CRITICAL: "🔴 Serveur — Surchargé",
+        }
+
+        embed = discord.Embed(
+            title=TITLE[r.status],
+            color=COLOR[r.status],
+            timestamp=datetime.now(timezone.utc),
+        )
+        embed.set_footer(text="Dernière mise à jour")
+
+        # ── Ressources globales ──
+        cpu_bar  = self._bar(r.cpu,      100)
+        ram_bar  = self._bar(r.ram_pct,  100)
+        swap_bar = self._bar(r.swap_pct, 100)
+
+        temp_str = f"  🌡️ {r.cpu_temp}°C" if r.cpu_temp else ""
+
+        resources = (
+            f"`CPU ` {cpu_bar} **{r.cpu:.0f}%**{temp_str}\n"
+            f"`RAM ` {ram_bar} **{r.ram_pct:.0f}%** "
+            f"({r.ram_used_gb:.1f}/{r.ram_total_gb:.1f} GB)\n"
+        )
+        if r.swap_pct > 5:
+            resources += f"`SWAP` {swap_bar} **{r.swap_pct:.0f}%**\n"
+
+        resources += (
+            f"\n📈 Charge (1m / 5m / 15m) : "
+            f"`{r.load_avg[0]:.2f}` / `{r.load_avg[1]:.2f}` / `{r.load_avg[2]:.2f}`"
+        )
+
+        embed.add_field(name="📊 Ressources", value=resources, inline=False)
+
+        # ── Services ──
+        running  = [s for s in r.services if s.running]
+        stopped  = [s for s in r.services if not s.running]
+
+        if running:
+            lines = []
+            for svc in running:
+                cpu_indicator = ""
+                if svc.cpu >= 40:
+                    cpu_indicator = " 🔥"
+                elif svc.cpu >= 20:
+                    cpu_indicator = " ⚡"
+                lines.append(
+                    f"▸ {svc.name}{cpu_indicator} — "
+                    f"CPU **{svc.cpu:.0f}%** | RAM **{svc.ram_gb:.1f} GB**"
+                )
+            embed.add_field(
+                name="🎮 Services actifs",
+                value="\n".join(lines),
+                inline=False
+            )
+
+        if stopped:
+            stopped_names = " • ".join(s.name for s in stopped)
+            embed.add_field(name="💤 Services arrêtés", value=stopped_names, inline=False)
+
+        # ── Cause ──
+        if r.status != STATUS_OK:
+            cause_text = "\n".join(r.cause_lines)
+            embed.add_field(name="🔍 Pourquoi ça rame ?", value=cause_text, inline=False)
+
+        # ── ETA ──
+        embed.add_field(name="⏱️ Retour à la normale", value=r.eta_text, inline=False)
+
+        return embed
+
+    def _bar(self, value: float, max_val: float, length: int = 10) -> str:
+        """Génère une barre de progression ASCII."""
+        filled = round((value / max_val) * length)
+        filled = max(0, min(length, filled))
+        bar = "█" * filled + "░" * (length - filled)
+        return f"`{bar}`"
+
+
+# ─── Slash commands ───────────────────────────────────────────────────────────
+
+def setup_commands(bot: StatusBot):
+
+    @bot.tree.command(name="status", description="Affiche l'état actuel du serveur")
+    async def cmd_status(interaction: discord.Interaction):
+        if not bot.engine.has_data:
+            await interaction.response.send_message(
+                "⏳ En attente des premières métriques de l'agent…", ephemeral=True
+            )
+            return
+
+        result = bot.engine.analyse()
+        embed  = bot._build_embed(result)
+        await interaction.response.send_message(embed=embed)
+
+    @bot.tree.command(name="setup", description="(Admin) Configure ce salon comme tableau de bord de statut")
+    @discord.app_commands.default_permissions(administrator=True)
+    async def cmd_setup(interaction: discord.Interaction):
+        guild_id = str(interaction.guild_id)
+
+        if guild_id not in bot.targets:
+            bot.targets[guild_id] = {}
+            bot.cfg["guilds"] = bot.targets
+
+        bot.targets[guild_id]["channel_id"] = interaction.channel_id
+        bot.targets[guild_id]["message_id"] = None   # forcera la recréation
+        save_config(bot.cfg)
+
+        await interaction.response.send_message(
+            f"✅ Ce salon sera utilisé pour le tableau de bord de statut.\n"
+            f"Le message épinglé apparaîtra dans les 20 secondes.",
+            ephemeral=True
+        )
+        log.info(f"Setup: guilde {guild_id} → channel {interaction.channel_id}")
+
+    @bot.tree.command(name="services", description="Détail de tous les services détectés")
+    async def cmd_services(interaction: discord.Interaction):
+        if not bot.engine.has_data:
+            await interaction.response.send_message("⏳ Pas encore de données.", ephemeral=True)
+            return
+
+        result = bot.engine.analyse()
+        lines  = []
+        for svc in result.services:
+            status_icon = "🟢" if svc.running else "⚫"
+            if svc.running:
+                lines.append(f"{status_icon} **{svc.name}** — CPU {svc.cpu:.0f}% | RAM {svc.ram_gb:.2f} GB")
+            else:
+                lines.append(f"{status_icon} **{svc.name}** — arrêté")
+
+        embed = discord.Embed(
+            title="🖥️ État des services",
+            description="\n".join(lines) or "Aucun service détecté",
+            color=0x3498DB,
+            timestamp=datetime.now(timezone.utc),
+        )
+        await interaction.response.send_message(embed=embed, ephemeral=True)
+
+
+# ─── Entry point ──────────────────────────────────────────────────────────────
+
+def main():
+    if not CONFIG_PATH.exists():
+        print(f"[ERREUR] Fichier {CONFIG_PATH} introuvable. Copie config.example.json → config.json et remplis-le.")
+        return
+
+    cfg = load_config()
+    token = cfg.get("discord_token") or os.environ.get("DISCORD_TOKEN")
+    if not token:
+        print("[ERREUR] Aucun token Discord dans config.json (clé 'discord_token') ni en variable d'env DISCORD_TOKEN.")
+        return
+
+    bot = StatusBot(cfg)
+    setup_commands(bot)
+    bot.run(token, log_handler=None)
+
+
+if __name__ == "__main__":
+    main()