AuthorityBase.cs

Go to the documentation of this file.

using System;
using System.Globalization;
 
using Microsoft.Extensions.Logging;
 
using Octokit;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Response;
using Tgstation.Server.Host.Database;
using Tgstation.Server.Host.Security;
 
namespace Tgstation.Server.Host.Authority.Core
{
    abstract class AuthorityBase : IAuthority
    {
        protected IAuthenticationContext AuthenticationContext { get; }
 
        protected IDatabaseContext DatabaseContext { get; }
 
        protected ILogger<AuthorityBase> Logger { get; }
 
        protected static AuthorityResponse<TResult> BadRequest<TResult>(ErrorCode errorCode)
            => new(
                new ErrorMessageResponse(errorCode),
                HttpFailureResponse.BadRequest);
 
        protected static AuthorityResponse<TResult> Unauthorized<TResult>()
            => new(
                new ErrorMessageResponse(),
                HttpFailureResponse.Unauthorized);
 
        protected static AuthorityResponse<TResult> Gone<TResult>()
            => new(
                new ErrorMessageResponse(ErrorCode.ResourceNotPresent),
                HttpFailureResponse.Gone);
 
        protected static AuthorityResponse<TResult> Forbid<TResult>()
            => new(
                new ErrorMessageResponse(),
                HttpFailureResponse.Forbidden);
 
        protected static AuthorityResponse<TResult> NotFound<TResult>()
            => new(
                new ErrorMessageResponse(ErrorCode.ResourceNeverPresent),
                HttpFailureResponse.NotFound);
 
        protected static AuthorityResponse<TResult> Conflict<TResult>(ErrorCode errorCode)
            => new(
                new ErrorMessageResponse(errorCode),
                HttpFailureResponse.Conflict);
 
        protected AuthorityBase(
            IAuthenticationContext authenticationContext,
            IDatabaseContext databaseContext,
            ILogger<AuthorityBase> logger)
        {
            AuthenticationContext = authenticationContext ?? throw new ArgumentNullException(nameof(authenticationContext));
            DatabaseContext = databaseContext ?? throw new ArgumentNullException(nameof(databaseContext));
            Logger = logger ?? throw new ArgumentNullException(nameof(logger));
        }
 
        protected AuthorityResponse<TResult> RateLimit<TResult>(RateLimitExceededException rateLimitException)
        {
            Logger.LogWarning(rateLimitException, "Exceeded GitHub rate limit!");
            var secondsString = Math.Ceiling(rateLimitException.GetRetryAfterTimeSpan().TotalSeconds).ToString(CultureInfo.InvariantCulture);
            return new(
                new ErrorMessageResponse(ErrorCode.GitHubApiRateLimit)
                {
                    AdditionalData = $"Retry-After: {secondsString}s",
                },
                HttpFailureResponse.RateLimited);
        }
    }
}