From c35e1a2693ce319e5f78756f5da7f32e0f376606 Mon Sep 17 00:00:00 2001
From: polamoros <info@polamoros.com>
Date: Tue, 23 Jan 2024 09:53:02 +0100
Subject: [PATCH] wip

---
 websockets/platform/tf-aws.w | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/websockets/platform/tf-aws.w b/websockets/platform/tf-aws.w
index 9cfcea35..d32d1344 100644
--- a/websockets/platform/tf-aws.w
+++ b/websockets/platform/tf-aws.w
@@ -60,7 +60,7 @@ pub class WebSocket_tfaws impl awsapi.IAwsWebSocket {
       unsafeCast(inflight (event: awsapi.WebSocketAwsRequest): awsapi.WebSocketAwsResponse => {
         if event.requestContext.routeKey == routeKey {
           handler(event.requestContext.connectionId, {
-            headers: event.headers,
+            headers: event.requestContext.headers,
           });
         }
         return {
@@ -79,12 +79,9 @@ pub class WebSocket_tfaws impl awsapi.IAwsWebSocket {
     let onDisconnectFunction = new cloud.Function(
       unsafeCast(inflight (event: awsapi.WebSocketAwsRequest): awsapi.WebSocketAwsResponse => {
         if event.requestContext.routeKey == routeKey {
-          handler(
-            event.requestContext.connectionId,
-            {
-              headers: event.headers,
-            }
-          );
+          handler(event.requestContext.connectionId, {
+            headers: event.requestContext.headers,
+          });
         }
 
         return {