redweb 0.6.1 → 0.6.2

package/index.d.ts

@@ -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.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "redweb",
-  "version": "0.6.1",
+  "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.