AuthenticatedGraphQLServerClient.cs

Go to the documentation of this file.

using System;
using System.Net.Http.Headers;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Logging;
 
using StrawberryShake;
 
using Tgstation.Server.Common.Extensions;
 
namespace Tgstation.Server.Client.GraphQL
{
    sealed class AuthenticatedGraphQLServerClient : GraphQLServerClient, IAuthenticatedGraphQLServerClient
    {
        public ITransferClient TransferClient => restClient!.Transfer;
 
        readonly Func<string, IRestServerClient>? getRestClientForToken;
 
        IRestServerClient? restClient;
 
        public AuthenticatedGraphQLServerClient(
            IGraphQLClient graphQLClient,
            IAsyncDisposable serviceProvider,
            ILogger<GraphQLServerClient> logger,
            IRestServerClient restClient)
            : base(graphQLClient, serviceProvider, logger)
        {
            this.restClient = restClient ?? throw new ArgumentNullException(nameof(restClient));
        }
 
        public AuthenticatedGraphQLServerClient(
            IGraphQLClient graphQLClient,
            IAsyncDisposable serviceProvider,
            ILogger<GraphQLServerClient> logger,
            Action<AuthenticationHeaderValue> setAuthenticationHeader,
            AuthenticationHeaderValue? basicCredentialsHeader,
            IOperationResult<ILoginResult> loginResult,
            Func<string, IRestServerClient> getRestClientForToken)
            : base(
                  graphQLClient,
                  serviceProvider,
                  logger,
                  setAuthenticationHeader,
                  basicCredentialsHeader,
                  loginResult)
        {
            this.getRestClientForToken = getRestClientForToken ?? throw new ArgumentNullException(nameof(getRestClientForToken));
            restClient = getRestClientForToken(loginResult.Data!.Login.LoginResult!.Bearer.EncodedToken);
        }
 
        public sealed override ValueTask DisposeAsync()
#pragma warning disable CA2012 // Use ValueTasks correctly
            => ValueTaskExtensions.WhenAll(
                base.DisposeAsync(),
                restClient!.DisposeAsync());
#pragma warning restore CA2012 // Use ValueTasks correctly
 
        protected sealed override async ValueTask<AuthenticationHeaderValue> CreateUpdatedAuthenticationHeader(string bearer)
        {
            var baseTask = base.CreateUpdatedAuthenticationHeader(bearer);
            if (restClient != null)
                await restClient.DisposeAsync().ConfigureAwait(false);
 
            if (getRestClientForToken != null)
                restClient = getRestClientForToken(bearer);
 
            return await baseTask.ConfigureAwait(false);
        }
    }
}