paneful 0.1.0

package/dist/server/ws-handler.js

@@ -0,0 +1,103 @@
+import { WebSocket, WebSocketServer } from 'ws';
+import { newProject } from './project-store.js';
+export class WsHandler {
+    wss;
+    client = null;
+    ptyManager;
+    projectStore;
+    constructor(server, ptyManager, projectStore) {
+        this.ptyManager = ptyManager;
+        this.projectStore = projectStore;
+        this.wss = new WebSocketServer({ noServer: true });
+        server.on('upgrade', (req, socket, head) => {
+            if (req.url === '/ws') {
+                this.wss.handleUpgrade(req, socket, head, (ws) => {
+                    this.wss.emit('connection', ws, req);
+                });
+            }
+            else {
+                socket.destroy();
+            }
+        });
+        this.wss.on('connection', (ws) => {
+            this.client = ws;
+            console.log('WebSocket client connected');
+            ws.on('message', (raw) => {
+                try {
+                    const msg = JSON.parse(raw.toString());
+                    this.handleMessage(msg);
+                }
+                catch (e) {
+                    this.send({ type: 'error', message: `Invalid message: ${e}` });
+                }
+            });
+            ws.on('close', () => {
+                console.log('WebSocket client disconnected');
+                this.client = null;
+            });
+            ws.on('error', (err) => {
+                console.error('WebSocket error:', err.message);
+                this.client = null;
+            });
+        });
+    }
+    send(msg) {
+        if (this.client && this.client.readyState === WebSocket.OPEN) {
+            this.client.send(JSON.stringify(msg));
+        }
+    }
+    handleMessage(msg) {
+        switch (msg.type) {
+            case 'pty:spawn':
+                this.handlePtySpawn(msg.terminalId, msg.projectId, msg.cwd);
+                break;
+            case 'pty:input':
+                this.ptyManager.write(msg.terminalId, msg.data);
+                break;
+            case 'pty:resize':
+                this.ptyManager.resize(msg.terminalId, msg.cols, msg.rows);
+                break;
+            case 'pty:kill': {
+                const projectId = this.ptyManager.kill(msg.terminalId);
+                if (projectId) {
+                    this.projectStore.removeTerminal(projectId, msg.terminalId);
+                    this.send({ type: 'pty:exit', terminalId: msg.terminalId, exitCode: 0 });
+                }
+                break;
+            }
+            case 'project:kill': {
+                const killed = this.ptyManager.killProject(msg.projectId);
+                for (const tid of killed) {
+                    this.send({ type: 'pty:exit', terminalId: tid, exitCode: 0 });
+                }
+                break;
+            }
+            case 'project:create': {
+                const project = newProject(msg.projectId, msg.name, msg.cwd);
+                this.projectStore.create(project);
+                break;
+            }
+            case 'project:remove': {
+                const killed = this.ptyManager.killProject(msg.projectId);
+                for (const tid of killed) {
+                    this.send({ type: 'pty:exit', terminalId: tid, exitCode: 0 });
+                }
+                this.projectStore.remove(msg.projectId);
+                break;
+            }
+        }
+    }
+    handlePtySpawn(terminalId, projectId, cwd) {
+        try {
+            this.ptyManager.spawn(terminalId, projectId, cwd, (tid, data) => {
+                this.send({ type: 'pty:output', terminalId: tid, data });
+            }, (tid, exitCode) => {
+                this.send({ type: 'pty:exit', terminalId: tid, exitCode });
+            });
+            this.projectStore.addTerminal(projectId, terminalId);
+        }
+        catch (e) {
+            this.send({ type: 'error', message: `Failed to spawn terminal: ${e}` });
+        }
+    }
+}