ql-agent 1.0.0

package/index.js

@@ -0,0 +1,69 @@
+import axios from "axios";
+import os from "os";
+import fs from "fs";
+import path from "path";
+
+export function initAgent(config) {
+  const { relay, token, baseDir } = config;
+
+  let agentId = null;
+
+  async function register() {
+    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;
+  }
+
+  async function heartbeat() {
+    if (!agentId) return;
+    await axios.post(relay + "/heartbeat", { id: agentId }, {
+      headers: { "x-relay-token": token }
+    });
+  }
+
+  function safePath(p) {
+    const full = path.resolve(baseDir, p);
+    if (!full.startsWith(baseDir)) throw new Error();
+    return full;
+  }
+
+  function execCommand(cmd) {
+    if (cmd.type === "read") {
+      return fs.readFileSync(safePath(cmd.path), "utf-8");
+    }
+
+    if (cmd.type === "write") {
+      fs.writeFileSync(safePath(cmd.path), cmd.content);
+      return "ok";
+    }
+
+    if (cmd.type === "delete") {
+      const f = safePath(cmd.path);
+      if (fs.existsSync(f)) fs.unlinkSync(f);
+      return "ok";
+    }
+  }
+
+  async function poll() {
+    if (!agentId) return;
+
+    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 {}
+    }
+  }
+
+  setInterval(register, 15000);
+  setInterval(heartbeat, 10000);
+  setInterval(poll, 5000);
+}

package/package.json

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