61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package router
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"github.com/gin-gonic/gin"
|
|
ws "github.com/gorilla/websocket"
|
|
"github.com/pterodactyl/wings/router/websocket"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// Upgrades a connection to a websocket and passes events along between.
|
|
func getServerWebsocket(c *gin.Context) {
|
|
s := GetServer(c.Param("server"))
|
|
handler, err := websocket.GetHandler(s, c.Writer, c.Request)
|
|
if err != nil {
|
|
TrackedServerError(err, s).AbortWithServerError(c)
|
|
return
|
|
}
|
|
defer handler.Connection.Close()
|
|
|
|
// Create a context that can be canceled when the user disconnects from this
|
|
// socket that will also cancel listeners running in separate threads.
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
go handler.ListenForServerEvents(ctx)
|
|
go handler.ListenForExpiration(ctx)
|
|
|
|
for {
|
|
j := websocket.Message{}
|
|
|
|
_, p, err := handler.Connection.ReadMessage()
|
|
if err != nil {
|
|
if !ws.IsCloseError(
|
|
err,
|
|
ws.CloseNormalClosure,
|
|
ws.CloseGoingAway,
|
|
ws.CloseNoStatusReceived,
|
|
ws.CloseServiceRestart,
|
|
ws.CloseAbnormalClosure,
|
|
) {
|
|
zap.S().Warnw("error handling websocket message", zap.Error(err))
|
|
}
|
|
break
|
|
}
|
|
|
|
// Discard and JSON parse errors into the void and don't continue processing this
|
|
// specific socket request. If we did a break here the client would get disconnected
|
|
// from the socket, which is NOT what we want to do.
|
|
if err := json.Unmarshal(p, &j); err != nil {
|
|
continue
|
|
}
|
|
|
|
if err := handler.HandleInbound(j); err != nil {
|
|
handler.SendErrorJson(err)
|
|
}
|
|
}
|
|
}
|
|
|