Initial commit — agent_deploy v2.0

2026-03-09 09:01:34 +00:00
commit f2da35e072
10 changed files with 768 additions and 0 deletions
@@ -0,0 +1,125 @@
+"""
+Skill DEPLOY — déployer un agent sur une machine distante ou locale.
+
+Le LLM collecte les paramètres via une conversation guidée, puis lance le déploiement.
+
+Usage LLM :
+  SKILL:deploy ARGS:start <type> <nom> <host> <user> <auth:password|key> <credential> <xmpp_jid> <xmpp_pass> <mqtt_host>
+  SKILL:deploy ARGS:local <type> <nom> <xmpp_jid> <xmpp_pass> <mqtt_host>
+  SKILL:deploy ARGS:catalog
+  SKILL:deploy ARGS:status <host>
+"""
+import json
+import os
+import sys
+
+sys.path.insert(0, "/opt/agent_deploy")
+
+from deployer import Deployer, DeployConfig, AgentCatalog
+
+DESCRIPTION = "Déployer un agent sur une machine distante (SSH) ou locale"
+USAGE = (
+    "SKILL:deploy ARGS:catalog  — liste les agents déployables\n"
+    "SKILL:deploy ARGS:start <type> <nom> <host> <user> password <mdp> <xmpp_jid> <xmpp_pass> <mqtt_host>\n"
+    "SKILL:deploy ARGS:local <type> <nom> <xmpp_jid> <xmpp_pass> <mqtt_host>"
+)
+
+
+def run(args: str, context) -> str:
+    parts = args.strip().split(None, 1)
+    action = parts[0].lower() if parts else "catalog"
+    rest   = parts[1] if len(parts) > 1 else ""
+
+    if action == "catalog":
+        catalog = AgentCatalog()
+        return catalog.summary()
+
+    if action == "local":
+        # ARGS: local <type> <nom> <xmpp_jid> <xmpp_pass> <mqtt_host>
+        p = rest.split()
+        if len(p) < 5:
+            return "Format : local <type> <nom> <xmpp_jid> <xmpp_pass> <mqtt_host>"
+        agent_type, agent_name, xmpp_jid, xmpp_pass, mqtt_host = p[0], p[1], p[2], p[3], p[4]
+
+        cfg = DeployConfig(
+            agent_type=agent_type,
+            agent_name=agent_name,
+            host="localhost",
+            ssh_user="root",
+            ssh_auth="",
+            ssh_credential="",
+            xmpp_jid=xmpp_jid,
+            xmpp_password=xmpp_pass,
+            mqtt_host=mqtt_host,
+            local=True,
+        )
+        return _do_deploy(cfg, context)
+
+    if action == "start":
+        # ARGS: start <type> <nom> <host> <user> <auth> <credential> <xmpp_jid> <xmpp_pass> <mqtt_host>
+        p = rest.split()
+        if len(p) < 9:
+            return (
+                "Format : start <type> <nom> <host> <user> password|key <credential> "
+                "<xmpp_jid> <xmpp_pass> <mqtt_host>"
+            )
+        cfg = DeployConfig(
+            agent_type=p[0],
+            agent_name=p[1],
+            host=p[2],
+            ssh_user=p[3],
+            ssh_auth=p[4],
+            ssh_credential=p[5],
+            xmpp_jid=p[6],
+            xmpp_password=p[7],
+            mqtt_host=p[8],
+            mqtt_port=int(p[9]) if len(p) > 9 else 1883,
+        )
+        return _do_deploy(cfg, context)
+
+    if action == "status":
+        host = rest.strip()
+        if not host:
+            return "Précise l'hôte."
+        import subprocess
+        try:
+            result = subprocess.run(
+                f"ssh -o ConnectTimeout=5 -o StrictHostKeyChecking=no {host} 'systemctl list-units --type=service --state=active | grep agent'",
+                shell=True, text=True, capture_output=True, timeout=15
+            )
+            return result.stdout.strip() or "Aucun agent actif trouvé sur cet hôte."
+        except Exception as e:
+            return str(e)
+
+    return "Action inconnue. Disponible : catalog, start, local, status"
+
+
+def _do_deploy(cfg: DeployConfig, context) -> str:
+    """Lance le déploiement et notifie via MQTT."""
+    messages = []
+
+    def progress(msg: str):
+        messages.append(msg)
+        # Notifie en temps réel via MQTT
+        context.mqtt.send_to(
+            "nexus",
+            f"[Deploy {cfg.agent_name}] {msg}",
+            msg_type="result",
+        )
+
+    deployer = Deployer(cfg, progress_cb=progress)
+    success, result = deployer.deploy()
+
+    # Notification finale
+    if success:
+        # Notifie Nexus de l'enregistrement du nouvel agent
+        registration = json.dumps({
+            "agent_id": cfg.agent_name,
+            "agent_type": cfg.agent_type,
+            "host": cfg.host,
+            "xmpp_jid": cfg.xmpp_jid,
+            "mqtt_inbox": f"agents/{cfg.agent_name}/inbox",
+        })
+        context.mqtt.publish_raw("agents/nexus/inbox", registration)
+
+    return result