aboutsummaryrefslogtreecommitdiff
path: root/Jellyfin.Api/Middleware/WebSocketHandlerMiddleware.cs
blob: 009fb6269d83d7288bd011db1688352a0b137670 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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);
    }
}