PullRequestsCommand.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.EntityFrameworkCore;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Host.Components.Deployment;
using Tgstation.Server.Host.Components.Interop;
using Tgstation.Server.Host.Components.Repository;
using Tgstation.Server.Host.Components.Watchdog;
using Tgstation.Server.Host.Database;
 
namespace Tgstation.Server.Host.Components.Chat.Commands
{
    sealed class PullRequestsCommand : ICommand
    {
        public string Name => "prs";
 
        public string HelpText => "Display live test merge numbers. Add --repo to view test merges in the repository as opposed to live. Add --staged to view PRs in the upcoming deployment.";
 
        public bool AdminOnly => false;
 
        readonly IWatchdog watchdog;
 
        readonly IRepositoryManager repositoryManager;
 
        readonly IDatabaseContextFactory databaseContextFactory;
 
        readonly ILatestCompileJobProvider compileJobProvider;
 
        readonly Models.Instance instance;
 
        public PullRequestsCommand(
            IWatchdog watchdog,
            IRepositoryManager repositoryManager,
            IDatabaseContextFactory databaseContextFactory,
            ILatestCompileJobProvider compileJobProvider,
            Models.Instance instance)
        {
            this.watchdog = watchdog ?? throw new ArgumentNullException(nameof(watchdog));
            this.repositoryManager = repositoryManager ?? throw new ArgumentNullException(nameof(repositoryManager));
            this.databaseContextFactory = databaseContextFactory ?? throw new ArgumentNullException(nameof(databaseContextFactory));
            this.compileJobProvider = compileJobProvider ?? throw new ArgumentNullException(nameof(compileJobProvider));
            this.instance = instance ?? throw new ArgumentNullException(nameof(instance));
        }
 
        // TODO: Decomplexify
#pragma warning disable CA1506
        public async ValueTask<MessageContent> Invoke(string arguments, ChatUser user, CancellationToken cancellationToken)
        {
            IEnumerable<Models.TestMerge> results;
            var splits = arguments.Split(' ');
            var hasRepo = splits.Any(x => x.Equals("--repo", StringComparison.OrdinalIgnoreCase));
            var hasStaged = splits.Any(x => x.Equals("--staged", StringComparison.OrdinalIgnoreCase));
 
            if (hasStaged && hasRepo)
                return new MessageContent
                {
                    Text = "Please use only one of `--staged` or `--repo`",
                };
 
            if (hasRepo)
            {
                if (repositoryManager.CloneInProgress || repositoryManager.InUse)
                    return new MessageContent
                    {
                        Text = "Repository busy! Try again later",
                    };
 
                string head;
                using (var repo = await repositoryManager.LoadRepository(cancellationToken))
                {
                    if (repo == null)
                        return new MessageContent
                        {
                            Text = "Repository unavailable!",
                        };
 
                    head = repo.Head;
                }
 
                results = null!;
                await databaseContextFactory.UseContext(
                    async db => results = await db
                        .RevisionInformations
                        .AsQueryable()
                        .Where(x => x.Instance!.Id == instance.Id && x.CommitSha == head)
                        .SelectMany(x => x.ActiveTestMerges!)
                        .Select(x => x.TestMerge)
                        .Select(x => new Models.TestMerge
                        {
                            Number = x.Number,
                            TargetCommitSha = x.TargetCommitSha,
                        })
                        .ToListAsync(cancellationToken));
            }
            else if (watchdog.Status == WatchdogStatus.Offline)
                return new MessageContent
                {
                    Text = "Server offline!",
                };
            else
            {
                var compileJobToUse = watchdog.ActiveCompileJob;
                if (hasStaged)
                {
                    var latestCompileJob = await compileJobProvider.LatestCompileJob();
                    if (latestCompileJob?.Id != compileJobToUse?.Id)
                        compileJobToUse = latestCompileJob;
                    else
                        compileJobToUse = null;
                }
 
                results = compileJobToUse?.RevisionInformation.ActiveTestMerges?.Select(x => x.TestMerge).ToList() ?? Enumerable.Empty<Models.TestMerge>();
            }
 
            return new MessageContent
            {
                Text = !results.Any()
                    ? "None!"
                    : String.Join(
                        ", ",
                        results.Select(
                            x => $"#{x.Number} at {x.TargetCommitSha![..7]}")),
            };
        }
#pragma warning restore CA1506
    }
}