nostr-websocket-utils 0.3.1 → 0.3.3

package/dist/server.js

@@ -1,132 +0,0 @@
-import { WebSocket } from 'ws';
-import { EventEmitter } from 'events';
-import { v4 as uuidv4 } from 'uuid';
-/**
- * WebSocket server implementation for Nostr protocol
- * Extends EventEmitter to provide event-based message handling
- */
-export class NostrWSServer extends EventEmitter {
-    constructor(wss, options = {}) {
-        super();
-        this.wss = null;
-        this.clients = new Map();
-        this.heartbeatInterval = null;
-        if (!options.logger) {
-            throw new Error('Logger is required');
-        }
-        this.wss = wss;
-        this.options = {
-            heartbeatInterval: 30000,
-            logger: options.logger,
-            WebSocketImpl: WebSocket,
-            ...options,
-            handlers: {
-                message: async (_ws, _message) => { },
-                ...options.handlers,
-            },
-        };
-        this.setupServer();
-    }
-    setupServer() {
-        if (!this.wss)
-            return;
-        this.wss.on('connection', (ws) => {
-            const extWs = ws;
-            this.handleConnection(extWs);
-        });
-        if (this.options.heartbeatInterval) {
-            this.startHeartbeat();
-        }
-    }
-    handleConnection(ws) {
-        ws.isAlive = true;
-        ws.subscriptions = new Set();
-        ws.clientId = uuidv4();
-        ws.messageQueue = [];
-        this.clients.set(ws.clientId, ws);
-        ws.on('message', async (data) => {
-            try {
-                const message = JSON.parse(data.toString());
-                if (this.options.handlers?.message) {
-                    await this.options.handlers.message(ws, message);
-                }
-            }
-            catch (error) {
-                this.options.logger.error('Error handling message:', error);
-                if (this.options.handlers?.error) {
-                    this.options.handlers.error(ws, error);
-                }
-            }
-        });
-        ws.on('close', () => {
-            if (ws.clientId) {
-                this.clients.delete(ws.clientId);
-            }
-            if (this.options.handlers?.close) {
-                this.options.handlers.close(ws);
-            }
-        });
-        ws.on('error', (error) => {
-            this.options.logger.error('WebSocket error:', error);
-            if (this.options.handlers?.error) {
-                this.options.handlers.error(ws, error);
-            }
-        });
-    }
-    startHeartbeat() {
-        if (this.heartbeatInterval)
-            return;
-        this.heartbeatInterval = setInterval(() => {
-            if (!this.wss)
-                return;
-            this.wss.clients.forEach((client) => {
-                const extClient = client;
-                if (extClient.isAlive === false) {
-                    if (extClient.clientId) {
-                        this.clients.delete(extClient.clientId);
-                    }
-                    return extClient.terminate();
-                }
-                extClient.isAlive = false;
-                extClient.ping();
-            });
-        }, this.options.heartbeatInterval);
-    }
-    broadcast(message) {
-        if (!this.wss)
-            return;
-        this.wss.clients.forEach((client) => {
-            if (client.readyState === WebSocket.OPEN) {
-                client.send(JSON.stringify(message));
-            }
-        });
-    }
-    broadcastToChannel(channel, message) {
-        if (!this.wss)
-            return;
-        this.wss.clients.forEach((ws) => {
-            const extWs = ws;
-            if (extWs.readyState === WebSocket.OPEN && extWs.subscriptions?.has(channel)) {
-                extWs.send(JSON.stringify(message));
-            }
-        });
-    }
-    close() {
-        if (this.heartbeatInterval) {
-            clearInterval(this.heartbeatInterval);
-            this.heartbeatInterval = null;
-        }
-        if (this.wss) {
-            this.wss.close();
-            this.wss = null;
-        }
-    }
-    /**
-     * Check if a client with the given ID exists
-     * @param clientId - The ID of the client to check
-     * @returns boolean indicating if the client exists
-     */
-    hasClient(clientId) {
-        return this.clients.has(clientId);
-    }
-}