Update server.js

server.js

@@ -1,5 +1,5 @@
 const WebSocket = require('ws');
-const { exec } = require('child_process');
+const { spawn } = require('child_process');
 
 // Create a WebSocket server listening on port 7860
 const wss = new WebSocket.Server({ port: 7860 });
@@ -18,24 +18,30 @@ wss.on('connection', (ws) => {
         // Convert message (Buffer) to string
         const command = message.toString();
 
-        // Execute the command without sudo
-        exec(command, (error, stdout, stderr) => {
-            if (error) {
-                ws.send(`Error: ${error.message}`);
-                console.log(`LOG: ${stderr}`);
-                return;
-            }
-            if (stderr) {
-                ws.send(`stderr: ${stderr}`);
-                console.log(`LOG: ${stderr}`);
-                return;
-            }
-            if (stdout) {
-                ws.send(`stdout: ${stdout}`);
-                console.log(`LOG: ${stdout}`);
-                return;
-            }
-            //ws.send(`stdout: ${stdout}`);
+        // Split command into executable and arguments
+        const [cmd, ...args] = command.split(' ');
+
+        // Spawn a new process to execute the command
+        const child = spawn(cmd, args, { shell: true });
+
+        // Send stdout and stderr to the client
+        child.stdout.on('data', (data) => {
+            ws.send(`stdout: ${data}`);
+            console.log(`stdout: ${data}`);
+        });
+
+        child.stderr.on('data', (data) => {
+            ws.send(`stderr: ${data}`);
+            console.log(`stderr: ${data}`);
+        });
+
+        child.on('error', (error) => {
+            ws.send(`Error: ${error.message}`);
+            console.log(`Error: ${error.message}`);
+        });
+
+        child.on('close', (code) => {
+            console.log(`Child process exited with code ${code}`);
         });
     });
 
@@ -45,4 +51,4 @@ wss.on('connection', (ws) => {
     });
 });
 
-console.log('WebSocket server is running on ws://localhost:7860');
+console.log('WebSocket server is running on ws://localhost:7860');