EngineCommand.cs

Go to the documentation of this file.

using System;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Host.Components.Engine;
using Tgstation.Server.Host.Components.Interop;
using Tgstation.Server.Host.Components.Watchdog;
 
namespace Tgstation.Server.Host.Components.Chat.Commands
{
    sealed class EngineCommand : ICommand
    {
        public string Name => "engine";
 
        public string HelpText => "Displays the running engine version. Use --active for the version used in future deployments";
 
        public bool AdminOnly => false;
 
        readonly IEngineManager engineManager;
 
        readonly IWatchdog watchdog;
 
        public EngineCommand(IEngineManager engineManager, IWatchdog watchdog)
        {
            this.engineManager = engineManager ?? throw new ArgumentNullException(nameof(engineManager));
            this.watchdog = watchdog ?? throw new ArgumentNullException(nameof(watchdog));
        }
 
        public ValueTask<MessageContent> Invoke(string arguments, ChatUser user, CancellationToken cancellationToken)
        {
            EngineVersion? engineVersion;
            if (arguments.Split(' ').Any(x => x.Equals("--active", StringComparison.OrdinalIgnoreCase)))
                engineVersion = engineManager.ActiveVersion;
            else
            {
                if (watchdog.Status == WatchdogStatus.Offline)
                    return ValueTask.FromResult(
                        new MessageContent
                        {
                            Text = "Server offline!",
                        });
 
                if (watchdog.ActiveCompileJob == null)
                    return ValueTask.FromResult(
                        new MessageContent
                        {
                            Text = "None!",
                        });
 
                engineVersion = EngineVersion.Parse(watchdog.ActiveCompileJob.EngineVersion);
            }
 
            string text;
            if (engineVersion == null)
                text = "None!";
            else
            {
                text = engineVersion.Engine!.Value switch
                {
                    EngineType.OpenDream => $"OpenDream: {engineVersion.SourceSHA}",
                    EngineType.Byond => $"BYOND {engineVersion.Version!.Major}.{engineVersion.Version.Minor}",
                    _ => throw new InvalidOperationException($"Invalid EngineType: {engineVersion.Engine.Value}"),
                };
 
                if (engineVersion.CustomIteration.HasValue)
                    text += $" (Custom Upload #{engineVersion.CustomIteration.Value})";
            }
 
            return ValueTask.FromResult(
                new MessageContent
                {
                    Text = text,
                });
        }
    }
}