RevisionCommand.cs

Go to the documentation of this file.

using System;
using System.Globalization;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Host.Components.Interop;
using Tgstation.Server.Host.Components.Repository;
using Tgstation.Server.Host.Components.Watchdog;
 
namespace Tgstation.Server.Host.Components.Chat.Commands
{
    sealed class RevisionCommand : ICommand
    {
        public string Name => "revision";
 
        public string HelpText => "Display live origin commit sha. Add --repo to view repository revision";
 
        public bool AdminOnly => false;
 
        readonly IWatchdog watchdog;
 
        readonly IRepositoryManager repositoryManager;
 
        public RevisionCommand(IWatchdog watchdog, IRepositoryManager repositoryManager)
        {
            this.watchdog = watchdog ?? throw new ArgumentNullException(nameof(watchdog));
            this.repositoryManager = repositoryManager ?? throw new ArgumentNullException(nameof(repositoryManager));
        }
 
        public async ValueTask<MessageContent> Invoke(string arguments, ChatUser user, CancellationToken cancellationToken)
        {
            string result;
            if (arguments.Split(' ').Any(x => x.ToUpperInvariant() == "--REPO"))
            {
                if (repositoryManager.CloneInProgress || repositoryManager.InUse)
                    return new MessageContent
                    {
                        Text = "Repository busy! Try again later",
                    };
 
                using (var repo = await repositoryManager.LoadRepository(cancellationToken))
                {
                    if (repo == null)
                        return new MessageContent
                        {
                            Text = "Repository unavailable!",
                        };
                    result = repo.Head;
                }
            }
            else
            {
                if (watchdog.Status == WatchdogStatus.Offline)
                    return new MessageContent
                    {
                        Text = "Server offline!",
                    };
                result = watchdog.ActiveCompileJob?.RevisionInformation.OriginCommitSha!;
            }
 
            return new MessageContent
            {
                Text = String.Format(CultureInfo.InvariantCulture, "^{0}", result),
            };
        }
    }
}