109 lines
3.1 KiB
Go
109 lines
3.1 KiB
Go
package ws
|
|
|
|
import (
|
|
"encoding/json"
|
|
"homestead/homestead_gateway/util/cache"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func (wsg *WebsocketGateway) handleSync(w http.ResponseWriter, r *http.Request) {
|
|
conn, err := wsg.validateAndUpgradeConnection(w, r)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
if wsg.bodySizeBytes > 0 {
|
|
conn.SetReadLimit(wsg.bodySizeBytes)
|
|
} else {
|
|
conn.SetReadLimit(1 << 20) // sensible default 1MiB
|
|
}
|
|
|
|
_ = conn.SetReadDeadline(time.Now().Add(60 * time.Second))
|
|
conn.SetPongHandler(func(appData string) error {
|
|
_ = conn.SetReadDeadline(time.Now().Add(60 * time.Second))
|
|
return nil
|
|
})
|
|
|
|
typ, data, err := conn.ReadMessage()
|
|
if err != nil {
|
|
wsg.sendWebsocketError(conn, "Internal Server Error", 500, true)
|
|
wsg.logger.Error("Failed to read handshake.", "remote", conn.RemoteAddr().String(), "err", err)
|
|
return
|
|
}
|
|
|
|
if typ != websocket.TextMessage && typ != websocket.BinaryMessage {
|
|
wsg.sendWebsocketError(conn, "Initial message must be a handshake.", 400, true)
|
|
wsg.logger.Warn("Invalid handshake message type.", "remote", conn.RemoteAddr().String())
|
|
return
|
|
}
|
|
|
|
var handshake Handshake
|
|
if err := json.Unmarshal(data, &handshake); err != nil {
|
|
wsg.sendWebsocketError(conn, "Malformed handshake.", 400, true)
|
|
wsg.logger.Warn("Malformed handshake.", "remote", conn.RemoteAddr().String(), "err", err)
|
|
return
|
|
}
|
|
|
|
meta := cache.ConnectionMetaData{ConnectionType: handshake.Type}
|
|
|
|
switch handshake.Type {
|
|
case "mod":
|
|
var mhs ModHandshake
|
|
|
|
if err := json.Unmarshal(handshake.Data, &mhs); err != nil {
|
|
wsg.sendWebsocketError(conn, "Malformed mod handshake.", 400, true)
|
|
wsg.logger.Warn("Malformed mod handshake.", "remote", conn.RemoteAddr().String(), "err", err)
|
|
return
|
|
}
|
|
|
|
meta.ID = mhs.ServerID
|
|
|
|
if err = wsg.sendWebsocketResponse(conn, GatewayAck{Status: "connected", Type: "mod"}); err != nil {
|
|
return
|
|
}
|
|
|
|
wsg.registerConn(conn, meta, "mod")
|
|
wsg.logger.Info("Mod connected via Websocket.", "remote", conn.RemoteAddr().String(), "server_id", mhs.ServerID)
|
|
|
|
go wsg.read(conn, meta, "mod")
|
|
|
|
case "bot":
|
|
var bhs BotHandshake
|
|
|
|
if err := json.Unmarshal(handshake.Data, &bhs); err != nil {
|
|
wsg.sendWebsocketError(conn, "Malformed bot handshake.", 400, true)
|
|
wsg.logger.Warn("Malformed bot handshake.", "remote", conn.RemoteAddr().String(), "err", err)
|
|
return
|
|
}
|
|
|
|
meta.ID = bhs.BotID
|
|
|
|
if ok := wsg.registerConn(conn, meta, "bot"); !ok {
|
|
wsg.sendWebsocketError(conn, "Bot already connected.", 409, true)
|
|
return
|
|
}
|
|
|
|
if err = wsg.sendWebsocketResponse(conn, GatewayAck{Status: "connected", Type: "bot"}); err != nil {
|
|
return
|
|
}
|
|
|
|
wsg.logger.Info("Bot connected via Websocket.", "remote", conn.RemoteAddr().String(), "bot_id", bhs.BotID)
|
|
|
|
go wsg.read(conn, meta, "bot")
|
|
|
|
default:
|
|
wsg.sendWebsocketError(conn, "Unknown handshake.", 400, true)
|
|
wsg.logger.Warn("Unknown connection type.", "remote", conn.RemoteAddr().String(), "type", handshake.Type)
|
|
return
|
|
}
|
|
}
|
|
|
|
func (wsg *WebsocketGateway) handleHealth(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(200)
|
|
_ = json.NewEncoder(w).Encode(map[string]interface{}{"status": "healthy"})
|
|
}
|