From 66f2cf6cc26bb812f5f89af55625665df24e1044 Mon Sep 17 00:00:00 2001
From: Evgeny Kovalev <e.kovalev@daydev.org>
Date: Thu, 25 May 2023 13:31:06 +0300
Subject: [PATCH] Fixed CORS for WS

---
 internal/server/server.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/internal/server/server.go b/internal/server/server.go
index 11573b8..b46d94e 100644
--- a/internal/server/server.go
+++ b/internal/server/server.go
@@ -27,6 +27,9 @@ func wsEndpoint(w http.ResponseWriter, r *http.Request) {
 	var upgrader = websocket.Upgrader{
 		ReadBufferSize:  1024,
 		WriteBufferSize: 1024,
+		CheckOrigin: func(r *http.Request) bool {
+			return true // DANGEROUS. USING FOR DEV PURPOSES ONLY
+		},
 	}
 
 	conn, err := upgrader.Upgrade(w, r, nil)
@@ -36,6 +39,8 @@ func wsEndpoint(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	log.Println(r.RemoteAddr)
+
 	for {
 		messageType, p, err := conn.ReadMessage()
 		if err != nil {
@@ -48,7 +53,6 @@ func wsEndpoint(w http.ResponseWriter, r *http.Request) {
 		}
 	}
 
-	fmt.Fprintf(w, "hello")
 }
 
 func setupRoutes() {