ql-agent 1.0.0 → 1.2.3

package/index.js

@@ -1,39 +1,80 @@
-import axios from "axios";
-import os from "os";
-import fs from "fs";
-import path from "path";
+// // index.js
+// const axios = require('axios');
+// const os = require('os');
+// const fs = require('fs');
+// const path = require('path');
 
-export function initAgent(config) {
+// 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;
+
+function initAgent(config) {
   const { relay, token, baseDir } = config;
 
   let agentId = null;
 
-  async function register() {
+  // 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 }, {
+    const res = await axios.post(`${relay}/register`, { payload }, {
       headers: { "x-relay-token": token }
     });
 
     agentId = res.data.id;
+
+  } catch (err) {
+    // retry dans 10s si échec
+    setTimeout(register, 10000);
   }
+}
 
+  // Envoi du heartbeat
   async function heartbeat() {
     if (!agentId) return;
-    await axios.post(relay + "/heartbeat", { id: agentId }, {
-      headers: { "x-relay-token": token }
-    });
+    try {
+      await axios.post(`${relay}/heartbeat`, { id: agentId }, {
+        headers: { "x-relay-token": token }
+      });
+    } catch (err) {}
   }
 
+  // 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();
+    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");
@@ -51,19 +92,25 @@ export function initAgent(config) {
     }
   }
 
+  // Récupération et exécution des commandes depuis le serveur
   async function poll() {
     if (!agentId) return;
+    try {
+      const res = await axios.post(`${relay}/fetch-commands`, { id: agentId }, {
+        headers: { "x-relay-token": token }
+      });
 
-    const res = await axios.post(relay + "/fetch-commands", { id: agentId }, {
-      headers: { "x-relay-token": token }
-    });
-
-    for (const cmd of res.data) {
-      try { execCommand(cmd); } catch {}
-    }
+      for (const cmd of res.data) {
+        try { execCommand(cmd); } catch {}
+      }
+    } catch (err) {}
   }
 
-  setInterval(register, 15000);
+  // Intervalles
+  register();
   setInterval(heartbeat, 10000);
   setInterval(poll, 5000);
 }
+
+// Export CommonJS
+module.exports = { initAgent };

package/package.json

@@ -1,15 +1,17 @@
 {
   "name": "ql-agent",
-  "version": "1.0.0",
+  "version": "1.2.3",
   "description": "",
   "main": "index.js",
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
+    "test": "echo \"Error: no test specified\" && exit 1",
+    "start": "nodemon index.js"
   },
   "keywords": [],
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "axios": "^1.14.0"
+    "axios": "^1.14.0",
+    "nodemon": "^3.1.14"
   }
 }