1 files changed, 39 insertions, 0 deletions
diff --git a/Jellyfin.Api/Middleware/WebSocketHandlerMiddleware.cs b/Jellyfin.Api/Middleware/WebSocketHandlerMiddleware.cs
new file mode 100644
index 000000000..009fb6269
--- /dev/null
+++ b/Jellyfin.Api/Middleware/WebSocketHandlerMiddleware.cs
@@ -0,0 +1,39 @@
+using System.Threading.Tasks;
+using MediaBrowser.Controller.Net;
+using Microsoft.AspNetCore.Http;
+
+namespace Jellyfin.Api.Middleware;
+
+/// <summary>
+/// Handles WebSocket requests.
+/// </summary>
+public class WebSocketHandlerMiddleware
+{
+    private readonly RequestDelegate _next;
+
+    /// <summary>
+    /// Initializes a new instance of the <see cref="WebSocketHandlerMiddleware"/> class.
+    /// </summary>
+    /// <param name="next">The next delegate in the pipeline.</param>
+    public WebSocketHandlerMiddleware(RequestDelegate next)
+    {
+        _next = next;
+    }
+
+    /// <summary>
+    /// Executes the middleware action.
+    /// </summary>
+    /// <param name="httpContext">The current HTTP context.</param>
+    /// <param name="webSocketManager">The WebSocket connection manager.</param>
+    /// <returns>The async task.</returns>
+    public async Task Invoke(HttpContext httpContext, IWebSocketManager webSocketManager)
+    {
+        if (!httpContext.WebSockets.IsWebSocketRequest)
+        {
+            await _next(httpContext).ConfigureAwait(false);
+            return;
+        }
+
+        await webSocketManager.WebSocketRequestHandler(httpContext).ConfigureAwait(false);
+    }
+}