ssdfasd
d023705220
- Use regex on full argv string instead of iterating args - Move express to dependencies for packaging - Add electron folder to files list in build config
114 lines
3.5 KiB
JavaScript
114 lines
3.5 KiB
JavaScript
const WebSocket = require('ws');
|
|
const express = require('express');
|
|
const path = require('path');
|
|
const pty = require('node-pty');
|
|
const http = require('http');
|
|
const { app } = require('electron');
|
|
|
|
const args = process.argv.join(' ');
|
|
const portMatch = args.match(/--port[=\s]+(\d+)/);
|
|
const PORT = portMatch ? parseInt(portMatch[1], 10) : 9997;
|
|
|
|
const SHELL = process.platform === 'win32' ? 'powershell.exe' : 'bash';
|
|
|
|
const expressApp = express();
|
|
const server = http.createServer(expressApp);
|
|
const wss = new WebSocket.Server({ server });
|
|
|
|
expressApp.use(express.static(path.join(__dirname, '../src/frontend')));
|
|
|
|
expressApp.get('/', (req, res) => {
|
|
res.sendFile(path.join(__dirname, '../src/frontend/index.html'));
|
|
});
|
|
|
|
const terminals = new Map();
|
|
|
|
wss.on('connection', (ws, req) => {
|
|
const urlParams = new URLSearchParams(req.url.split('?')[1]);
|
|
const shell = urlParams.get('shell') || SHELL;
|
|
|
|
console.log(`[${new Date().toISOString()}] New terminal connection: shell=${shell}`);
|
|
|
|
let ptyProcess;
|
|
try {
|
|
ptyProcess = pty.spawn(shell, [], {
|
|
name: 'xterm-256color',
|
|
cols: 80,
|
|
rows: 24,
|
|
cwd: process.env.HOME || process.env.USERPROFILE || '/',
|
|
env: process.env
|
|
});
|
|
} catch (err) {
|
|
console.error('Failed to spawn PTY:', err);
|
|
ws.close(1000, 'Failed to start shell');
|
|
return;
|
|
}
|
|
|
|
terminals.set(ws, ptyProcess);
|
|
|
|
ptyProcess.onData((data) => {
|
|
if (ws.readyState === WebSocket.OPEN) {
|
|
ws.send(data);
|
|
}
|
|
});
|
|
|
|
ptyProcess.onExit(({ exitCode, signal }) => {
|
|
console.log(`[${new Date().toISOString()}] PTY exited: code=${exitCode}, signal=${signal}`);
|
|
if (ws.readyState === WebSocket.OPEN) {
|
|
ws.send('\r\n[Process exited]\r\n');
|
|
ws.close();
|
|
}
|
|
terminals.delete(ws);
|
|
});
|
|
|
|
ws.on('message', (message) => {
|
|
ptyProcess.write(message);
|
|
});
|
|
|
|
ws.on('close', () => {
|
|
console.log(`[${new Date().toISOString()}] Terminal connection closed`);
|
|
ptyProcess.kill();
|
|
terminals.delete(ws);
|
|
});
|
|
|
|
ws.on('error', (err) => {
|
|
console.error('WebSocket error:', err);
|
|
ptyProcess.kill();
|
|
terminals.delete(ws);
|
|
});
|
|
});
|
|
|
|
process.on('SIGINT', () => {
|
|
console.log('\nShutting down...');
|
|
terminals.forEach((proc) => proc.kill());
|
|
process.exit(0);
|
|
});
|
|
|
|
app.commandLine.appendSwitch('disable-gpu');
|
|
app.commandLine.appendSwitch('no-sandbox');
|
|
app.commandLine.appendSwitch('headless');
|
|
|
|
app.disableHardwareAcceleration();
|
|
|
|
app.on('window-all-closed', () => {
|
|
});
|
|
|
|
app.whenReady().then(() => {
|
|
server.listen(PORT, () => {
|
|
console.log(`
|
|
╔═══════════════════════════════════════════════════════════╗
|
|
║ XCTerminal (Electron Headless) ║
|
|
╠═══════════════════════════════════════════════════════════╣
|
|
║ Server running at: http://localhost:${PORT} ║
|
|
║ WebSocket endpoint: ws://localhost:${PORT} ║
|
|
║ ║
|
|
║ Usage: XCTerminal.exe --port 9997 ║
|
|
╚═══════════════════════════════════════════════════════════╝
|
|
`);
|
|
});
|
|
});
|
|
|
|
process.on('exit', () => {
|
|
terminals.forEach((proc) => proc.kill());
|
|
});
|