Files
silo-server/internal/jellycompat/handlers_websocket.go
T

66 lines
1.6 KiB
Go

package jellycompat
import (
"encoding/json"
"log/slog"
"net/http"
"time"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
CheckOrigin: func(r *http.Request) bool { return true },
}
type wsMessage struct {
MessageType string `json:"MessageType"`
Data json.RawMessage `json:"Data,omitempty"`
}
// HandleSocket upgrades the connection to a WebSocket and keeps it alive.
// Jellyfin clients (e.g. Streamyfin) open a persistent WebSocket for
// remote-control commands. This stub accepts the connection and responds
// to KeepAlive pings but sends no server-initiated commands.
func HandleSocket(w http.ResponseWriter, r *http.Request) {
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
slog.Error("websocket upgrade failed", "error", err)
return
}
defer conn.Close()
slog.Info("websocket connected",
"remote_addr", r.RemoteAddr,
"device_id", r.URL.Query().Get("deviceId"),
)
conn.SetReadDeadline(time.Now().Add(60 * time.Second))
conn.SetPongHandler(func(string) error {
conn.SetReadDeadline(time.Now().Add(60 * time.Second))
return nil
})
for {
_, raw, err := conn.ReadMessage()
if err != nil {
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseNormalClosure) {
slog.Warn("websocket read error", "error", err)
}
return
}
var msg wsMessage
if err := json.Unmarshal(raw, &msg); err != nil {
continue
}
switch msg.MessageType {
case "KeepAlive":
conn.SetReadDeadline(time.Now().Add(60 * time.Second))
default:
slog.Debug("websocket message ignored", "type", msg.MessageType)
}
}
}