spooder 4.5.1 → 4.5.2

package/README.md

@@ -933,6 +933,10 @@ server.websocket('/path/to/websocket', {
 		// validates a request before it is upgraded
 		// returns HTTP 401 if FALSE is returned
 		// allows you to check headers/authentication
+
+		// if an OBJECT is returned, the object will
+		// be accessible on the websocket as ws.data.*
+
 		return true;
 	},
 

package/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "spooder",
 	"type": "module",
-	"version": "4.5.1",
+	"version": "4.5.2",
 	"exports": {
 		".": {
 			"bun": "./src/api.ts",

package/src/api.ts

@@ -810,10 +810,18 @@ export function serve(port: number) {
 		/** Add a route to upgrade connections to websockets. */
 		websocket: (path: string, handlers: WebsocketHandlers): void => {
 			routes.push([path.split('/'), async (req: Request, url: URL) => {
-				if (await handlers.accept?.(req) === false)
-					return 401; // Unauthorized
+				let context_data = undefined;
+				if (handlers.accept) {
+					const res = await handlers.accept(req);
+					
+					if (typeof res === 'object') {
+						context_data = res;
+					} else if (!res) {
+						return 401; // Unauthorized
+					}
+				}
 
-				if (server.upgrade(req))
+				if (server.upgrade(req, { data: context_data }))
 					return 101; // Switching Protocols
 
 				return new Response('WebSocket upgrade failed', { status: 500 });