ql-agent 1.3.4 → 1.3.6

package/index.js

@@ -8,63 +8,146 @@ 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()
-    };
+  async function sendResult(result) {
+    if (!agentId) return;
 
-    const res = await axios.post(`${relay}/register`, { payload }, {
-      headers: { "x-relay-token": token }
-    });
+    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;
+      agentId = res.data.id;
 
-  } catch (err) {
-    // retry dans 10s si échec
-    console.log("register échoué, "+err)
-    setTimeout(register, 20000);
+    } 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");
+    const full = path.resolve(baseDir, p || '.');
+    const normalizedBase = path.resolve(baseDir);
+    if (!full.startsWith(normalizedBase)) {
+      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");
-    }
+  function buildTree(dir) {
+    const stats = fs.statSync(dir);
+    const name = path.basename(dir);
 
-    if (cmd.type === "write") {
-      fs.writeFileSync(safePath(cmd.path), cmd.content);
-      return "ok";
+    if (stats.isFile()) {
+      return {
+        type: 'file',
+        name,
+        path: dir,
+        size: stats.size
+      };
     }
 
-    if (cmd.type === "delete") {
-      const f = safePath(cmd.path);
-      if (fs.existsSync(f)) fs.unlinkSync(f);
-      return "ok";
+    const children = fs.readdirSync(dir).map((entry) => {
+      return buildTree(path.join(dir, entry));
+    });
+
+    return {
+      type: 'directory',
+      name,
+      path: dir,
+      children
+    };
+  }
+
+  function execCommand(cmd) {
+    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') {
+        const target = safePath(cmd.path);
+        fs.mkdirSync(path.dirname(target), { recursive: true });
+        fs.writeFileSync(target, cmd.content);
+        sendResult({ type: 'write', path: cmd.path, status: 'ok' });
+      }
+
+      if (cmd.type === 'delete') {
+        const f = safePath(cmd.path);
+        if (fs.existsSync(f)) {
+          const stats = fs.statSync(f);
+          if (stats.isDirectory()) {
+            fs.rmSync(f, { recursive: true, force: true });
+          } else {
+            fs.unlinkSync(f);
+          }
+        }
+        sendResult({ type: 'delete', path: cmd.path, status: 'ok' });
+      }
+
+      if (cmd.type === 'list') {
+        const root = safePath(cmd.path || '.');
+        const tree = buildTree(root);
+        sendResult({ type: 'list', path: cmd.path || '.', tree });
+      }
+
+      if (cmd.type === 'mkdir') {
+        const dir = safePath(cmd.path);
+        fs.mkdirSync(dir, { recursive: true });
+        sendResult({ type: 'mkdir', path: cmd.path, status: 'ok' });
+      }
+
+      if (cmd.type === 'rename') {
+        const oldPath = safePath(cmd.path);
+        const newPath = safePath(cmd.newPath);
+        fs.mkdirSync(path.dirname(newPath), { recursive: true });
+        fs.renameSync(oldPath, newPath);
+        sendResult({
+          type: 'rename',
+          oldPath: cmd.path,
+          newPath: cmd.newPath,
+          status: 'ok'
+        });
+      }
+
+      if (cmd.type === 'touch') {
+        const file = safePath(cmd.path);
+        fs.mkdirSync(path.dirname(file), { recursive: true });
+        if (!fs.existsSync(file)) {
+          fs.writeFileSync(file, '');
+        }
+        sendResult({ type: 'touch', path: cmd.path, status: 'ok' });
+      }
+
+    } 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 +156,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, 93000);
   setInterval(heartbeat, 10000);
   setInterval(poll, 5000);
 }
 
-// Export CommonJS
-module.exports = { initAgent };
+module.exports = { initAgent };

package/package.json

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