Port most of the HTTP code over to gin

router/router_server_ws.go

@@ -0,0 +1,60 @@
+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 seperate 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)
+		}
+	}
+}
+