@fenwave/agent 1.1.0

package/docker-actions/general.js

@@ -0,0 +1,171 @@
+import os from 'os';
+import { formatUptime } from '../helper-functions.js';
+import { docker } from './containers.js';
+import agentStore from '../store/agentStore.js';
+import { loadSession } from '../auth.js';
+import agentSessionStore from '../store/agentSessionStore.js';
+
+// Version from package.json
+import packageJson from '../package.json' with { type: 'json' };
+const { version } = packageJson;
+
+// Agent informations - startTime will be loaded from store or set when server starts
+let agentStartTime = new Date();
+const agentInfo = {
+  version,
+  hostname: os.hostname(),
+  startTime: agentStartTime,
+  cpus: os.cpus().length,
+  memory: {
+    total: Math.round(os.totalmem() / (1024 * 1024)), // MB
+    free: Math.round(os.freemem() / (1024 * 1024)), // MB
+  },
+};
+
+/**
+ * Initialize agent start time - called when server actually starts
+ * Always creates a fresh start time when the server starts
+ */
+async function initializeAgentStartTime() {
+  try {
+    // Always create a fresh start time when the server starts
+    agentStartTime = new Date();
+    agentInfo.startTime = agentStartTime;
+    await agentStore.saveAgentStartTime(agentStartTime);
+  } catch (err) {
+    console.error('Failed to initialize agent start time:', err);
+  }
+}
+
+async function handleGeneralAction(ws, clientId, action, payload) {
+  switch (action) {
+    case 'getAgentInfo':
+      return await handleGetAgentInfo(ws, payload);
+    case 'getDockerInfo':
+      return await handleGetDockerInfo(ws, payload);
+    case 'execInContainer':
+      return await handleExecInContainer(ws, clientId, payload);
+    default:
+      throw new Error(`Unknown general action: ${action}`);
+  }
+}
+
+async function handleGetAgentInfo(ws, payload = {}) {
+  try {
+    const currentTime = Date.now();
+    const uptimeMs = currentTime - new Date(agentInfo.startTime).getTime();
+    const formattedUptime = formatUptime(uptimeMs);
+
+    // Get session info
+    const sessionInfo = agentSessionStore.getAgentSessionInfo();
+    const session = loadSession();
+
+    const info = {
+      ...agentInfo,
+      timestamp: Date.now(),
+      uptime: formattedUptime,
+      agentId: sessionInfo.agentId,
+      userEntityRef:
+        sessionInfo.userEntityRef || (session && session.userEntityRef),
+      sessionExpiresAt:
+        sessionInfo.sessionExpiresAt || (session && session.expiresAt),
+      status: 'connected', // Agent is connected if responding
+    };
+
+    ws.send(
+      JSON.stringify({
+        type: 'agentInfo',
+        info,
+        requestId: payload.requestId,
+      })
+    );
+  } catch (error) {
+    console.error('Error getting agent info:', error);
+    ws.send(
+      JSON.stringify({
+        type: 'error',
+        error: 'Failed to get agent info: ' + error.message,
+        requestId: payload.requestId,
+      })
+    );
+  }
+}
+
+async function handleExecInContainer(ws, clientId, payload) {
+  try {
+    const { containerId, command, requestId } = payload;
+
+    const container = docker.getContainer(containerId);
+    const exec = await container.exec({
+      AttachStdin: false,
+      AttachStdout: true,
+      AttachStderr: true,
+      Tty: false,
+      Cmd: ['/bin/sh', '-c', command],
+    });
+
+    const stream = await exec.start({ hijack: true, stdin: false });
+    let output = '';
+
+    stream.on('data', (chunk) => {
+      output += chunk.toString();
+    });
+
+    stream.on('end', () => {
+      ws.send(
+        JSON.stringify({
+          type: 'execResult',
+          containerId,
+          output,
+          requestId,
+        })
+      );
+    });
+
+    stream.on('error', (err) => {
+      ws.send(
+        JSON.stringify({
+          type: 'error',
+          error: `Error executing command: ${err.message}`,
+          requestId,
+        })
+      );
+    });
+  } catch (error) {
+    console.error('Error executing command in container:', error);
+    ws.send(
+      JSON.stringify({
+        type: 'error',
+        error: 'Failed to execute command in container: ' + error.message,
+        requestId: payload.requestId,
+      })
+    );
+  }
+}
+
+async function handleGetDockerInfo(ws, payload = {}) {
+  try {
+    const info = await docker.info();
+
+    ws.send(
+      JSON.stringify({
+        type: 'dockerInfo',
+        info,
+        requestId: payload.requestId,
+      })
+    );
+  } catch (error) {
+    console.error('Error getting Docker info:', error);
+    ws.send(
+      JSON.stringify({
+        type: 'error',
+        error: 'Failed to get Docker info: ' + error.message,
+        requestId: payload.requestId,
+      })
+    );
+  }
+}
+
+export default { handleGeneralAction, agentInfo, initializeAgentStartTime };
+
+export { handleGeneralAction, agentInfo, initializeAgentStartTime };