AdministrationAuthority.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.Extensions.Caching.Memory;
using Microsoft.Extensions.Logging;
 
using Octokit;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Response;
using Tgstation.Server.Api.Rights;
using Tgstation.Server.Host.Authority.Core;
using Tgstation.Server.Host.Core;
using Tgstation.Server.Host.Database;
using Tgstation.Server.Host.Security;
using Tgstation.Server.Host.Transfer;
using Tgstation.Server.Host.Utils.GitHub;
 
namespace Tgstation.Server.Host.Authority
{
    sealed class AdministrationAuthority : AuthorityBase, IAdministrationAuthority
    {
        const string OctokitException = "Bad GitHub API response, check configuration!";
 
        static readonly object ReadCacheKey = new();
 
        readonly IGitHubServiceFactory gitHubServiceFactory;
 
        readonly IServerControl serverControl;
 
        readonly IServerUpdateInitiator serverUpdateInitiator;
 
        readonly IFileTransferTicketProvider fileTransferService;
 
        readonly IMemoryCache cacheService;
 
        public AdministrationAuthority(
            IAuthenticationContext authenticationContext,
            IDatabaseContext databaseContext,
            ILogger<UserAuthority> logger,
            IGitHubServiceFactory gitHubServiceFactory,
            IServerControl serverControl,
            IServerUpdateInitiator serverUpdateInitiator,
            IFileTransferTicketProvider fileTransferService,
            IMemoryCache cacheService)
            : base(
                  authenticationContext,
                  databaseContext,
                  logger)
        {
            this.gitHubServiceFactory = gitHubServiceFactory ?? throw new ArgumentNullException(nameof(gitHubServiceFactory));
            this.serverControl = serverControl ?? throw new ArgumentNullException(nameof(serverControl));
            this.serverUpdateInitiator = serverUpdateInitiator ?? throw new ArgumentNullException(nameof(serverUpdateInitiator));
            this.fileTransferService = fileTransferService ?? throw new ArgumentNullException(nameof(fileTransferService));
            this.cacheService = cacheService ?? throw new ArgumentNullException(nameof(cacheService));
        }
 
        public async ValueTask<AuthorityResponse<AdministrationResponse>> GetUpdateInformation(bool forceFresh, CancellationToken cancellationToken)
        {
            try
            {
                async Task<AdministrationResponse> CacheFactory()
                {
                    Version? greatestVersion = null;
                    Uri? repoUrl = null;
                    try
                    {
                        var gitHubService = await gitHubServiceFactory.CreateService(cancellationToken);
                        var repositoryUrlTask = gitHubService.GetUpdatesRepositoryUrl(cancellationToken);
                        var releases = await gitHubService.GetTgsReleases(cancellationToken);
 
                        foreach (var kvp in releases)
                        {
                            var version = kvp.Key;
                            var release = kvp.Value;
                            if (version.Major > 3 // Forward/backward compatible but not before TGS4
                                && (greatestVersion == null || version > greatestVersion))
                                greatestVersion = version;
                        }
 
                        repoUrl = await repositoryUrlTask;
                    }
                    catch (NotFoundException e)
                    {
                        Logger.LogWarning(e, "Not found exception while retrieving upstream repository info!");
                    }
 
                    return new AdministrationResponse
                    {
                        LatestVersion = greatestVersion,
                        TrackedRepositoryUrl = repoUrl,
                        GeneratedAt = DateTimeOffset.UtcNow,
                    };
                }
 
                var ttl = TimeSpan.FromMinutes(30);
                Task<AdministrationResponse> task;
                if (forceFresh || !cacheService.TryGetValue(ReadCacheKey, out var rawCacheObject))
                {
                    using var entry = cacheService.CreateEntry(ReadCacheKey);
                    entry.AbsoluteExpirationRelativeToNow = ttl;
                    entry.Value = task = CacheFactory();
                }
                else
                    task = (Task<AdministrationResponse>)rawCacheObject!;
 
                return new AuthorityResponse<AdministrationResponse>(await task);
            }
            catch (RateLimitExceededException e)
            {
                return RateLimit<AdministrationResponse>(e);
            }
            catch (ApiException e)
            {
                Logger.LogWarning(e, OctokitException);
                return new AuthorityResponse<AdministrationResponse>(
                    new ErrorMessageResponse(ErrorCode.RemoteApiError)
                    {
                        AdditionalData = e.Message,
                    },
                    HttpFailureResponse.FailedDependency);
            }
        }
 
        public async ValueTask<AuthorityResponse<ServerUpdateResponse>> TriggerServerVersionChange(Version targetVersion, bool uploadZip, CancellationToken cancellationToken)
        {
            var attemptingUpload = uploadZip == true;
            if (attemptingUpload)
            {
                if (!AuthenticationContext.PermissionSet.AdministrationRights!.Value.HasFlag(AdministrationRights.UploadVersion))
                    return Forbid<ServerUpdateResponse>();
            }
            else if (!AuthenticationContext.PermissionSet.AdministrationRights!.Value.HasFlag(AdministrationRights.ChangeVersion))
                return Forbid<ServerUpdateResponse>();
 
            if (targetVersion.Major < 4)
                return BadRequest<ServerUpdateResponse>(ErrorCode.CannotChangeServerSuite);
 
            if (!serverControl.WatchdogPresent)
                return new AuthorityResponse<ServerUpdateResponse>(
                    new ErrorMessageResponse(ErrorCode.MissingHostWatchdog),
                    HttpFailureResponse.UnprocessableEntity);
 
            IFileUploadTicket? uploadTicket = attemptingUpload
                ? fileTransferService.CreateUpload(FileUploadStreamKind.None)
                : null;
 
            ServerUpdateResult updateResult;
            try
            {
                try
                {
                    updateResult = await serverUpdateInitiator.InitiateUpdate(uploadTicket, targetVersion, cancellationToken);
                }
                catch
                {
                    if (attemptingUpload)
                        await uploadTicket!.DisposeAsync();
 
                    throw;
                }
            }
            catch (RateLimitExceededException ex)
            {
                return RateLimit<ServerUpdateResponse>(ex);
            }
            catch (ApiException e)
            {
                Logger.LogWarning(e, OctokitException);
                return new AuthorityResponse<ServerUpdateResponse>(
                    new ErrorMessageResponse(ErrorCode.RemoteApiError)
                    {
                        AdditionalData = e.Message,
                    },
                    HttpFailureResponse.FailedDependency);
            }
 
            return updateResult switch
            {
                ServerUpdateResult.Started => new AuthorityResponse<ServerUpdateResponse>(new ServerUpdateResponse(targetVersion, uploadTicket?.Ticket.FileTicket), HttpSuccessResponse.Accepted),
                ServerUpdateResult.ReleaseMissing => Gone<ServerUpdateResponse>(),
                ServerUpdateResult.UpdateInProgress => BadRequest<ServerUpdateResponse>(ErrorCode.ServerUpdateInProgress),
                ServerUpdateResult.SwarmIntegrityCheckFailed => new AuthorityResponse<ServerUpdateResponse>(
                    new ErrorMessageResponse(ErrorCode.SwarmIntegrityCheckFailed),
                    HttpFailureResponse.FailedDependency),
                _ => throw new InvalidOperationException($"Unexpected ServerUpdateResult: {updateResult}"),
            };
        }
 
        public async ValueTask<AuthorityResponse> TriggerServerRestart()
        {
            if (!serverControl.WatchdogPresent)
            {
                Logger.LogDebug("Restart request failed due to lack of host watchdog!");
                return new AuthorityResponse(
                    new ErrorMessageResponse(ErrorCode.MissingHostWatchdog),
                    HttpFailureResponse.UnprocessableEntity);
            }
 
            await serverControl.Restart();
            return new AuthorityResponse();
        }
    }
}