Implement websocket echo, get ready for IRC chat relay

extractor/chat/chatRelayServer.go

@@ -0,0 +1,24 @@
+package chat
+
+import (
+	"net"
+
+	"github.com/gin-gonic/gin"
+	"github.com/gobwas/ws/wsutil"
+)
+
+func CreateWebsocketServer(conn net.Conn, context *gin.Context) {
+	defer conn.Close()
+
+	for {
+		msg, op, err := wsutil.ReadClientData(conn)
+		if err != nil {
+			break
+		}
+
+		err = wsutil.WriteServerMessage(conn, op, msg)
+		if err != nil {
+			break
+		}
+	}
+}

routes/root/root.go

@@ -2,14 +2,26 @@ package root
 
 import (
 	"safetwitch-backend/extractor"
+	"safetwitch-backend/extractor/chat"
 
 	"github.com/gin-gonic/gin"
+	"github.com/gobwas/ws"
 )
 
 func Routes(route *gin.Engine) {
 	auth := route.Group("/")
 
 	auth.GET("/", func(context *gin.Context) {
-		context.JSON(200, extractor.FormatMessage("SafeTwitch backend running", true))
+		upgradeHeader := context.Request.Header.Get("Upgrade")
+		if upgradeHeader == "websocket" {
+			conn, _, _, err := ws.UpgradeHTTP(context.Request, context.Writer)
+			if err != nil {
+				context.Error(err)
+			}
+
+			go chat.CreateWebsocketServer(conn, context)
+		} else {
+			context.JSON(200, extractor.FormatMessage("SafeTwitch backend running", true))
+		}
 	})
 }