From 89418fdbbd57ea68a224dd0eedc55833ae3a4f0a Mon Sep 17 00:00:00 2001
From: poiuty <poiuty@protonmail.com>
Date: Sun, 13 Nov 2022 13:00:53 +0300
Subject: [PATCH] func Upgrade DEPRECATED

https://pkg.go.dev/github.com/gorilla/websocket#Upgrade
---
 app/echo.go | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/app/echo.go b/app/echo.go
index 3bfddf1..5178320 100644
--- a/app/echo.go
+++ b/app/echo.go
@@ -6,6 +6,17 @@ import (
 	"github.com/gorilla/websocket"
 )
 
+var upgrader = websocket.Upgrader{
+    ReadBufferSize:  1024,
+    WriteBufferSize: 1024,
+    CheckOrigin: func(r *http.Request) bool {
+		if r.Header.Get("origin") == "https://statbate.com" {
+			return true
+		}
+		return false
+	},
+}
+
 var (
 	wsClients = make(map[*websocket.Conn]struct{})
 
@@ -41,7 +52,7 @@ func broadcast() {
 }
 
 func wsHandler(w http.ResponseWriter, r *http.Request) {
-	conn, err := websocket.Upgrade(w, r, w.Header(), 1024, 1024)
+	conn, err := upgrader.Upgrade(w, r, nil)
 	if err != nil {
 		return
 	}