From d61d333c5b211bb5d371125f76264c5213c4bd65 Mon Sep 17 00:00:00 2001
From: Adhiraj Singh <adhirajsingh1001@gmail.com>
Date: Wed, 22 Jun 2022 12:07:09 +0530
Subject: [PATCH] chore: nicer retry errors

---
 src/Socket/messages-send.ts | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/src/Socket/messages-send.ts b/src/Socket/messages-send.ts
index 449685c..b7b847a 100644
--- a/src/Socket/messages-send.ts
+++ b/src/Socket/messages-send.ts
@@ -521,7 +521,11 @@ export const makeMessagesSocket = (config: SocketConfig) => {
 								try {
 									const media = decryptMediaRetryData(result.media!, mediaKey, result.key.id)
 									if(media.result !== proto.MediaRetryNotification.MediaRetryNotificationResultType.SUCCESS) {
-										throw new Boom(`Media re-upload failed by device (${media.result})`, { data: media })
+										const resultStr = proto.MediaRetryNotification.MediaRetryNotificationResultType[media.result]
+										throw new Boom(
+											`Media re-upload failed by device (${resultStr})`,
+											{ data: media, statusCode: MEDIA_RETRY_STATUS_MAP[media.result] }
+										)
 									}
 
 									content.directPath = media.directPath
@@ -601,3 +605,10 @@ export const makeMessagesSocket = (config: SocketConfig) => {
 		}
 	}
 }
+
+const MEDIA_RETRY_STATUS_MAP = {
+	[proto.MediaRetryNotification.MediaRetryNotificationResultType.SUCCESS]: 200,
+	[proto.MediaRetryNotification.MediaRetryNotificationResultType.DECRYPTION_ERROR]: 412,
+	[proto.MediaRetryNotification.MediaRetryNotificationResultType.NOT_FOUND]: 404,
+	[proto.MediaRetryNotification.MediaRetryNotificationResultType.GENERAL_ERROR]: 418,
+} as const