From c5e7f6c545860bfbf2b4b6f9e6a022f8ac590d42 Mon Sep 17 00:00:00 2001
From: qowevisa <qowevisa@gmail.com>
Date: Sat, 8 Jun 2024 21:39:35 +0300
Subject: [PATCH] Fixes

---
 cmd/cgui/main.go | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/cmd/cgui/main.go b/cmd/cgui/main.go
index 0e76793..4fac333 100644
--- a/cmd/cgui/main.go
+++ b/cmd/cgui/main.go
@@ -70,6 +70,7 @@ func main() {
 }
 
 var r com.RegisteredUser
+var tmpLink *com.Link
 
 func readFromServer(conn net.Conn, ws *websocket.Conn) {
 	buf := make([]byte, 70000)
@@ -96,6 +97,12 @@ func readFromServer(conn net.Conn, ws *websocket.Conn) {
 			r.ID = newID
 			r.IsRegistered = true
 			break
+		case com.ID_SERVER_APPROVE_CLIENT_LINK:
+			if tmpLink == nil {
+				continue
+			}
+			msg.ToID = tmpLink.UseCount
+			msg.Data = tmpLink.Data
 		}
 		log.Printf("client: readServer: sending message to websocket: %v", *msg)
 		ws.WriteJSON(*msg)
@@ -113,12 +120,16 @@ func readFromWebSocket(conn net.Conn, ws *websocket.Conn) {
 		log.Printf("client: readWS: received message from Electron: %v", msg)
 		msg.Version = com.V1
 		switch msg.ID {
-		case com.ID_CLIENT_ASK_SERVER_LINK:
+		case com.ID_CLIENT_SEND_SERVER_LINK:
+			if !r.IsRegistered {
+				continue
+			}
 			l, err := r.GenerateLink(msg.ToID)
 			if err != nil {
 				log.Printf("Error: link: %v", err)
 				continue
 			}
+			tmpLink = &l
 			answ, err := com.ClientSendServerLink(l)
 			if err != nil {
 				log.Printf("Error: com: %v", err)