PullRequestsCommand.cs

Go to the documentation of this file.

using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using Tgstation.Server.Api.Models;
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 pull request numbers. Add --repo to view repository test merges";

        public bool AdminOnly => false;

        readonly IWatchdog watchdog;

        readonly IRepositoryManager repositoryManager;

        readonly IDatabaseContextFactory databaseContextFactory;

        readonly Models.Instance instance;

        public PullRequestsCommand(IWatchdog watchdog, IRepositoryManager repositoryManager, IDatabaseContextFactory databaseContextFactory, 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.instance = instance ?? throw new ArgumentNullException(nameof(instance));
        }

        // TODO: Decomplexify
        #pragma warning disable CA1506
        public async Task<string> Invoke(string arguments, ChatUser user, CancellationToken cancellationToken)
        {
            IEnumerable<Models.TestMerge> results = null;
            if (arguments.Split(' ').Any(x => x.ToUpperInvariant() == "--REPO"))
            {
                string head;
                using (var repo = await repositoryManager.LoadRepository(cancellationToken).ConfigureAwait(false))
                {
                    if (repo == null)
                        return "Repository unavailable!";
                    head = repo.Head;
                }

                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,
                            PullRequestRevision = x.PullRequestRevision
                        })
                        .ToListAsync(cancellationToken)
                        .ConfigureAwait(false))
                    .ConfigureAwait(false);
            }
            else
            {
                if (watchdog.Status == WatchdogStatus.Offline)
                    return "Server offline!";
                results = watchdog.ActiveCompileJob?.RevisionInformation.ActiveTestMerges.Select(x => x.TestMerge).ToList() ?? new List<Models.TestMerge>();
            }

            return !results.Any() ? "None!" : String.Join(", ", results.Select(x => String.Format(CultureInfo.InvariantCulture, "#{0} at {1}", x.Number, x.PullRequestRevision.Substring(0, 7))));
        }
        #pragma warning restore CA1506
    }
}