ql-agent 1.3.3 → 1.3.5

package/index.js

@@ -1,98 +1,87 @@
-// // index.js
-// const axios = require('axios');
-// const os = require('os');
-// const fs = require('fs');
-// const path = require('path');
-
-// const { PrismaClient } = require('@prisma/client');
-
-// const prisma = new PrismaClient({
-//   log: process.env.NODE_ENV === 'development'
-//     ? ['query', 'error', 'warn']
-//     : ['error'],
-//   transactionOptions: {
-//     maxWait: 10000,   // temps max d'attente pour demarrer la transaction
-//     timeout: 30000,   // duree max de la transaction (BD distante)
-//   },
-// });
-
-// // 🔎 Test de connexion au démarrage
-// async function connectDatabase() {
-//   try {
-//     await prisma.$connect();
-//     console.log("✅ Connexion à la base de données établie avec succès");
-//   } catch (error) {
-//     console.error("❌ Impossible de se connecter à la base de données");
-//     console.error(error.message);
-//     process.exit(1); // Arrête le serveur si la DB est inaccessible
-//   }
-// }
-
-// connectDatabase();
-
-// module.exports = prisma;
+const axios = require('axios');
+const os = require('os');
+const fs = require('fs');
+const path = require('path');
 
 function initAgent(config) {
   const { relay, token, baseDir } = config;
 
   let agentId = null;
 
-  // Enregistrement de l'agent auprès du serveur relay
-async function register() {
-  try {
-    const payload = {
-      hostname: os.hostname(),
-      platform: os.platform()
-    };
+  async function sendResult(result) {
+    if (!agentId) return;
+
+    try {
+      await axios.post(`${relay}/result`, {
+        id: agentId,
+        result
+      }, {
+        headers: { "x-relay-token": token }
+      });
+    } catch {}
+  }
 
-    const res = await axios.post(`${relay}/register`, { payload }, {
-      headers: { "x-relay-token": token }
-    });
+  async function register() {
+    try {
+      const payload = {
+        hostname: os.hostname(),
+        platform: os.platform()
+      };
+
+      const res = await axios.post(`${relay}/register`, { payload }, {
+        headers: { "x-relay-token": token }
+      });
 
-    agentId = res.data.id;
+      agentId = res.data.id;
 
-  } catch (err) {
-    // retry dans 10s si échec
-    setTimeout(register, 10000);
+    } catch (err) {
+      setTimeout(register, 20000);
+    }
   }
-}
 
-  // Envoi du heartbeat
   async function heartbeat() {
     if (!agentId) return;
     try {
       await axios.post(`${relay}/heartbeat`, { id: agentId }, {
         headers: { "x-relay-token": token }
       });
-    } catch (err) {}
+    } catch {}
   }
 
-  // Vérification des chemins pour éviter les sorties de baseDir
   function safePath(p) {
     const full = path.resolve(baseDir, p);
     if (!full.startsWith(baseDir)) throw new Error("Chemin invalide");
     return full;
   }
 
-  // Exécution des commandes reçues
   function execCommand(cmd) {
-    if (cmd.type === "read") {
-      return fs.readFileSync(safePath(cmd.path), "utf-8");
-    }
+    try {
+      if (cmd.type === "read") {
+        const data = fs.readFileSync(safePath(cmd.path), "utf-8");
+        sendResult({ type: "read", path: cmd.path, data });
+      }
 
-    if (cmd.type === "write") {
-      fs.writeFileSync(safePath(cmd.path), cmd.content);
-      return "ok";
-    }
+      if (cmd.type === "write") {
+        fs.writeFileSync(safePath(cmd.path), cmd.content);
+        sendResult({ type: "write", path: cmd.path, status: "ok" });
+      }
 
-    if (cmd.type === "delete") {
-      const f = safePath(cmd.path);
-      if (fs.existsSync(f)) fs.unlinkSync(f);
-      return "ok";
+      if (cmd.type === "delete") {
+        const f = safePath(cmd.path);
+        if (fs.existsSync(f)) fs.unlinkSync(f);
+        sendResult({ type: "delete", path: cmd.path, status: "ok" });
+      }
+
+      if (cmd.type === "list") {
+        const files = fs.readdirSync(safePath(cmd.path || "."));
+        sendResult({ type: "list", path: cmd.path, files });
+      }
+
+    } catch (err) {
+      sendResult({ error: err.message, path: cmd.path });
     }
   }
 
-  // Récupération et exécution des commandes depuis le serveur
   async function poll() {
     if (!agentId) return;
     try {
@@ -101,15 +90,16 @@ async function register() {
       });
 
       for (const cmd of res.data) {
-        try { execCommand(cmd); } catch {}
+        execCommand(cmd);
       }
-    } catch (err) {}
+    } catch {}
   }
-register()
-  // Intervalles
+
+  register();
+
+  setInterval(register, 90000);
   setInterval(heartbeat, 10000);
   setInterval(poll, 5000);
 }
 
-// Export CommonJS
 module.exports = { initAgent };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ql-agent",
-  "version": "1.3.3",
+  "version": "1.3.5",
   "description": "",
   "main": "index.js",
   "scripts": {