Fix the websocket not working

main.go

@@ -197,6 +197,13 @@ var (
 	upgrade           = websocket.Upgrader{
 		ReadBufferSize:  1024,
 		WriteBufferSize: 1024,
+		CheckOrigin: func(r *http.Request) bool {
+			if r.Header.Get("Origin") == configFile.Origin {
+				return true
+			} else {
+				return false
+			}
+		},
 	}
 )
 
@@ -275,6 +282,7 @@ type Bet struct {
 
 type Config struct {
 	ApiKey     string  `json:"key"`
+	Origin     string  `json:"origin"`
 	Rate       float64 `json:"rate"`
 	Multiplier float64 `json:"multiplier"`
 }