BaseRemoteDeploymentManager.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.Extensions.Logging;
 
using Tgstation.Server.Host.Components.Repository;
using Tgstation.Server.Host.Models;
 
namespace Tgstation.Server.Host.Components.Deployment.Remote
{
    abstract class BaseRemoteDeploymentManager : IRemoteDeploymentManager
    {
        protected Api.Models.Instance Metadata { get; }
 
        protected ILogger<BaseRemoteDeploymentManager> Logger { get; }
 
        protected BaseRemoteDeploymentManager(ILogger<BaseRemoteDeploymentManager> logger, Api.Models.Instance metadata)
        {
            Logger = logger ?? throw new ArgumentNullException(nameof(logger));
            Metadata = metadata ?? throw new ArgumentNullException(nameof(metadata));
        }
 
        public async Task PostDeploymentComments(
            CompileJob compileJob,
            RevisionInformation previousRevisionInformation,
            RepositorySettings repositorySettings,
            string repoOwner,
            string repoName,
            CancellationToken cancellationToken)
        {
            if (repositorySettings?.AccessToken == null)
                return;
 
            var deployedRevisionInformation = compileJob.RevisionInformation;
            if ((previousRevisionInformation != null && previousRevisionInformation.CommitSha == deployedRevisionInformation.CommitSha)
                || !repositorySettings.PostTestMergeComment.Value)
                return;
 
            previousRevisionInformation ??= new RevisionInformation();
            previousRevisionInformation.ActiveTestMerges ??= new List<RevInfoTestMerge>();
 
            deployedRevisionInformation.ActiveTestMerges ??= new List<RevInfoTestMerge>();
            var tasks = new List<Task>();
 
            // added prs
            var addedTestMerges = deployedRevisionInformation
                .ActiveTestMerges
                .Select(x => x.TestMerge)
                .Where(x => !previousRevisionInformation
                    .ActiveTestMerges
                    .Any(y => y.TestMerge.Number == x.Number))
                .ToList();
            var removedTestMerges = previousRevisionInformation
                .ActiveTestMerges
                .Select(x => x.TestMerge)
                .Where(x => !deployedRevisionInformation
                    .ActiveTestMerges
                    .Any(y => y.TestMerge.Number == x.Number))
                .ToList();
            var updatedTestMerges = deployedRevisionInformation
                .ActiveTestMerges
                .Select(x => x.TestMerge)
                .Where(x => previousRevisionInformation
                    .ActiveTestMerges
                    .Any(y => y.TestMerge.Number == x.Number))
                .ToList();
 
            if (!addedTestMerges.Any() && !removedTestMerges.Any() && !updatedTestMerges.Any())
                return;
 
            Logger.LogTrace(
                "Commenting on {0} added, {1} removed, and {2} updated test merge sources...",
                addedTestMerges.Count,
                removedTestMerges.Count,
                updatedTestMerges.Count);
            foreach (var addedTestMerge in addedTestMerges)
                tasks.Add(
                    CommentOnTestMergeSource(
                        repositorySettings,
                        repoOwner,
                        repoName,
                        FormatTestMerge(
                            repositorySettings,
                            compileJob,
                            addedTestMerge,
                            repoOwner,
                            repoName,
                            false),
                        addedTestMerge.Number,
                        cancellationToken));
 
            foreach (var removedTestMerge in removedTestMerges)
                tasks.Add(
                    CommentOnTestMergeSource(
                        repositorySettings,
                        repoOwner,
                        repoName,
                        "#### Test Merge Removed",
                        removedTestMerge.Number,
                        cancellationToken));
 
            foreach (var updatedTestMerge in updatedTestMerges)
                tasks.Add(
                    CommentOnTestMergeSource(
                        repositorySettings,
                        repoOwner,
                        repoName,
                        FormatTestMerge(
                            repositorySettings,
                            compileJob,
                            updatedTestMerge,
                            repoOwner,
                            repoName,
                            true),
                        updatedTestMerge.Number,
                        cancellationToken));
 
            if (tasks.Any())
                await Task.WhenAll(tasks);
        }
 
        public abstract Task ApplyDeployment(CompileJob compileJob, CompileJob oldCompileJob, CancellationToken cancellationToken);
 
        public abstract Task FailDeployment(CompileJob compileJob, string errorMessage, CancellationToken cancellationToken);
 
        public abstract Task MarkInactive(CompileJob compileJob, CancellationToken cancellationToken);
 
        public abstract Task<IReadOnlyCollection<TestMerge>> RemoveMergedTestMerges(
            IRepository repository,
            RepositorySettings repositorySettings,
            RevisionInformation revisionInformation,
            CancellationToken cancellationToken);
 
        public abstract Task StageDeployment(CompileJob compileJob, CancellationToken cancellationToken);
 
        public abstract Task StartDeployment(
            Api.Models.Internal.IGitRemoteInformation remoteInformation,
            CompileJob compileJob,
            CancellationToken cancellationToken);
 
        protected abstract string FormatTestMerge(
            RepositorySettings repositorySettings,
            CompileJob compileJob,
            TestMerge testMerge,
            string remoteRepositoryOwner,
            string remoteRepositoryName,
            bool updated);
 
        protected abstract Task CommentOnTestMergeSource(
            RepositorySettings repositorySettings,
            string remoteRepositoryOwner,
            string remoteRepositoryName,
            string comment,
            int testMergeNumber,
            CancellationToken cancellationToken);
    }
}