ProjectLighthouse/ProjectLighthouse/Startup/DebugWarmupLifetime.cs

using System;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;
using LBPUnion.ProjectLighthouse.Configuration;
using LBPUnion.ProjectLighthouse.Extensions;
using LBPUnion.ProjectLighthouse.Logging;
using LBPUnion.ProjectLighthouse.Types;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Hosting.Internal;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;

namespace LBPUnion.ProjectLighthouse.Startup;

public class DebugWarmupLifetime : IHostLifetime
{
    public IHostApplicationLifetime ApplicationLifetime { get; }

    private CancellationTokenRegistration applicationStartedRegistration;

    private readonly ConsoleLifetime consoleLifetime;
    public static ServerType ServerType;

    public DebugWarmupLifetime
    (
        IOptions<ConsoleLifetimeOptions> options,
        IHostEnvironment environment,
        IHostApplicationLifetime applicationLifetime,
        IOptions<HostOptions> hostOptions,
        ILoggerFactory loggerFactory
    )
    {
        this.consoleLifetime = new ConsoleLifetime(options, environment, applicationLifetime, hostOptions, loggerFactory);
        this.ApplicationLifetime = applicationLifetime;
    }

    public static void OnApplicationStarted()
    {
        using HttpClient client = new();

        string url = ServerType switch
        {
            ServerType.GameServer => ServerConfiguration.Instance.GameApiListenUrl,
            ServerType.Website => ServerConfiguration.Instance.WebsiteListenUrl,
            ServerType.Api => ServerConfiguration.Instance.ApiListenUrl,
            _ => throw new ArgumentOutOfRangeException(),
        };

        url = url.Replace("0.0.0.0", "127.0.0.1");

        Logger.Debug("Warming up Hot Reload...", LogArea.Startup);
        try
        {
            client.GetAsync(url).Wait();
        }
        catch(Exception e)
        {
            Logger.Debug("An error occurred while attempting to warm up hot reload. Initial page load will be delayed.", LogArea.Startup);
            Logger.Debug(e.ToDetailedException(), LogArea.Startup);
            return;
        }
        Logger.Success("Hot Reload is ready to go!", LogArea.Startup);
    }

    public Task StopAsync(CancellationToken cancellationToken) => this.consoleLifetime.StopAsync(cancellationToken);

    public Task WaitForStartAsync(CancellationToken cancellationToken)
    {
        this.applicationStartedRegistration = this.ApplicationLifetime.ApplicationStarted.Register((Action<object>)(_ => OnApplicationStarted()), (object)this);

        return this.consoleLifetime.WaitForStartAsync(cancellationToken);
    }
}