From 8c1d4f2ee2bbf89499f0165795f48f1394101434 Mon Sep 17 00:00:00 2001
From: Tarashish Mishra <sunu@sunu.in>
Date: Fri, 31 May 2024 15:19:31 +0530
Subject: [PATCH] retry a few time if a connection fails

Sometimes the initial websocket connection fails. This is a workaround that retries the connection a few times with some delay in between connection attempts.
---
 js/index.js | 83 ++++++++++++++++++++++++++++++++---------------------
 1 file changed, 51 insertions(+), 32 deletions(-)

diff --git a/js/index.js b/js/index.js
index 72ae4474..ab78f460 100644
--- a/js/index.js
+++ b/js/index.js
@@ -11,6 +11,9 @@ import RFB from "@novnc/novnc/core/rfb";
 
 import { setupTooltip } from "./tooltip.js";
 
+const maxRetryCount = 5;
+const retryInterval = 3;  // seconds
+
 // When this function is called we have successfully connected to a server
 function connectedToServer() {
   status("Connected");
@@ -22,6 +25,15 @@ function disconnectedFromServer(e) {
     status("Disconnected");
   } else {
     status("Something went wrong, connection is closed");
+    if (retryCount < maxRetryCount) {
+      status(`Reconnecting in ${retryInterval} seconds`);
+      setTimeout(() => {
+        connect();
+        retryCount++;
+      }, retryInterval * 1000);
+    } else {
+      status("Failed to connect, giving up");
+    }
   }
 }
 
@@ -36,38 +48,45 @@ function status(text) {
 let websockifyUrl = new URL("../desktop-websockify/", window.location);
 websockifyUrl.protocol = window.location.protocol === "https:" ? "wss" : "ws";
 
-// Creating a new RFB object will start a new connection
-const rfb = new RFB(
-  document.getElementById("screen"),
-  websockifyUrl.toString(),
-  {},
-);
-
-// Add listeners to important events from the RFB module
-rfb.addEventListener("connect", connectedToServer);
-rfb.addEventListener("disconnect", disconnectedFromServer);
-
-// Scale our viewport so the user doesn't have to scroll
-rfb.scaleViewport = true;
-
-// Use a CSS variable to set background color
-rfb.background = "var(--jupyter-medium-dark-grey)";
+let retryCount = 0;
 
-// Clipboard
-function clipboardReceive(e) {
-  document.getElementById("clipboard-text").value = e.detail.text;
-}
-rfb.addEventListener("clipboard", clipboardReceive);
-
-function clipboardSend() {
-  const text = document.getElementById("clipboard-text").value;
-  rfb.clipboardPasteFrom(text);
+function connect() {
+  // Creating a new RFB object will start a new connection
+  let rfb = new RFB(
+    document.getElementById("screen"),
+    websockifyUrl.toString(),
+    {},
+  );
+  
+  // Add listeners to important events from the RFB module
+  rfb.addEventListener("connect", connectedToServer);
+  rfb.addEventListener("disconnect", disconnectedFromServer);
+  
+  // Scale our viewport so the user doesn't have to scroll
+  rfb.scaleViewport = true;
+  
+  // Use a CSS variable to set background color
+  rfb.background = "var(--jupyter-medium-dark-grey)";
+  
+  // Clipboard
+  function clipboardReceive(e) {
+    document.getElementById("clipboard-text").value = e.detail.text;
+  }
+  rfb.addEventListener("clipboard", clipboardReceive);
+  
+  function clipboardSend() {
+    const text = document.getElementById("clipboard-text").value;
+    rfb.clipboardPasteFrom(text);
+  }
+  document
+    .getElementById("clipboard-text")
+    .addEventListener("change", clipboardSend);
+  
+  setupTooltip(
+    document.getElementById("clipboard-button"),
+    document.getElementById("clipboard-container"),
+  );
 }
-document
-  .getElementById("clipboard-text")
-  .addEventListener("change", clipboardSend);
 
-setupTooltip(
-  document.getElementById("clipboard-button"),
-  document.getElementById("clipboard-container"),
-);
+// Start the connection
+connect();