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.Host.Components.Repository;
using Tgstation.Server.Host.Components.Watchdog;
using Tgstation.Server.Host.Core;

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));
        }

        public async Task<string> Invoke(string arguments, User 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.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.Running)
                    return "Server offline!";
                results = watchdog.ActiveCompileJob?.RevisionInformation.ActiveTestMerges.Select(x => x.TestMerge).ToList() ?? new List<Models.TestMerge>();
            }

            if (!results.Any())
                return "None!";

            return String.Join(", ", results.Select(x => String.Format(CultureInfo.InvariantCulture, "#{0} at {1}", x.Number, x.PullRequestRevision.Substring(0, 7))));
        }
    }
}