From 6bb44afd91b030facec486c128f80d533996e98b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Karol=20S=C3=B3jko?= <karolsojko@gmail.com>
Date: Tue, 5 Sep 2023 13:50:30 +0200
Subject: [PATCH] fix(websockets): add missing region parameter in api gateway
 client

---
 packages/websockets/.env.sample                | 3 +++
 packages/websockets/src/Bootstrap/Container.ts | 1 +
 2 files changed, 4 insertions(+)

diff --git a/packages/websockets/.env.sample b/packages/websockets/.env.sample
index 7169356b0..962258349 100644
--- a/packages/websockets/.env.sample
+++ b/packages/websockets/.env.sample
@@ -13,6 +13,9 @@ SNS_AWS_REGION=
 SQS_QUEUE_URL=
 SQS_AWS_REGION=
 
+# (Optional) AWS API Gateway Setup
+API_GATEWAY_AWS_REGION=
+
 WEB_SOCKET_CONNECTION_TOKEN_SECRET=
 WEB_SOCKET_CONNECTION_TOKEN_TTL=
 
diff --git a/packages/websockets/src/Bootstrap/Container.ts b/packages/websockets/src/Bootstrap/Container.ts
index 3f973ed52..45b99bc00 100644
--- a/packages/websockets/src/Bootstrap/Container.ts
+++ b/packages/websockets/src/Bootstrap/Container.ts
@@ -94,6 +94,7 @@ export class ContainerConfigLoader {
     container.bind<ApiGatewayManagementApiClient>(TYPES.WebSockets_ApiGatewayManagementApiClient).toConstantValue(
       new ApiGatewayManagementApiClient({
         endpoint: container.get(TYPES.WEBSOCKETS_API_URL),
+        region: env.get('API_GATEWAY_AWS_REGION', true) ?? 'us-east-1',
       }),
     )