Add configuration option publicServer.

This commit is contained in:
Juliusz Chroboczek committed 2023-07-10 23:59:24 +02:00
1 parent bb0a01895e
commit 90e2de0b2d
3 files changed
+23 -1

No files matched your search

+19 -1
View File
@@ -479,8 +479,26 @@ var wsUpgrader = websocket.Upgrader{
HandshakeTimeout: 30 * time.Second,
}
var wsPublicUpgrader = websocket.Upgrader{
HandshakeTimeout: 30 * time.Second,
CheckOrigin: func(r *http.Request) bool {
return true
},
}
func wsHandler(w http.ResponseWriter, r *http.Request) {
conn, err := wsUpgrader.Upgrade(w, r, nil)
conf, err := group.GetConfiguration()
if err != nil {
http.Error(w, "Internal server error",
http.StatusInternalServerError)
return
}
upgrader := wsUpgrader
if conf.PublicServer {
upgrader = wsPublicUpgrader
}
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Printf("Websocket upgrade: %v", err)
return