RemoteDeploymentManagerFactory.cs

Go to the documentation of this file.

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
 
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Host.Components.Repository;
using Tgstation.Server.Host.Database;
using Tgstation.Server.Host.Utils.GitHub;
 
namespace Tgstation.Server.Host.Components.Deployment.Remote
{
    sealed class RemoteDeploymentManagerFactory : IRemoteDeploymentManagerFactory
    {
        readonly IDatabaseContextFactory databaseContextFactory;
 
        readonly IGitHubServiceFactory gitHubServiceFactory;
 
        readonly IGitRemoteFeaturesFactory gitRemoteFeaturesFactory;
 
        readonly ILoggerFactory loggerFactory;
 
        readonly ILogger<RemoteDeploymentManagerFactory> logger;
 
        readonly ConcurrentDictionary<long, Action<bool>> activationCallbacks;
 
        public RemoteDeploymentManagerFactory(
            IDatabaseContextFactory databaseContextFactory,
            IGitHubServiceFactory gitHubServiceFactory,
            IGitRemoteFeaturesFactory gitRemoteFeaturesFactory,
            ILoggerFactory loggerFactory,
            ILogger<RemoteDeploymentManagerFactory> logger)
        {
            this.databaseContextFactory = databaseContextFactory ?? throw new ArgumentNullException(nameof(databaseContextFactory));
            this.gitHubServiceFactory = gitHubServiceFactory ?? throw new ArgumentNullException(nameof(gitHubServiceFactory));
            this.gitRemoteFeaturesFactory = gitRemoteFeaturesFactory ?? throw new ArgumentNullException(nameof(gitRemoteFeaturesFactory));
            this.loggerFactory = loggerFactory ?? throw new ArgumentNullException(nameof(loggerFactory));
            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
 
            activationCallbacks = new ConcurrentDictionary<long, Action<bool>>();
        }
 
        public IRemoteDeploymentManager CreateRemoteDeploymentManager(Api.Models.Instance metadata, RemoteGitProvider remoteGitProvider)
        {
            ArgumentNullException.ThrowIfNull(metadata);
 
            logger.LogTrace("Creating remote deployment manager for remote git provider {remoteGitProvider}...", remoteGitProvider);
            return remoteGitProvider switch
            {
                RemoteGitProvider.GitHub => new GitHubRemoteDeploymentManager(
                    databaseContextFactory,
                    gitHubServiceFactory,
                    loggerFactory.CreateLogger<GitHubRemoteDeploymentManager>(),
                    metadata,
                    activationCallbacks),
                RemoteGitProvider.GitLab => new GitLabRemoteDeploymentManager(
                    loggerFactory.CreateLogger<GitLabRemoteDeploymentManager>(),
                    metadata,
                    activationCallbacks),
                RemoteGitProvider.Unknown => new NoOpRemoteDeploymentManager(
                    loggerFactory.CreateLogger<NoOpRemoteDeploymentManager>(),
                    metadata,
                    activationCallbacks),
                _ => throw new InvalidOperationException($"Invalid RemoteGitProvider: {remoteGitProvider}!"),
            };
        }
 
        public IRemoteDeploymentManager CreateRemoteDeploymentManager(Api.Models.Instance metadata, Models.CompileJob compileJob)
        {
            ArgumentNullException.ThrowIfNull(compileJob);
 
            RemoteGitProvider remoteGitProvider;
 
            // Pre 4.7.X
            if (compileJob.RepositoryOrigin == null)
                remoteGitProvider = RemoteGitProvider.Unknown;
            else
                remoteGitProvider = gitRemoteFeaturesFactory.ParseRemoteGitProviderFromOrigin(
                    new Uri(
                        compileJob.RepositoryOrigin));
 
            return CreateRemoteDeploymentManager(metadata, remoteGitProvider);
        }
 
        public void ForgetLocalStateForCompileJobs(IEnumerable<long> compileJobIds)
        {
            ArgumentNullException.ThrowIfNull(compileJobIds);
 
            foreach (var compileJobId in compileJobIds)
                activationCallbacks.Remove(compileJobId, out _);
        }
    }
}