From b149883e23f68653169f0c5a7db6951157ed03ac Mon Sep 17 00:00:00 2001
From: qowevisa <qowevisa@gmail.com>
Date: Fri, 7 Jun 2024 12:28:01 +0300
Subject: [PATCH] Add reconnect to ws

---
 cmd/cgui/main.go | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/cmd/cgui/main.go b/cmd/cgui/main.go
index f53e4cc..adcfd1e 100644
--- a/cmd/cgui/main.go
+++ b/cmd/cgui/main.go
@@ -9,6 +9,7 @@ import (
 	"net"
 	"net/url"
 	"os"
+	"time"
 
 	com "git.qowevisa.me/Qowevisa/gotell/communication"
 	"git.qowevisa.me/Qowevisa/gotell/env"
@@ -51,9 +52,15 @@ func main() {
 
 	// Connect to the Electron.js application via WebSocket
 	u := url.URL{Scheme: "ws", Host: "localhost:8081", Path: "/ws"}
-	ws, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
-	if err != nil {
-		log.Fatal("dial:", err)
+	var ws *websocket.Conn
+	for {
+		ws, _, err = websocket.DefaultDialer.Dial(u.String(), nil)
+		if err != nil {
+			log.Printf("Error: dial: %v\n", err)
+			time.Sleep(5 * time.Second)
+		} else {
+			break
+		}
 	}
 	defer ws.Close()