AdministrationMutations.cs

Go to the documentation of this file.

using System;
using System.Threading;
using System.Threading.Tasks;
 
using HotChocolate;
using HotChocolate.Types;
 
using Tgstation.Server.Api.Models.Response;
using Tgstation.Server.Api.Rights;
using Tgstation.Server.Host.Authority;
using Tgstation.Server.Host.GraphQL.Scalars;
using Tgstation.Server.Host.Security;
 
namespace Tgstation.Server.Host.GraphQL.Mutations
{
    [ExtendObjectType(typeof(Mutation))]
    [GraphQLDescription(Mutation.GraphQLDescription)]
    public sealed class AdministrationMutations
    {
        [TgsGraphQLAuthorize<IAdministrationAuthority>(nameof(IAdministrationAuthority.TriggerServerRestart))]
        [Error(typeof(ErrorMessageException))]
        public async ValueTask<Query> RestartServerNode(
            [Service] IGraphQLAuthorityInvoker<IAdministrationAuthority> administrationAuthority)
        {
            ArgumentNullException.ThrowIfNull(administrationAuthority);
            await administrationAuthority.Invoke(
                authority => authority.TriggerServerRestart());
 
            return new Query();
        }
 
        [TgsGraphQLAuthorize(AdministrationRights.ChangeVersion)]
        [Error(typeof(ErrorMessageException))]
        public async ValueTask<Query> ChangeServerNodeVersionViaTrackedRepository(
            Version targetVersion,
            [Service] IGraphQLAuthorityInvoker<IAdministrationAuthority> administrationAuthority,
            CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(targetVersion);
            ArgumentNullException.ThrowIfNull(administrationAuthority);
            await administrationAuthority.Invoke<ServerUpdateResponse, ServerUpdateResponse>(
                authority => authority.TriggerServerVersionChange(targetVersion, false, cancellationToken));
            return new Query();
        }
 
        [TgsGraphQLAuthorize(AdministrationRights.UploadVersion)]
        [Error(typeof(ErrorMessageException))]
        [GraphQLType<FileUploadTicketType>]
        public async ValueTask<string> ChangeServerNodeVersionViaUpload(
            Version targetVersion,
            [Service] IGraphQLAuthorityInvoker<IAdministrationAuthority> administrationAuthority,
            CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(targetVersion);
            ArgumentNullException.ThrowIfNull(administrationAuthority);
            var response = await administrationAuthority.Invoke<ServerUpdateResponse, FileTicketResponse>(
                authority => authority.TriggerServerVersionChange(targetVersion, true, cancellationToken));
 
            return response.FileTicket ?? throw new InvalidOperationException("Administration authority did not generate a FileUploadTicket!");
        }
    }
}