From 994011009460901811b1bc77d61c71bbf9beea74 Mon Sep 17 00:00:00 2001
From: Nour Alharithi <nour.alharithi@gmail.com>
Date: Fri, 24 Nov 2023 16:05:29 -0800
Subject: [PATCH] don't send stack errors

---
 src/index.ts               |  6 +-----
 src/wsConnectionManager.ts | 27 +++++++++++++++++++++++----
 2 files changed, 24 insertions(+), 9 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 8ff170f..1f92b71 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -212,11 +212,7 @@ const main = async () => {
 	await userMap.subscribe();
 	const userStatsMap = new UserStatsMap(driftClient, {
 		type: 'polling',
-		accountLoader: new BulkAccountLoader(
-			connection,
-			stateCommitment,
-			0
-		),
+		accountLoader: new BulkAccountLoader(connection, stateCommitment, 0),
 	});
 	await userStatsMap.subscribe();
 
diff --git a/src/wsConnectionManager.ts b/src/wsConnectionManager.ts
index cb383aa..bae00bc 100644
--- a/src/wsConnectionManager.ts
+++ b/src/wsConnectionManager.ts
@@ -131,11 +131,20 @@ async function main() {
 							ws.send(
 								JSON.stringify({
 									channel: requestChannel,
-									error: error.message,
+									error:
+										'Error subscribing to channel with data: ' +
+										JSON.stringify(parsedMessage),
 								})
 							);
 						} else {
-							ws.close(1003, JSON.stringify({ error: error.message }));
+							ws.close(
+								1003,
+								JSON.stringify({
+									error:
+										'Error subscribing to channel with data: ' +
+										JSON.stringify(parsedMessage),
+								})
+							);
 						}
 						return;
 					}
@@ -189,14 +198,24 @@ async function main() {
 					} catch (error) {
 						const requestChannel = safeGetRawChannelFromMessage(parsedMessage);
 						if (requestChannel) {
+							console.log('Error unsubscribing from channel:', error.message);
 							ws.send(
 								JSON.stringify({
 									channel: requestChannel,
-									error: error.message,
+									error:
+										'Error unsubscribing from channel with data: ' +
+										JSON.stringify(parsedMessage),
 								})
 							);
 						} else {
-							ws.close(1003, JSON.stringify({ error: error.message }));
+							ws.close(
+								1003,
+								JSON.stringify({
+									error:
+										'Error unsubscribing from channel with data: ' +
+										JSON.stringify(parsedMessage),
+								})
+							);
 						}
 						return;
 					}