ApiHeadersProvider.cs

Go to the documentation of this file.

using System;
 
using Microsoft.AspNetCore.Http;
 
using Tgstation.Server.Api;
 
namespace Tgstation.Server.Host.Utils
{
    sealed class ApiHeadersProvider : IApiHeadersProvider
    {
        public ApiHeaders? ApiHeaders => attemptedApiHeadersCreation
            ? apiHeaders
            : CreateApiHeaders(false);
 
        public HeadersException? HeadersException { get; private set; }
 
        readonly IHttpContextAccessor httpContextAccessor;
 
        ApiHeaders? apiHeaders;
 
        bool attemptedApiHeadersCreation;
 
        public ApiHeadersProvider(IHttpContextAccessor httpContextAccessor)
        {
            this.httpContextAccessor = httpContextAccessor ?? throw new ArgumentNullException(nameof(httpContextAccessor));
        }
 
        public ApiHeaders CreateAuthlessHeaders() => CreateApiHeaders(true)!;
 
        ApiHeaders? CreateApiHeaders(bool authless)
        {
            if (httpContextAccessor.HttpContext == null)
                throw new InvalidOperationException("httpContextAccessor has no HttpContext!");
 
            var typedHeaders = httpContextAccessor.HttpContext.Request.GetTypedHeaders();
            if (!authless)
                attemptedApiHeadersCreation = true;
 
            try
            {
                var headers = new ApiHeaders(typedHeaders, authless, !authless);
                if (!authless)
                    apiHeaders = headers;
 
                return headers;
            }
            catch (HeadersException ex) when (!authless)
            {
                HeadersException = ex;
                return null;
            }
        }
    }
}