AuthorizationMessageHandler.cs

Go to the documentation of this file.

using System.Net.Http;
using System.Net.Http.Headers;
using System.Threading;
using System.Threading.Tasks;
 
namespace Tgstation.Server.Client.GraphQL
{
    sealed class AuthorizationMessageHandler : DelegatingHandler
    {
        public static AsyncLocal<AuthenticationHeaderValue?> Header { get; } = new AsyncLocal<AuthenticationHeaderValue?>();
 
        readonly AuthenticationHeaderValue? headerOverride;
 
        public AuthorizationMessageHandler(AuthenticationHeaderValue? headerOverride)
        {
            this.headerOverride = headerOverride;
        }
 
        protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken)
        {
            var currentAuthHeader = headerOverride ?? Header.Value;
            if (currentAuthHeader != null)
                request.Headers.Authorization = currentAuthHeader;
 
            return base.SendAsync(request, cancellationToken);
        }
    }
}