From 8a153e3807e8f87c566e05c693d7d9fbbf1a4cc3 Mon Sep 17 00:00:00 2001
From: garrettmills <shout@garrettmills.dev>
Date: Tue, 9 Aug 2022 22:27:19 -0500
Subject: [PATCH] Properly unwrap StateEvent state before calling handler

---
 package.json                           | 2 +-
 src/http/routing/SocketRouteBuilder.ts | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/package.json b/package.json
index 810f9e7..b65faa9 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@extollo/lib",
-  "version": "0.13.4",
+  "version": "0.13.5",
   "description": "The framework library that lifts up your code.",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",
diff --git a/src/http/routing/SocketRouteBuilder.ts b/src/http/routing/SocketRouteBuilder.ts
index 356f8a3..603d8c5 100644
--- a/src/http/routing/SocketRouteBuilder.ts
+++ b/src/http/routing/SocketRouteBuilder.ts
@@ -93,7 +93,9 @@ export class SocketRouteBuilder {
      */
     async build(request: Request, ws: WebSocketBus): Promise<void> {
         await this.handlers.promiseMap(handler => {
-            ws.subscribe(handler.eventClass, handler.handler.apply(request))
+            ws.subscribe(handler.eventClass, (event: StateEvent<JSONState>) => {
+                return handler.handler.apply(request)(event.getState())
+            })
         })
 
         if ( this.connectionCallback ) {