Update server.js

server.js

@@ -2,6 +2,7 @@ const express = require('express');
 const http = require('http');
 const { Server } = require('socket.io');
 const Convert = require('ansi-to-html');
+const { exec } = require('child_process');
 const path = require('path');
 
 const app = express();
@@ -20,7 +21,7 @@ const convert = new Convert({
   escapeXML: true
 });
 
-// 🧠 logs
+// 🧠 logs memory
 const logs = [];
 const MAX_LOGS = 10000;
 
@@ -37,7 +38,7 @@ function addLog(log) {
   if (logs.length > MAX_LOGS) logs.shift();
 }
 
-// ===== STATIC FILE =====
+// ===== STATIC =====
 app.use(express.static(path.join(__dirname, 'public')));
 
 // ===== API =====
@@ -49,26 +50,71 @@ app.get('/api/logs', (req, res) => {
 io.on('connection', (socket) => {
   console.log('Client connected:', socket.id);
 
+  // receive logs from clients
   socket.on('log', (data) => {
     const log = formatLog(data);
     addLog(log);
     io.emit('log', log);
   });
 
+  // handle command execution
   socket.on('command', (data) => {
-    const cmd = data.command;
+    const cmd = data.command.trim();
 
-    const response = formatLog({
-      message: "\x1b[36mExecuted:\x1b[0m " + cmd,
-      level: "info"
+    if (!cmd) return;
+
+    // 🔒 Basic security (edit as needed)
+    const blocked = ['rm', 'shutdown', 'reboot', 'mkfs'];
+    if (blocked.some(b => cmd.includes(b))) {
+      const log = formatLog({
+        message: "\x1b[31mBlocked command!\x1b[0m",
+        level: "error"
+      });
+      addLog(log);
+      return io.emit('log', log);
+    }
+
+    // show command
+    const cmdLog = formatLog({
+      message: "\x1b[32m$ " + cmd + "\x1b[0m"
+    });
+
+    addLog(cmdLog);
+    io.emit('log', cmdLog);
+
+    // execute
+    exec(cmd, { timeout: 20000 }, (error, stdout, stderr) => {
+
+      if (stdout) {
+        const outLog = formatLog({ message: stdout });
+        addLog(outLog);
+        io.emit('log', outLog);
+      }
+
+      if (stderr) {
+        const errLog = formatLog({
+          message: "\x1b[31m" + stderr + "\x1b[0m",
+          level: "error"
+        });
+        addLog(errLog);
+        io.emit('log', errLog);
+      }
+
+      if (error) {
+        const errLog = formatLog({
+          message: "\x1b[31m" + error.message + "\x1b[0m",
+          level: "error"
+        });
+        addLog(errLog);
+        io.emit('log', errLog);
+      }
     });
 
-    addLog(response);
-    io.emit('log', response);
   });
+
 });
 
 // ===== START =====
 server.listen(PORT, () => {
-  console.log('🚀 Server running on', PORT);
+  console.log('🚀 Server running on http://localhost:' + PORT);
 });