fix: origin for websockets

TheTipo01 · Oct 25, 2023 · 100e20b · 100e20b
1 parent cfa2e87
commit 100e20b
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/api/api.go b/api/api.go
@@ -14,7 +14,7 @@ func NewApi(servers map[string]*manager.Server, address, owner string, clients *
 
 	conf := cors.DefaultConfig()
 	conf.AllowMethods = []string{"GET", "POST", "DELETE"}
-	// TODO: Set this to false
+	// TODO: Set this to false, and add the frontend url
 	conf.AllowAllOrigins = true
 
 	r.Use(gin.Recovery(), cors.New(conf))

diff --git a/api/handlers.go b/api/handlers.go
@@ -13,6 +13,10 @@ import (
 var wsupgrader = websocket.Upgrader{
 	ReadBufferSize:  1024,
 	WriteBufferSize: 1024,
+	CheckOrigin: func(r *http.Request) bool {
+		// TODO: Set this to false, and add the frontend url
+		return true
+	},
 }
 
 func (a *Api) getQueue(c *gin.Context) {

diff --git a/manager/sound.go b/manager/sound.go
@@ -88,7 +88,6 @@ func (server *Server) playSound(el *queue.Element) (SkipReason, error) {
 				lit.Debug("VC is nil, triggering reconnection")
 				server.VC, _ = server.Clients.Discord.ChannelVoiceJoin(server.GuildID, server.VoiceChannel, false, true)
 			}
-
 		}
 	}
 }