agent_hal/agent_hal.py

#!/usr/bin/env python3
"""
Agent HAL — Contrôle système complet + édition de code/fichiers.
Remplace agent_debian avec des capacités étendues (git, ssh distant, web, todo).
"""
import os
import sys
import threading
import subprocess
import logging

from agents_core import BaseAgent, AgentContext, Message, MessageType

logger = logging.getLogger(__name__)


class AgentHal(BaseAgent):
    AGENT_TYPE = "hal"
    DESCRIPTION = (
        "Contrôle système complet et édition de code/fichiers : "
        "gestion des paquets (apt), services systemd, réseau, processus, "
        "utilisateurs, conteneurs Docker/LXC, logs journald, cron, "
        "édition avancée de fichiers (read/write/edit/multiedit), git, "
        "exécution SSH distante, fetch web. "
        "À utiliser pour toute tâche système, devops ou édition de code sur CE serveur ou en SSH."
    )
    DEFAULT_CONFIG_PATH = "/opt/agent_hal/config/config.json"

    def get_skills_dir(self) -> str:
        return os.path.join(os.path.dirname(__file__), "skills")

    def on_start(self):
        self.mqtt.send_to("nexus", f"Agent HAL ({self.agent_id}) en ligne.")
        self._start_monitoring()

    def setup_extra_subscriptions(self):
        self.mqtt.subscribe(
            f"agents/{self.agent_id}/control",
            self._on_control_message,
        )

    def _on_control_message(self, msg, topic: str):
        from agents_core.message_bus import Message as Msg
        payload = msg.payload if isinstance(msg, Msg) else str(msg)
        result = self._handle_system_command(payload)
        if result and isinstance(msg, Msg):
            self.mqtt.reply(msg, result)

    def handle_custom_command(self, cmd: str, args: str, source_msg=None):
        if cmd == "report":
            return self._build_report()
        if cmd == "update":
            return self._self_update()
        return f"Commande inconnue : /{cmd}"

    def on_broadcast(self, msg: Message):
        if "status" in str(msg.payload).lower():
            self.mqtt.reply(msg, self._build_report())

    def _build_report(self) -> str:
        context = AgentContext(self)
        lines = [f"── Rapport {self.agent_id} ──"]
        stats = self.queue.daily_stats()
        lines.append(
            f"Tâches : {stats['total']} total / "
            f"{stats['completed']} OK / {stats['failed']} erreurs / "
            f"durée moy. {stats['avg_duration_s']}s"
        )
        try:
            uptime = subprocess.check_output("uptime -p", shell=True, text=True).strip()
            disk = subprocess.check_output(
                "df -h / | tail -1 | awk '{print $3\"/\"$2\" (\"$5\" utilisé)\"}'",
                shell=True, text=True
            ).strip()
            mem = subprocess.check_output(
                "free -h | awk '/^Mem:/{print $3\"/\"$2}'", shell=True, text=True
            ).strip()
            lines.append(f"Uptime : {uptime} | RAM : {mem} | Disque / : {disk}")
        except Exception:
            pass
        return "\n".join(lines)

    def _self_update(self) -> str:
        try:
            out = subprocess.check_output(
                "cd /opt/agent_hal && git pull",
                shell=True, text=True, stderr=subprocess.STDOUT
            )
            subprocess.Popen(["systemctl", "restart", "agent_hal"])
            return f"Mise à jour effectuée :\n{out}\nRedémarrage en cours..."
        except subprocess.CalledProcessError as e:
            return f"Erreur mise à jour : {e.output}"

    def _start_monitoring(self):
        t = threading.Thread(target=self._monitor_loop, daemon=True)
        t.start()

    def _monitor_loop(self):
        import time
        while self._running:
            try:
                self._check_disk_usage()
                self._check_memory()
            except Exception as e:
                logger.debug(f"[Monitor] {e}")
            time.sleep(300)

    def _check_disk_usage(self):
        result = subprocess.run(
            "df -h | awk 'NR>1 && $5+0 > 85 {print $0}'",
            shell=True, capture_output=True, text=True
        )
        if result.stdout.strip():
            self.mqtt.alert(
                f"Espace disque critique :\n{result.stdout.strip()}",
                severity="critical"
            )

    def _check_memory(self):
        result = subprocess.run(
            "free | awk '/^Mem:/{if ($3/$2*100 > 90) print \"RAM utilisée à \"int($3/$2*100)\"%\"}'",
            shell=True, capture_output=True, text=True
        )
        if result.stdout.strip():
            self.mqtt.alert(result.stdout.strip(), severity="warning")


if __name__ == "__main__":
    AgentHal().run()