From 2db9396dea749f90b79baa8b712e99f6111197fd Mon Sep 17 00:00:00 2001
From: Fijxu <fijxu@nadeko.net>
Date: Wed, 18 Sep 2024 18:14:28 -0300
Subject: [PATCH] SigHelper: Reconnect to signature helper

---
 src/invidious/helpers/sig_helper.cr | 23 ++++++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/src/invidious/helpers/sig_helper.cr b/src/invidious/helpers/sig_helper.cr
index 9e72c1c7..6705711d 100644
--- a/src/invidious/helpers/sig_helper.cr
+++ b/src/invidious/helpers/sig_helper.cr
@@ -176,7 +176,9 @@ module Invidious::SigHelper
 
     @conn : Connection
 
-    def initialize(uri_or_path)
+    @uri_or_path : String
+
+    def initialize(@uri_or_path)
       @conn = Connection.new(uri_or_path)
       listen
     end
@@ -186,10 +188,25 @@ module Invidious::SigHelper
 
       LOGGER.debug("SigHelper: Multiplexor listening")
 
-      # TODO: reopen socket if unexpectedly closed
       spawn do
         loop do
-          receive_data
+          begin
+            receive_data
+          rescue ex : IO::EOFError
+            LOGGER.info("Connection to helper died, trying to reconnect...")
+            # We close the socket because for some reason is not closed.
+            @conn.close
+            loop do
+              begin
+                @conn = Connection.new(@uri_or_path)
+              rescue
+                LOGGER.debug("Reconnection to helper unsuccessful, retrying.")
+                sleep 1
+                next
+              end
+              break if !@conn.closed?
+            end
+          end
           Fiber.yield
         end
       end