feat: fully functional terminal for command center

terminal-server.js

@@ -0,0 +1,163 @@
+import { WebSocketServer } from 'ws';
+import http from 'http';
+import pty from 'node-pty';
+
+const server = http.createServer((req, res) => {
+    if (req.url === '/ready') {
+        res.writeHead(200, { 'Content-Type': 'text/plain' });
+        res.end('OK');
+    } else {
+        res.writeHead(404, { 'Content-Type': 'text/plain' });
+        res.end('Not Found');
+    }
+});
+
+const wss = new WebSocketServer({ server, path: '/terminal' });
+const userSessions = new Map();
+
+wss.on('connection', (ws) => {
+    const userId = generateUserId();
+    const userSession = { ws, userId, ptyProcess: null, isActive: false };
+    userSessions.set(userId, userSession);
+
+    ws.on('message', (message) => handleMessage(userSession, message));
+    ws.on('error', (err) => handleError(err, userId));
+    ws.on('close', () => handleClose(userId));
+});
+
+const messageHandlers = {
+    message: (session, data) => session.ptyProcess.write(data),
+    resize: (session, { cols, rows }) => session.ptyProcess.resize(cols, rows),
+    pause: (session) => session.ptyProcess.pause(),
+    resume: (session) => session.ptyProcess.resume(),
+    checkActive: (session, data) => {
+        if (data === 'force' && session.isActive) {
+            killPtyProcess(session.userId);
+        } else {
+            session.ws.send(session.isActive);
+        }
+    },
+    command: (session, data) => handleCommand(session.ws, data, session.userId)
+};
+
+function handleMessage(userSession, message) {
+    const parsed = parseMessage(message);
+    if (!parsed) return;
+
+    Object.entries(parsed).forEach(([key, value]) => {
+        const handler = messageHandlers[key];
+        if (handler && (userSession.isActive || key === 'checkActive' || key === 'command')) {
+            handler(userSession, value);
+        }
+    });
+}
+
+function parseMessage(message) {
+    try {
+        return JSON.parse(message);
+    } catch (e) {
+        console.error('Failed to parse message:', e);
+        return null;
+    }
+}
+
+async function handleCommand(ws, command, userId) {
+    const userSession = userSessions.get(userId);
+
+    if (userSession && userSession.isActive) {
+        await killPtyProcess(userId);
+    }
+
+    const commandString = command[0].split('\n').join(' ');
+    const timeout = extractTimeout(commandString);
+    const sshArgs = extractSshArgs(commandString);
+    const hereDocContent = extractHereDocContent(commandString);
+    const options = {
+        name: 'xterm-color',
+        cols: 80,
+        rows: 30,
+        cwd: process.env.HOME,
+        env: process.env
+    };
+
+    // NOTE: - Initiates a process within the Terminal container
+    //         Establishes an SSH connection to root@coolify with RequestTTY enabled
+    //         Executes the 'docker exec' command to connect to a specific container
+    //         If the user types 'exit', it terminates the container connection and reverts to the server.
+    const ptyProcess = pty.spawn('ssh', sshArgs.concat(['bash']), options);
+    userSession.ptyProcess = ptyProcess;
+    userSession.isActive = true;
+    ptyProcess.write(hereDocContent + '\n');
+    ptyProcess.write('clear\n');
+
+    ws.send('pty-ready');
+
+    ptyProcess.onData((data) => ws.send(data));
+
+    ptyProcess.onExit(({ exitCode, signal }) => {
+        console.error(`Process exited with code ${exitCode} and signal ${signal}`);
+        userSession.isActive = false;
+    });
+
+    if (timeout) {
+        setTimeout(async () => {
+            await killPtyProcess(userId);
+        }, timeout * 1000);
+    }
+}
+
+async function handleError(err, userId) {
+    console.error('WebSocket error:', err);
+    await killPtyProcess(userId);
+}
+
+async function handleClose(userId) {
+    await killPtyProcess(userId);
+    userSessions.delete(userId);
+}
+
+async function killPtyProcess(userId) {
+    const session = userSessions.get(userId);
+    if (!session?.ptyProcess) return false;
+
+    return new Promise((resolve) => {
+        session.ptyProcess.on('exit', () => {
+            session.isActive = false;
+            resolve(true);
+        });
+
+        session.ptyProcess.kill();
+    });
+}
+
+function generateUserId() {
+    return Math.random().toString(36).substring(2, 11);
+}
+
+function extractTimeout(commandString) {
+    const timeoutMatch = commandString.match(/timeout (\d+)/);
+    return timeoutMatch ? parseInt(timeoutMatch[1], 10) : null;
+}
+
+function extractSshArgs(commandString) {
+    const sshCommandMatch = commandString.match(/ssh (.+?) 'bash -se'/);
+    let sshArgs = sshCommandMatch ? sshCommandMatch[1].split(' ') : [];
+    sshArgs = sshArgs.map(arg => arg === 'RequestTTY=no' ? 'RequestTTY=yes' : arg);
+    if (!sshArgs.includes('RequestTTY=yes')) {
+        sshArgs.push('-o', 'RequestTTY=yes');
+    }
+    return sshArgs;
+}
+
+function extractHereDocContent(commandString) {
+    const delimiterMatch = commandString.match(/<< (\S+)/);
+    const delimiter = delimiterMatch ? delimiterMatch[1] : null;
+    const escapedDelimiter = delimiter.slice(1).trim().replace(/[/\-\\^$*+?.()|[\]{}]/g, '\\$&');
+    const hereDocRegex = new RegExp(`<< \\\\${escapedDelimiter}([\\s\\S\\.]*?)${escapedDelimiter}`);
+    const hereDocMatch = commandString.match(hereDocRegex);
+    return hereDocMatch ? hereDocMatch[1] : '';
+}
+
+server.listen(6002, () => {
+    console.log('Server listening on port 6002');
+});