GatewayClientHostedService.cs

using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Options;

using NetCord.Gateway;

namespace NetCord.Hosting.Gateway;

internal partial class GatewayClientHostedService(IServiceProvider services) : IHostedService
{
    public Task StartAsync(CancellationToken cancellationToken)
    {
        var client = services.GetRequiredService<GatewayClient>();

        foreach (var handler in services.GetServices<IGatewayHandler>())
        {
            if (handler is IDelegateGatewayHandlerBase delegateHandler)
                RegisterDelegateHandler(client, delegateHandler);
            else
                RegisterClassHandler(client, handler);
        }

        var options = services.GetRequiredService<IOptions<GatewayClientOptions>>().Value;

        return options.AutoStartStop.GetValueOrDefault(true)
            ? client.StartAsync(cancellationToken: cancellationToken).AsTask()
            : Task.CompletedTask;
    }

    public Task StopAsync(CancellationToken cancellationToken)
    {
        var options = services.GetRequiredService<IOptions<GatewayClientOptions>>().Value;

        if (!options.AutoStartStop.GetValueOrDefault(true))
            return Task.CompletedTask;

        var client = services.GetRequiredService<GatewayClient>();

        return client.CloseAsync(cancellationToken: cancellationToken).AsTask();
    }
}