From d4e88dd3a7df285f1311b7380c28cd22c4a4db05 Mon Sep 17 00:00:00 2001
From: Fijxu <fijxu@nadeko.net>
Date: Thu, 1 Jan 2026 21:47:43 -0300
Subject: [PATCH] refactor: move backend status handling to own function

---
 src/invidious/helpers/backend_info.cr | 33 ++++++++++++---------------
 1 file changed, 14 insertions(+), 19 deletions(-)

diff --git a/src/invidious/helpers/backend_info.cr b/src/invidious/helpers/backend_info.cr
index a610b431..ed60cc80 100644
--- a/src/invidious/helpers/backend_info.cr
+++ b/src/invidious/helpers/backend_info.cr
@@ -45,32 +45,19 @@ module BackendInfo
               body = response.body
               status_json = CompanionData.from_json(body)
               if status_json.blocked
-                @@check_mutex.synchronize do
-                  updated_status[index] = Status::Blocked.to_i
-                  updated_ends.push(index)
-                end
+                updated_ends, updated_status = self.set_status(index, updated_ends, updated_status, Status::Blocked, true)
               else
-                @@check_mutex.synchronize do
-                  updated_status[index] = Status::Working.to_i
-                  updated_ends.push(index)
-                end
+                updated_ends, updated_status = self.set_status(index, updated_ends, updated_status, Status::Working, true)
               end
             else
-              @@check_mutex.synchronize do
-                updated_status[index] = Status::Working.to_i
-                updated_ends.push(index)
-              end
+              updated_ends, updated_status = self.set_status(index, updated_ends, updated_status, Status::Working, true)
             end
-            generate_csp([companion.public_url, companion.i2p_public_url], index)
+            self.generate_csp([companion.public_url, companion.i2p_public_url], index)
           else
-            @@check_mutex.synchronize do
-              updated_status[index] = Status::Dead.to_i
-            end
+            _, updated_status = self.set_status(index, updated_ends, updated_status, Status::Dead, false)
           end
         rescue
-          @@check_mutex.synchronize do
-            updated_status[index] = Status::Dead.to_i
-          end
+          _, updated_status = self.set_status(index, updated_ends, updated_status, Status::Dead, false)
         ensure
           LOGGER.trace("Invidious companion: Done Index: \"#{index}\"")
           channels.send(nil)
@@ -94,6 +81,14 @@ module BackendInfo
     end
   end
 
+  private def set_status(index, updated_ends, updated_status, status, push = false)
+    @@check_mutex.synchronize do
+      updated_status[index] = status.to_i
+      updated_ends.push(index) if push
+    end
+    return {updated_ends, updated_status}
+  end
+
   def get_status
     # Shouldn't need to lock since we never edit this array, only change the pointer.
     return @@status