DelegatingEngineInstaller.cs

Go to the documentation of this file.

using System;
using System.Collections.Frozen;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Host.Jobs;
 
namespace Tgstation.Server.Host.Components.Engine
{
    sealed class DelegatingEngineInstaller : IEngineInstaller
    {
        readonly FrozenDictionary<EngineType, IEngineInstaller> delegatedInstallers;
 
        public DelegatingEngineInstaller(FrozenDictionary<EngineType, IEngineInstaller> delegatedInstallers)
        {
            this.delegatedInstallers = delegatedInstallers ?? throw new ArgumentNullException(nameof(delegatedInstallers));
        }
 
        public Task CleanCache(CancellationToken cancellationToken)
            => Task.WhenAll(delegatedInstallers.Values.Select(installer => installer.CleanCache(cancellationToken)));
 
        public ValueTask<IEngineInstallation> CreateInstallation(EngineVersion version, string path, Task installationTask, CancellationToken cancellationToken)
            => DelegateCall(version, installer => installer.CreateInstallation(version, path, installationTask, cancellationToken));
 
        public ValueTask<IEngineInstallationData> DownloadVersion(EngineVersion version, JobProgressReporter jobProgressReporter, CancellationToken cancellationToken)
            => DelegateCall(version, installer => installer.DownloadVersion(version, jobProgressReporter, cancellationToken));
 
        public ValueTask Install(EngineVersion version, string path, bool deploymentPipelineProcesses, CancellationToken cancellationToken)
            => DelegateCall(version, installer => installer.Install(version, path, deploymentPipelineProcesses, cancellationToken));
 
        public ValueTask TrustDmbPath(EngineVersion version, string fullDmbPath, CancellationToken cancellationToken)
            => DelegateCall(version, installer => installer.TrustDmbPath(version, fullDmbPath, cancellationToken));
 
        public ValueTask UpgradeInstallation(EngineVersion version, string path, CancellationToken cancellationToken)
            => DelegateCall(version, installer => installer.UpgradeInstallation(version, path, cancellationToken));
 
        TReturn DelegateCall<TReturn>(EngineVersion version, Func<IEngineInstaller, TReturn> call)
        {
            ArgumentNullException.ThrowIfNull(version);
            return call(delegatedInstallers[version.Engine!.Value]);
        }
    }
}