OAuthProviders.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Common.Http;
using Tgstation.Server.Host.Configuration;
using Tgstation.Server.Host.Utils.GitHub;
 
namespace Tgstation.Server.Host.Security.OAuth
{
    sealed class OAuthProviders : IOAuthProviders
    {
        readonly IReadOnlyCollection<IOAuthValidator> validators;
 
        public OAuthProviders(
            IGitHubServiceFactory gitHubServiceFactory,
            IAbstractHttpClientFactory httpClientFactory,
            ILoggerFactory loggerFactory,
            IOptions<SecurityConfiguration> securityConfigurationOptions)
        {
            ArgumentNullException.ThrowIfNull(loggerFactory);
 
            var securityConfiguration = securityConfigurationOptions?.Value ?? throw new ArgumentNullException(nameof(securityConfigurationOptions));
 
            var validatorsBuilder = new List<IOAuthValidator>();
            validators = validatorsBuilder;
 
            if (securityConfiguration.OAuth == null)
                return;
 
            if (securityConfiguration.OAuth.TryGetValue(OAuthProvider.GitHub, out var gitHubConfig))
                validatorsBuilder.Add(
                    new GitHubOAuthValidator(
                        gitHubServiceFactory,
                        loggerFactory.CreateLogger<GitHubOAuthValidator>(),
                        gitHubConfig));
 
            if (securityConfiguration.OAuth.TryGetValue(OAuthProvider.Discord, out var discordConfig))
                validatorsBuilder.Add(
                    new DiscordOAuthValidator(
                        httpClientFactory,
                        loggerFactory.CreateLogger<DiscordOAuthValidator>(),
                        discordConfig));
 
            if (securityConfiguration.OAuth.TryGetValue(OAuthProvider.TGForums, out var tgConfig))
                validatorsBuilder.Add(
                    new TGForumsOAuthValidator(
                        httpClientFactory,
                        loggerFactory.CreateLogger<TGForumsOAuthValidator>(),
                        tgConfig));
 
            if (securityConfiguration.OAuth.TryGetValue(OAuthProvider.Keycloak, out var keyCloakConfig))
                validatorsBuilder.Add(
                    new KeycloakOAuthValidator(
                        httpClientFactory,
                        loggerFactory.CreateLogger<KeycloakOAuthValidator>(),
                        keyCloakConfig));
 
            if (securityConfiguration.OAuth.TryGetValue(OAuthProvider.InvisionCommunity, out var invisionConfig))
                validatorsBuilder.Add(
                    new InvisionCommunityOAuthValidator(
                        httpClientFactory,
                        loggerFactory.CreateLogger<InvisionCommunityOAuthValidator>(),
                        invisionConfig));
        }
 
        public IOAuthValidator GetValidator(OAuthProvider oAuthProvider) => validators.FirstOrDefault(x => x.Provider == oAuthProvider);
 
        public async Task<Dictionary<OAuthProvider, OAuthProviderInfo>> ProviderInfos(CancellationToken cancellationToken)
        {
            var providersAndTasks = validators.ToDictionary(
                x => x.Provider,
                x => x.GetProviderInfo(cancellationToken));
 
            await Task.WhenAll(providersAndTasks.Values);
 
            return providersAndTasks
                .Where(x => x.Value.Result != null)
                .ToDictionary(
                    x => x.Key,
                    x => x.Value.Result);
        }
    }
}