From 687eb2e3f6be6c559b4b073a776baddd446f7155 Mon Sep 17 00:00:00 2001
From: Fijxu <fijxu@nadeko.net>
Date: Tue, 30 Dec 2025 22:01:03 -0300
Subject: [PATCH] fix current companion overflow checking

---
 src/invidious/routes/before_all.cr | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/invidious/routes/before_all.cr b/src/invidious/routes/before_all.cr
index 5e788b23..ca7e26dc 100644
--- a/src/invidious/routes/before_all.cr
+++ b/src/invidious/routes/before_all.cr
@@ -61,8 +61,12 @@ module Invidious::Routes::BeforeAll
             end
           end
 
-          if current_companion > CONFIG.invidious_companion.size
-            current_companion = current_companion % CONFIG.invidious_companion.size - 1
+          if current_companion < 0
+            current_companion = rand(CONFIG.invidious_companion.size)
+          end
+
+          if current_companion >= CONFIG.invidious_companion.size
+            current_companion = current_companion % CONFIG.invidious_companion.size
             env.response.cookies[CONFIG.server_id_cookie_name] = Invidious::User::Cookies.server_id(host, current_companion)
           end