aboutsummaryrefslogtreecommitdiff
path: root/Jellyfin.Server/Middleware/ServerStartupMessageMiddleware.cs
blob: 2ec06339242620a3510d1fb85d642adc490cde63 (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
40
41
42
43
44
45
46
47
48
49
50
51
using System;
using System.Net.Mime;
using System.Threading.Tasks;
using MediaBrowser.Controller;
using MediaBrowser.Model.Globalization;
using Microsoft.AspNetCore.Http;

namespace Jellyfin.Server.Middleware
{
    /// <summary>
    /// Shows a custom message during server startup.
    /// </summary>
    public class ServerStartupMessageMiddleware
    {
        private readonly RequestDelegate _next;

        /// <summary>
        /// Initializes a new instance of the <see cref="ServerStartupMessageMiddleware"/> class.
        /// </summary>
        /// <param name="next">The next delegate in the pipeline.</param>
        public ServerStartupMessageMiddleware(RequestDelegate next)
        {
            _next = next;
        }

        /// <summary>
        /// Executes the middleware action.
        /// </summary>
        /// <param name="httpContext">The current HTTP context.</param>
        /// <param name="serverApplicationHost">The server application host.</param>
        /// <param name="localizationManager">The localization manager.</param>
        /// <returns>The async task.</returns>
        public async Task Invoke(
            HttpContext httpContext,
            IServerApplicationHost serverApplicationHost,
            ILocalizationManager localizationManager)
        {
            if (serverApplicationHost.CoreStartupHasCompleted
                || httpContext.Request.Path.Equals("/system/ping", StringComparison.OrdinalIgnoreCase))
            {
                await _next(httpContext).ConfigureAwait(false);
                return;
            }

            var message = localizationManager.GetLocalizedString("StartupEmbyServerIsLoading");
            httpContext.Response.StatusCode = StatusCodes.Status503ServiceUnavailable;
            httpContext.Response.ContentType = MediaTypeNames.Text.Html;
            await httpContext.Response.WriteAsync(message, httpContext.RequestAborted).ConfigureAwait(false);
        }
    }
}