Email MCP Server

import { v4 as uuidv4 } from 'uuid'; import { WebSocket, WebSocketServer } from 'ws'; export class WebSocketServerTransport { wss; clients = new Map(); onclose; onerror; messageHandler; onconnection; ondisconnection; set onmessage(handler) { this.messageHandler = handler ? (msg, clientId) => { // @ts-ignore if (msg.id === undefined) { console.log('Broadcast message:', msg); return handler(msg); } // @ts-ignore return handler({ ...msg, // @ts-ignore id: clientId + ':' + msg.id, }); } : undefined; } constructor({ path, server }) { this.wss = new WebSocketServer({ path, server, }); } async start() { this.wss.on('connection', (ws) => { const clientId = uuidv4(); this.clients.set(clientId, ws); this.onconnection?.(clientId); ws.on('message', (data) => { try { const msg = JSON.parse(data.toString()); this.messageHandler?.(msg, clientId); } catch (err) { this.onerror?.(new Error(`Failed to parse message: ${err}`)); } }); ws.on('close', () => { this.clients.delete(clientId); this.ondisconnection?.(clientId); }); ws.on('error', (err) => { this.onerror?.(err); }); }); } async send(msg, clientId) { const [cId, msgId] = clientId?.split(':') ?? []; // @ts-ignore msg.id = parseInt(msgId); const data = JSON.stringify(msg); const deadClients = []; if (cId) { // Send to specific client const client = this.clients.get(cId); if (client?.readyState === WebSocket.OPEN) { client.send(data); } else { this.clients.delete(cId); this.ondisconnection?.(cId); } } for (const [id, client] of this.clients.entries()) { if (client.readyState !== WebSocket.OPEN) { deadClients.push(id); } } // Cleanup dead clients deadClients.forEach((id) => { this.clients.delete(id); this.ondisconnection?.(id); }); } async broadcast(msg) { return this.send(msg); } async close() { return new Promise((resolve) => { this.wss.close(() => { this.clients.clear(); resolve(); }); }); } }