Update src/terminal.js

src/terminal.js

@@ -1,54 +1,51 @@
-import http from 'http';
-import { WebSocketServer } from 'ws';
-import fs from 'fs';
-import path from 'path';
-import { fileURLToPath } from 'url';
-import { handleTerminalConnection, setSharedTerminalMode } from 'terminal.js';
-
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = path.dirname(__filename);
-
-// Config
-setSharedTerminalMode(false); // Set this to false to allow a shared session
-const port = 6060;
-
-const server = http.createServer((req, res) => {
-    if (req.method === 'GET') {
-        const routeName = req.url.slice(1);
-        const assetObj = {
-            '': { file: "index.html", contentType: "text/html" },
-            'client.js': { file: "client.js", contentType: "text/javascript" }
-        }[routeName];
-
-        if (!assetObj) {
-            res.writeHead(404, { 'Content-Type': 'text/plain' });
-            return res.end('Path not found');
-        }
+import os from 'os';
+import pty from 'node-pty';
+
+let sharedPtyProcess = null;
+let sharedTerminalMode = false;
+
+const shell = os.platform() === 'win32' ? 'powershell.exe' : 'bash';
+
+const spawnShell = () => {
+    return pty.spawn(shell, [], {
+        name: 'xterm-color',
+        env: process.env,
+    });
+};
 
-        const filePath = path.join(__dirname, assetObj.file);
-
-        fs.readFile(filePath, (err, data) => {
-            if (err) {
-                res.writeHead(500, { 'Content-Type': 'text/plain' });
-                res.end('Failed to load file');
-            } else {
-                res.writeHead(200, { 'Content-Type': assetObj.contentType });
-                res.end(data);
-            }
-        });
+export const setSharedTerminalMode = (useSharedTerminal) => {
+    sharedTerminalMode = useSharedTerminal;
+    if (sharedTerminalMode && !sharedPtyProcess) {
+        sharedPtyProcess = spawnShell();
     }
-});
+};
 
-const wss = new WebSocketServer({ noServer: true });
+export const handleTerminalConnection = (ws) => {
+    let ptyProcess = sharedTerminalMode ? sharedPtyProcess : spawnShell();
 
-wss.on('connection', handleTerminalConnection);
+    ws.on('message', command => {
+        const processedCommand = commandProcessor(command);
+        ptyProcess.write(processedCommand);
+    });
 
-server.on('upgrade', (request, socket, head) => {
-    wss.handleUpgrade(request, socket, head, (ws) => {
-        wss.emit('connection', ws, request);
+    ptyProcess.on('data', (rawOutput) => {
+        const processedOutput = outputProcessor(rawOutput);
+        ws.send(processedOutput);
     });
-});
 
-server.listen(port, () => {
-    console.log(`HTTP and WebSocket server is running on port ${port}`);
-});
+    ws.on('close', () => {
+        if (!sharedTerminalMode) {
+            ptyProcess.kill();
+        }
+    });
+};
+
+// Utility function to process commands
+const commandProcessor = (command) => {
+    return command;
+};
+
+// Utility function to process output
+const outputProcessor = (output) => {
+    return output;
+};