WebHostBuilderExtensions.cs

Go to the documentation of this file.

using System;
using System.IO.Abstractions;
 
using Microsoft.AspNetCore.Builder;
using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
 
using Tgstation.Server.Host.Configuration;
using Tgstation.Server.Host.Core;
using Tgstation.Server.Host.IO;
using Tgstation.Server.Host.Security;
using Tgstation.Server.Host.Setup;
using Tgstation.Server.Host.System;
 
namespace Tgstation.Server.Host.Extensions
{
    static class WebHostBuilderExtensions
    {
        public static IWebHostBuilder UseApplication(
            this IWebHostBuilder builder,
            IAssemblyInformationProvider assemblyInformationProvider,
            IIOManager ioManager,
            IPostSetupServices postSetupServices,
            IFileSystem fileSystem)
        {
            ArgumentNullException.ThrowIfNull(builder);
            ArgumentNullException.ThrowIfNull(assemblyInformationProvider);
            ArgumentNullException.ThrowIfNull(ioManager);
            ArgumentNullException.ThrowIfNull(postSetupServices);
            ArgumentNullException.ThrowIfNull(fileSystem);
 
            return builder.ConfigureServices(
                (context, services) =>
                {
                    var application = new Application(context.Configuration, context.HostingEnvironment);
                    application.ConfigureServices(services, assemblyInformationProvider, ioManager, postSetupServices, fileSystem);
                    services.AddSingleton(application);
                })
                .Configure(ConfigureApplication);
        }
 
        static void ConfigureApplication(IApplicationBuilder applicationBuilder)
            => applicationBuilder
                .ApplicationServices
                .GetRequiredService<Application>()
                .Configure(
                    applicationBuilder, // TODO: find a way to call this func and auto-resolve the services
                    applicationBuilder.ApplicationServices.GetRequiredService<IServerControl>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<ITokenFactory>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<IServerPortProvider>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<IAssemblyInformationProvider>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<IOptions<ControlPanelConfiguration>>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<IOptions<GeneralConfiguration>>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<IOptions<DatabaseConfiguration>>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<IOptions<SecurityConfiguration>>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<IOptions<SwarmConfiguration>>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<IOptions<InternalConfiguration>>(),
                    applicationBuilder.ApplicationServices.GetRequiredService<ILogger<Application>>());
    }
}