From 7bbc95bd4306c27ee990e6f5cb00ab2102a2199f Mon Sep 17 00:00:00 2001
From: Alan Hamlett <alan.hamlett@gmail.com>
Date: Wed, 9 Oct 2024 00:17:11 +0200
Subject: [PATCH] Accept any 2xx nested hearbeat status code

---
 src/core/WakaTimeCore.ts | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/core/WakaTimeCore.ts b/src/core/WakaTimeCore.ts
index 7c4d66c..64e0681 100644
--- a/src/core/WakaTimeCore.ts
+++ b/src/core/WakaTimeCore.ts
@@ -213,17 +213,19 @@ class WakaTimeCore {
       if (response.status === 202 || response.status === 201) {
         await Promise.all(
           (data.responses ?? []).map(async (resp, respNumber) => {
-            if (resp[0].error) {
+            const nestedResp = resp[0];
+            const nestedStatus = resp[1];
+            if (nestedResp.error) {
               await this.putHeartbeatsBackInQueue(heartbeats.filter((h, i) => i === respNumber));
               console.error(resp[0].error);
-            } else if (resp[1] === 201 || resp[1] === 202) {
+            } else if (nestedStatus >= 200 && nestedStatus <= 299) {
               await changeExtensionStatus('allGood');
             } else {
-              if (resp[1] !== 400) {
+              if (nestedStatus !== 400) {
                 await this.putHeartbeatsBackInQueue(heartbeats.filter((h, i) => i === respNumber));
               }
               console.error(
-                `Heartbeat ${resp[0].data?.id ?? respNumber} returned status: ${resp[1]}`,
+                `Heartbeat ${nestedResp.data?.id ?? respNumber} returned status: ${nestedStatus}`,
               );
             }
             return resp;