diff --git a/stream/stream.go b/stream/stream.go index f5bdc94..7242df4 100644 --- a/stream/stream.go +++ b/stream/stream.go @@ -8,11 +8,16 @@ import ( "github.com/gorilla/websocket" "github.com/gotify/server/auth" "github.com/gotify/server/model" + "net/http" + "github.com/gotify/server/mode" ) var upgrader = websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, + CheckOrigin: func(r *http.Request) bool { + return mode.IsDev(); + }, } // The API provides a handler for a WebSocket stream API.