OAuthProviders.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.Http;
 
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
 
using Tgstation.Server.Api.Models;
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,
            IHttpClientFactory httpClientFactory,
            ILoggerFactory loggerFactory,
            IOptionsSnapshot<SecurityConfiguration> securityConfigurationOptions)
        {
            ArgumentNullException.ThrowIfNull(loggerFactory);
            ArgumentNullException.ThrowIfNull(securityConfigurationOptions);
 
            var securityConfiguration = securityConfigurationOptions.Value;
 
            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));
 
#pragma warning disable CS0618 // Type or member is obsolete
            if (securityConfiguration.OAuth.TryGetValue(OAuthProvider.Keycloak, out var keyCloakConfig))
#pragma warning restore CS0618 // Type or member is obsolete
                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, bool forLogin)
            => validators.FirstOrDefault(
                x => x.Provider == oAuthProvider
                && ((forLogin && x.GatewayStatus != OAuthGatewayStatus.Only) || (!forLogin && x.GatewayStatus != OAuthGatewayStatus.Disabled)));
 
        public Dictionary<OAuthProvider, OAuthProviderInfo> ProviderInfos()
        {
            var providersAndTasks = validators.ToDictionary(
                x => x.Provider,
                x => x.GetProviderInfo());
 
            return providersAndTasks
                .Where(x => x.Value != null)
                .ToDictionary(
                    x => x.Key,
                    x => x.Value);
        }
    }
}