63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package handlers
|
|
|
|
import (
|
|
"github.com/ectrc/snow/aid"
|
|
"github.com/ectrc/snow/socket"
|
|
"github.com/gofiber/contrib/websocket"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func MiddlewareWebsocket(c *fiber.Ctx) error {
|
|
if !websocket.IsWebSocketUpgrade(c) {
|
|
return fiber.ErrUpgradeRequired
|
|
}
|
|
|
|
var protocol string
|
|
|
|
switch c.Get("Sec-WebSocket-Protocol") {
|
|
case "xmpp":
|
|
protocol = "jabber"
|
|
default:
|
|
protocol = "matchmaking"
|
|
}
|
|
|
|
c.Locals("identifier", uuid.New().String())
|
|
c.Locals("protocol", protocol)
|
|
|
|
return c.Next()
|
|
}
|
|
|
|
func WebsocketConnection(c *websocket.Conn) {
|
|
protocol := c.Locals("protocol").(string)
|
|
identifier := c.Locals("identifier").(string)
|
|
|
|
switch protocol {
|
|
case "jabber":
|
|
socket.JabberSockets.Set(identifier, socket.NewJabberSocket(c, identifier, socket.JabberData{}))
|
|
socket.HandleNewJabberSocket(identifier)
|
|
case "matchmaking":
|
|
// socket.MatchmakerSockets.Set(identifier, socket.NewMatchmakerSocket(c, socket.MatchmakerData{}))
|
|
default:
|
|
aid.Print("Invalid protocol: " + protocol)
|
|
}
|
|
}
|
|
|
|
func GetSnowConnectedSockets(c *fiber.Ctx) error {
|
|
jabber := aid.JSON{}
|
|
socket.JabberSockets.Range(func(key string, value *socket.Socket[socket.JabberData]) bool {
|
|
jabber[key] = value
|
|
return true
|
|
})
|
|
|
|
matchmaking := aid.JSON{}
|
|
socket.MatchmakerSockets.Range(func(key string, value *socket.Socket[socket.MatchmakerData]) bool {
|
|
matchmaking[key] = value
|
|
return true
|
|
})
|
|
|
|
return c.Status(200).JSON(aid.JSON{
|
|
"jabber": jabber,
|
|
"matchmaking": matchmaking,
|
|
})
|
|
} |