ql-agent 1.3.4 → 1.3.5

package/index.js

@@ -8,63 +8,80 @@ function initAgent(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()
-    };
-
-    const res = await axios.post(`${relay}/register`, { payload }, {
-      headers: { "x-relay-token": token }
-    });
-
-    agentId = res.data.id;
-
-  } catch (err) {
-    // retry dans 10s si échec
-    console.log("register échoué, "+err)
-    setTimeout(register, 20000);
+  async function sendResult(result) {
+    if (!agentId) return;
+
+    try {
+      await axios.post(`${relay}/result`, {
+        id: agentId,
+        result
+      }, {
+        headers: { "x-relay-token": token }
+      });
+    } catch {}
+  }
+
+  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;
+
+    } 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 {
@@ -73,16 +90,16 @@ async function register() {
       });
 
       for (const cmd of res.data) {
-        try { execCommand(cmd); } catch {}
+        execCommand(cmd);
       }
-    } catch (err) {}
+    } catch {}
   }
-register()
-  // Intervalles
-  setInterval(register, 90000)
+
+  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.4",
+  "version": "1.3.5",
   "description": "",
   "main": "index.js",
   "scripts": {