redweb 0.6.0 → 0.6.2

package/README.md

@@ -76,7 +76,7 @@ const app = new HttpsServer(options);
 
 ### WebSocket Server with Handlers
 
-RedWeb now supports **handler-based routing** for WebSocket connections, allowing you to modularize and secure your WebSocket message handling logic.
+RedWeb supports **handler-based routing** for WebSocket connections, allowing you to modularize and secure your WebSocket message handling logic.
 
 #### Defining a Custom Handler
 
@@ -122,6 +122,34 @@ const options = {
 const socketServer = new SocketServer(options);
 ```
 
+#### Adding Handlers Dynamically
+
+Handlers can be added after the WebSocket server has started using the `addHandler` method.
+
+```javascript
+class DynamicHandler {
+    constructor() {
+        this.name = 'DynamicHandler';
+        this.handlers = {
+            dynamicMessage: (socket, data) => {
+                console.log(`Received dynamic message: ${data.message}`);
+                socket.send(JSON.stringify({ type: 'dynamicResponse', message: 'Handled dynamically!' }));
+            }
+        };
+    }
+
+    newConnection(socket) {
+        console.log('DynamicHandler: New connection established');
+    }
+}
+
+const { SocketServer } = require('redweb');
+const socketServer = new SocketServer({ port: 3000 });
+
+// Dynamically add a new handler
+socketServer.addHandler(DynamicHandler);
+```
+
 #### Client Communication with a Handler
 
 The client must identify the handler during the initial connection with a `__handlerConnect` message.

package/index.d.ts

@@ -35,7 +35,7 @@ declare module 'redweb' {
     }
 
     /**
-     * Handler configuration for WebSocket.
+     * Configuration for a WebSocket handler.
      */
     export interface HandlerConfig {
         name: string;
@@ -90,10 +90,11 @@ declare module 'redweb' {
         constructor(config: HandlerConfig);
 
         /**
-         * Adds a new WebSocket connection and sets up message handling for this handler.
+         * Adds a new WebSocket connection, processes initial data, and sets up message handling for this handler.
          * @param socket - The WebSocket connection to add.
+         * @param data - Optional initial data sent during the connection handshake.
          */
-        newConnection(socket: WebSocket): void;
+        newConnection(socket: WebSocket, data?: any): void;
 
         /**
          * Handles an incoming message and routes it to the appropriate handler function.
@@ -109,6 +110,29 @@ declare module 'redweb' {
         broadcast(message: object): void;
     }
 
+    /**
+     * Base WebSocket server class.
+     */
+    export class BaseSocketServer {
+        /**
+         * Map of connected WebSocket clients by their IP address.
+         */
+        clients: Map<string, WebSocket>;
+
+        /**
+         * List of handler instances currently registered with the server.
+         */
+        handlers: BaseHandler[];
+
+        constructor(server: HTTPServer | HTTPSServer, options?: SocketServerOptions);
+
+        /**
+         * Adds a new handler to the WebSocket server.
+         * @param HandlerClass - A class extending `BaseHandler` to add to the server.
+         */
+        addHandler(HandlerClass: new () => BaseHandler): void;
+    }
+
     /**
      * HTTP server class.
      */
@@ -126,14 +150,14 @@ declare module 'redweb' {
     /**
      * WebSocket server class.
      */
-    export class SocketServer {
+    export class SocketServer extends BaseSocketServer {
         constructor(options?: SocketServerOptions);
     }
 
     /**
      * Secure WebSocket server class.
      */
-    export class SecureSocketServer {
+    export class SecureSocketServer extends BaseSocketServer {
         constructor(options?: SocketServerOptions);
     }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "redweb",
-  "version": "0.6.0",
+  "version": "0.6.2",
   "description": "A way to quickly set up an express server",
   "main": "index.js",
   "scripts": {

package/src/ws/BaseHandler.js

@@ -28,24 +28,34 @@ class BaseHandler {
         this.connections = [];
     }
 
-    /**
-     * Adds a new WebSocket connection and sets up message handling for this handler.
-     * @param {WebSocket} socket - The WebSocket connection to add.
-     */
-    newConnection(socket) {
-        this.connections.push(socket);
-
+    handleSocketMessages(socket) {
         socket.on('message', (message) => this.handleMessage(socket, message));
+    }
 
+    handleSocketClose(socket) {
         socket.on('close', () => {
             this.connections = this.connections.filter((conn) => conn !== socket);
             console.log(`Connection closed for handler "${this.name}".`);
         });
+    }
 
+    handleNewConnection(socket) {
+        this.connections.push(socket);
+        this.handleSocketMessages(socket);
+        this.handleSocketClose(socket);
         socket.isAssigned = true;
         console.log(`New connection added to handler "${this.name}".`);
     }
 
+    /**
+     * Adds a new WebSocket connection and sets up message handling for this handler.
+     * @param {WebSocket} socket - The WebSocket connection to add.
+     */
+    newConnection(socket, data) {
+        console.log(`Received newConnection data ${data}`);
+        this.handleNewConnection(socket);
+    }
+
     /**
      * Handles an incoming message and routes it to the appropriate handler function.
      * @param {WebSocket} socket - The WebSocket connection that sent the message.