DreamMaker.cs

Go to the documentation of this file.

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
 
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging;
 
using Tgstation.Server.Api.Models;
using Tgstation.Server.Api.Models.Internal;
using Tgstation.Server.Common.Extensions;
using Tgstation.Server.Host.Components.Chat;
using Tgstation.Server.Host.Components.Deployment.Remote;
using Tgstation.Server.Host.Components.Engine;
using Tgstation.Server.Host.Components.Events;
using Tgstation.Server.Host.Components.Repository;
using Tgstation.Server.Host.Components.Session;
using Tgstation.Server.Host.Configuration;
using Tgstation.Server.Host.Database;
using Tgstation.Server.Host.IO;
using Tgstation.Server.Host.Jobs;
using Tgstation.Server.Host.Models;
using Tgstation.Server.Host.System;
using Tgstation.Server.Host.Utils;
 
namespace Tgstation.Server.Host.Components.Deployment
{
    sealed class DreamMaker : IDreamMaker
    {
        const string DmeExtension = "dme";
 
        readonly IEngineManager engineManager;
 
        readonly IIOManager ioManager;
 
        readonly StaticFiles.IConfiguration configuration;
 
        readonly ISessionControllerFactory sessionControllerFactory;
 
        readonly IEventConsumer eventConsumer;
 
        readonly IChatManager chatManager;
 
        readonly IProcessExecutor processExecutor;
 
        readonly IRepositoryManager repositoryManager;
 
        readonly ICompileJobSink compileJobConsumer;
 
        readonly IRemoteDeploymentManagerFactory remoteDeploymentManagerFactory;
 
        readonly IAsyncDelayer asyncDelayer;
 
        readonly ILogger<DreamMaker> logger;
 
        readonly SessionConfiguration sessionConfiguration;
 
        readonly Api.Models.Instance metadata;
 
        readonly object deploymentLock;
 
        Func<string?, string, Action<bool>>? currentChatCallback;
 
        string? currentDreamMakerOutput;
 
        bool deploying;
 
        static string FormatExceptionForUsers(Exception exception)
            => exception is OperationCanceledException
                ? "The job was cancelled!"
                : exception.Message;
 
        public DreamMaker(
            IEngineManager engineManager,
            IIOManager ioManager,
            StaticFiles.IConfiguration configuration,
            ISessionControllerFactory sessionControllerFactory,
            IEventConsumer eventConsumer,
            IChatManager chatManager,
            IProcessExecutor processExecutor,
            ICompileJobSink compileJobConsumer,
            IRepositoryManager repositoryManager,
            IRemoteDeploymentManagerFactory remoteDeploymentManagerFactory,
            IAsyncDelayer asyncDelayer,
            ILogger<DreamMaker> logger,
            SessionConfiguration sessionConfiguration,
            Api.Models.Instance metadata)
        {
            this.engineManager = engineManager ?? throw new ArgumentNullException(nameof(engineManager));
            this.ioManager = ioManager ?? throw new ArgumentNullException(nameof(ioManager));
            this.configuration = configuration ?? throw new ArgumentNullException(nameof(configuration));
            this.sessionControllerFactory = sessionControllerFactory ?? throw new ArgumentNullException(nameof(sessionControllerFactory));
            this.eventConsumer = eventConsumer ?? throw new ArgumentNullException(nameof(eventConsumer));
            this.chatManager = chatManager ?? throw new ArgumentNullException(nameof(chatManager));
            this.processExecutor = processExecutor ?? throw new ArgumentNullException(nameof(processExecutor));
            this.compileJobConsumer = compileJobConsumer ?? throw new ArgumentNullException(nameof(compileJobConsumer));
            this.repositoryManager = repositoryManager ?? throw new ArgumentNullException(nameof(repositoryManager));
            this.asyncDelayer = asyncDelayer ?? throw new ArgumentNullException(nameof(asyncDelayer));
            this.remoteDeploymentManagerFactory = remoteDeploymentManagerFactory ?? throw new ArgumentNullException(nameof(remoteDeploymentManagerFactory));
            this.logger = logger ?? throw new ArgumentNullException(nameof(logger));
            this.sessionConfiguration = sessionConfiguration ?? throw new ArgumentNullException(nameof(sessionConfiguration));
            this.metadata = metadata ?? throw new ArgumentNullException(nameof(metadata));
 
            deploymentLock = new object();
        }
 
#pragma warning disable CA1506
        public async ValueTask DeploymentProcess(
            Models.Job job,
            IDatabaseContextFactory databaseContextFactory,
            JobProgressReporter progressReporter,
            CancellationToken cancellationToken)
        {
            ArgumentNullException.ThrowIfNull(job);
            ArgumentNullException.ThrowIfNull(databaseContextFactory);
            ArgumentNullException.ThrowIfNull(progressReporter);
 
            lock (deploymentLock)
            {
                if (deploying)
                    throw new JobException(ErrorCode.DeploymentInProgress);
                deploying = true;
            }
 
            currentChatCallback = null;
            currentDreamMakerOutput = null;
            Models.CompileJob? compileJob = null;
            try
            {
                string? repoOwner = null;
                string? repoName = null;
                TimeSpan? averageSpan = null;
                Models.RepositorySettings? repositorySettings = null;
                Models.DreamDaemonSettings? ddSettings = null;
                Models.DreamMakerSettings? dreamMakerSettings = null;
                IRepository? repo = null;
                IRemoteDeploymentManager? remoteDeploymentManager = null;
                Models.RevisionInformation? revInfo = null;
                await databaseContextFactory.UseContext(
                    async databaseContext =>
                    {
                        averageSpan = await CalculateExpectedDeploymentTime(databaseContext, cancellationToken);
 
                        ddSettings = await databaseContext
                            .DreamDaemonSettings
                            .AsQueryable()
                            .Where(x => x.InstanceId == metadata.Id)
                            .Select(x => new Models.DreamDaemonSettings
                            {
                                StartupTimeout = x.StartupTimeout,
                                LogOutput = x.LogOutput,
                            })
                            .FirstOrDefaultAsync(cancellationToken);
                        if (ddSettings == default)
                            throw new JobException(ErrorCode.InstanceMissingDreamDaemonSettings);
 
                        dreamMakerSettings = await databaseContext
                            .DreamMakerSettings
                            .AsQueryable()
                            .Where(x => x.InstanceId == metadata.Id)
                            .FirstAsync(cancellationToken);
                        if (dreamMakerSettings == default)
                            throw new JobException(ErrorCode.InstanceMissingDreamMakerSettings);
 
                        repositorySettings = await databaseContext
                            .RepositorySettings
                            .AsQueryable()
                            .Where(x => x.InstanceId == metadata.Id)
                            .Select(x => new Models.RepositorySettings
                            {
                                AccessToken = x.AccessToken,
                                AccessUser = x.AccessUser,
                                ShowTestMergeCommitters = x.ShowTestMergeCommitters,
                                PushTestMergeCommits = x.PushTestMergeCommits,
                                PostTestMergeComment = x.PostTestMergeComment,
                            })
                            .FirstOrDefaultAsync(cancellationToken);
                        if (repositorySettings == default)
                            throw new JobException(ErrorCode.InstanceMissingRepositorySettings);
 
                        repo = await repositoryManager.LoadRepository(cancellationToken);
                        try
                        {
                            if (repo == null)
                                throw new JobException(ErrorCode.RepoMissing);
 
                            remoteDeploymentManager = remoteDeploymentManagerFactory
                                .CreateRemoteDeploymentManager(metadata, repo.RemoteGitProvider!.Value);
 
                            var repoSha = repo.Head;
                            repoOwner = repo.RemoteRepositoryOwner;
                            repoName = repo.RemoteRepositoryName;
                            revInfo = await databaseContext
                                .RevisionInformations
                                .AsQueryable()
                                .Where(x => x.CommitSha == repoSha && x.InstanceId == metadata.Id)
                                .Include(x => x.ActiveTestMerges!)
                                    .ThenInclude(x => x.TestMerge!)
                                        .ThenInclude(x => x.MergedBy)
                                .FirstOrDefaultAsync(cancellationToken);
 
                            if (revInfo == null)
                            {
                                revInfo = new Models.RevisionInformation
                                {
                                    CommitSha = repoSha,
                                    Timestamp = await repo.TimestampCommit(repoSha, cancellationToken),
                                    OriginCommitSha = repoSha,
                                    Instance = new Models.Instance
                                    {
                                        Id = metadata.Id,
                                    },
                                    ActiveTestMerges = new List<RevInfoTestMerge>(),
                                };
 
                                logger.LogInformation(Repository.Repository.OriginTrackingErrorTemplate, repoSha);
                                databaseContext.RevisionInformations.Add(revInfo);
                                databaseContext.Instances.Attach(revInfo.Instance);
                                await databaseContext.Save(cancellationToken);
                            }
                        }
                        catch
                        {
                            repo?.Dispose();
                            throw;
                        }
                    });
 
                var likelyPushedTestMergeCommit =
                    repositorySettings!.PushTestMergeCommits!.Value
                    && repositorySettings.AccessToken != null
                    && repositorySettings.AccessUser != null;
                using (repo)
                    compileJob = await Compile(
                        job,
                        revInfo!,
                        dreamMakerSettings!,
                        ddSettings!,
                        repo!,
                        remoteDeploymentManager!,
                        progressReporter,
                        averageSpan,
                        likelyPushedTestMergeCommit,
                        cancellationToken);
 
                var activeCompileJob = await compileJobConsumer.LatestCompileJob();
                try
                {
                    await databaseContextFactory.UseContext(
                        async databaseContext =>
                        {
                            var fullJob = compileJob.Job;
                            compileJob.Job = new Models.Job(job.Require(x => x.Id));
                            var fullRevInfo = compileJob.RevisionInformation;
                            compileJob.RevisionInformation = new Models.RevisionInformation
                            {
                                Id = revInfo!.Id,
                            };
 
                            databaseContext.Jobs.Attach(compileJob.Job);
                            databaseContext.RevisionInformations.Attach(compileJob.RevisionInformation);
                            databaseContext.CompileJobs.Add(compileJob);
 
                            // The difficulty with compile jobs is they have a two part commit
                            await databaseContext.Save(cancellationToken);
                            logger.LogTrace("Created CompileJob {compileJobId}", compileJob.Id);
                            try
                            {
                                var chatNotificationAction = currentChatCallback!(null, compileJob.Output!);
                                await compileJobConsumer.LoadCompileJob(compileJob, chatNotificationAction, cancellationToken);
                            }
                            catch
                            {
                                // So we need to un-commit the compile job if the above throws
                                databaseContext.CompileJobs.Remove(compileJob);
 
                                // DCT: Cancellation token is for job, operation must run regardless
                                await databaseContext.Save(CancellationToken.None);
                                throw;
                            }
 
                            compileJob.Job = fullJob;
                            compileJob.RevisionInformation = fullRevInfo;
                        });
                }
                catch (Exception ex)
                {
                    await CleanupFailedCompile(compileJob, remoteDeploymentManager!, ex);
                    throw;
                }
 
                var commentsTask = remoteDeploymentManager!.PostDeploymentComments(
                    compileJob,
                    activeCompileJob?.RevisionInformation,
                    repositorySettings,
                    repoOwner,
                    repoName,
                    cancellationToken);
 
                var eventTask = eventConsumer.HandleEvent(EventType.DeploymentComplete, Enumerable.Empty<string>(), false, cancellationToken);
 
                try
                {
                    await ValueTaskExtensions.WhenAll(commentsTask, eventTask);
                }
                catch (Exception ex)
                {
                    throw new JobException(ErrorCode.PostDeployFailure, ex);
                }
                finally
                {
                    currentChatCallback = null;
                }
            }
            catch (Exception ex)
            {
                currentChatCallback?.Invoke(
                    FormatExceptionForUsers(ex),
                    currentDreamMakerOutput!);
 
                throw;
            }
            finally
            {
                deploying = false;
            }
        }
#pragma warning restore CA1506
 
        async ValueTask<TimeSpan?> CalculateExpectedDeploymentTime(IDatabaseContext databaseContext, CancellationToken cancellationToken)
        {
            var previousCompileJobs = await databaseContext
                .CompileJobs
                .AsQueryable()
                .Where(x => x.Job.Instance!.Id == metadata.Id)
                .OrderByDescending(x => x.Job.StoppedAt)
                .Take(10)
                .Select(x => new
                {
                    StoppedAt = x.Job.StoppedAt!.Value,
                    StartedAt = x.Job.StartedAt!.Value,
                })
                .ToListAsync(cancellationToken);
 
            TimeSpan? averageSpan = null;
            if (previousCompileJobs.Count != 0)
            {
                var totalSpan = TimeSpan.Zero;
                foreach (var previousCompileJob in previousCompileJobs)
                    totalSpan += previousCompileJob.StoppedAt - previousCompileJob.StartedAt;
                averageSpan = totalSpan / previousCompileJobs.Count;
            }
 
            return averageSpan;
        }
 
        async ValueTask<Models.CompileJob> Compile(
            Models.Job job,
            Models.RevisionInformation revisionInformation,
            Api.Models.Internal.DreamMakerSettings dreamMakerSettings,
            DreamDaemonLaunchParameters launchParameters,
            IRepository repository,
            IRemoteDeploymentManager remoteDeploymentManager,
            JobProgressReporter progressReporter,
            TimeSpan? estimatedDuration,
            bool localCommitExistsOnRemote,
            CancellationToken cancellationToken)
        {
            logger.LogTrace("Begin Compile");
 
            using var progressCts = CancellationTokenSource.CreateLinkedTokenSource(cancellationToken);
 
            progressReporter.StageName = "Reserving BYOND version";
            var progressTask = ProgressTask(progressReporter, estimatedDuration, progressCts.Token);
            try
            {
                using var engineLock = await engineManager.UseExecutables(null, null, cancellationToken);
                currentChatCallback = chatManager.QueueDeploymentMessage(
                    revisionInformation,
                    engineLock.Version,
                    DateTimeOffset.UtcNow + estimatedDuration,
                    repository.RemoteRepositoryOwner,
                    repository.RemoteRepositoryName,
                    localCommitExistsOnRemote);
 
                var compileJob = new Models.CompileJob(job, revisionInformation, engineLock.Version.ToString())
                {
                    DirectoryName = Guid.NewGuid(),
                    DmeName = dreamMakerSettings.ProjectName,
                    RepositoryOrigin = repository.Origin.ToString(),
                };
 
                progressReporter.StageName = "Creating remote deployment notification";
                await remoteDeploymentManager.StartDeployment(
                    repository,
                    compileJob,
                    cancellationToken);
 
                logger.LogTrace("Deployment will timeout at {timeoutTime}", DateTimeOffset.UtcNow + dreamMakerSettings.Timeout!.Value);
                using var timeoutTokenSource = new CancellationTokenSource(dreamMakerSettings.Timeout.Value);
                var timeoutToken = timeoutTokenSource.Token;
                using (timeoutToken.Register(() => logger.LogWarning("Deployment timed out!")))
                {
                    using var combinedTokenSource = CancellationTokenSource.CreateLinkedTokenSource(timeoutToken, cancellationToken);
                    try
                    {
                        await RunCompileJob(
                            progressReporter,
                            compileJob,
                            dreamMakerSettings,
                            launchParameters,
                            engineLock,
                            repository,
                            remoteDeploymentManager,
                            combinedTokenSource.Token);
                    }
                    catch (OperationCanceledException) when (timeoutToken.IsCancellationRequested)
                    {
                        throw new JobException(ErrorCode.DeploymentTimeout);
                    }
                }
 
                return compileJob;
            }
            catch (OperationCanceledException)
            {
                // DCT: Cancellation token is for job, delaying here is fine
                progressReporter.StageName = "Running CompileCancelled event";
                await eventConsumer.HandleEvent(EventType.CompileCancelled, Enumerable.Empty<string>(), true, CancellationToken.None);
                throw;
            }
            finally
            {
                progressCts.Cancel();
                await progressTask;
            }
        }
 
        async ValueTask RunCompileJob(
            JobProgressReporter progressReporter,
            Models.CompileJob job,
            Api.Models.Internal.DreamMakerSettings dreamMakerSettings,
            DreamDaemonLaunchParameters launchParameters,
            IEngineExecutableLock engineLock,
            IRepository repository,
            IRemoteDeploymentManager remoteDeploymentManager,
            CancellationToken cancellationToken)
        {
            var outputDirectory = job.DirectoryName!.Value.ToString();
            logger.LogTrace("Compile output GUID: {dirGuid}", outputDirectory);
 
            try
            {
                // copy the repository
                logger.LogTrace("Copying repository to game directory");
                progressReporter.StageName = "Copying repository";
                var resolvedOutputDirectory = ioManager.ResolvePath(outputDirectory);
                var repoOrigin = repository.Origin;
                var repoReference = repository.Reference;
                using (repository)
                    await repository.CopyTo(resolvedOutputDirectory, cancellationToken);
 
                // repository closed now
 
                // run precompile scripts
                progressReporter.StageName = "Running PreCompile event";
                await eventConsumer.HandleEvent(
                    EventType.CompileStart,
                    new List<string>
                    {
                        resolvedOutputDirectory,
                        repoOrigin.ToString(),
                        engineLock.Version.ToString(),
                        repoReference,
                    },
                    true,
                    cancellationToken);
 
                // determine the dme
                progressReporter.StageName = "Determining .dme";
                if (job.DmeName == null)
                {
                    logger.LogTrace("Searching for available .dmes");
                    var foundPaths = await ioManager.GetFilesWithExtension(resolvedOutputDirectory, DmeExtension, true, cancellationToken);
                    var foundPath = foundPaths.FirstOrDefault();
                    if (foundPath == default)
                        throw new JobException(ErrorCode.DeploymentNoDme);
                    job.DmeName = foundPath.Substring(
                        resolvedOutputDirectory.Length + 1,
                        foundPath.Length - resolvedOutputDirectory.Length - DmeExtension.Length - 2); // +1 for . in extension
                }
                else
                {
                    var targetDme = ioManager.ConcatPath(outputDirectory, String.Join('.', job.DmeName, DmeExtension));
                    if (!await ioManager.PathIsChildOf(outputDirectory, targetDme, cancellationToken))
                        throw new JobException(ErrorCode.DeploymentWrongDme);
 
                    var targetDmeExists = await ioManager.FileExists(targetDme, cancellationToken);
                    if (!targetDmeExists)
                        throw new JobException(ErrorCode.DeploymentMissingDme);
                }
 
                logger.LogDebug("Selected \"{dmeName}.dme\" for compilation!", job.DmeName);
 
                progressReporter.StageName = "Modifying .dme";
                await ModifyDme(job, cancellationToken);
 
                // run precompile scripts
                progressReporter.StageName = "Running PreDreamMaker event";
                await eventConsumer.HandleEvent(
                    EventType.PreDreamMaker,
                    new List<string>
                    {
                        resolvedOutputDirectory,
                        repoOrigin.ToString(),
                        engineLock.Version.ToString(),
                    },
                    true,
                    cancellationToken);
 
                // run compiler
                progressReporter.StageName = "Running Compiler";
                var compileSuceeded = await RunDreamMaker(engineLock, job, dreamMakerSettings.CompilerAdditionalArguments, cancellationToken);
 
                // Session takes ownership of the lock and Disposes it so save this for later
                var engineVersion = engineLock.Version;
 
                // verify api
                try
                {
                    if (!compileSuceeded)
                        throw new JobException(
                            ErrorCode.DeploymentExitCode,
                            new JobException($"Compilation failed:{Environment.NewLine}{Environment.NewLine}{job.Output}"));
 
                    await VerifyApi(
                        launchParameters.StartupTimeout!.Value,
                        dreamMakerSettings.ApiValidationSecurityLevel!.Value,
                        job,
                        progressReporter,
                        engineLock,
                        dreamMakerSettings.ApiValidationPort!.Value,
                        dreamMakerSettings.DMApiValidationMode!.Value,
                        launchParameters.LogOutput!.Value,
                        cancellationToken);
                }
                catch (JobException)
                {
                    // DD never validated or compile failed
                    progressReporter.StageName = "Running CompileFailure event";
                    await eventConsumer.HandleEvent(
                        EventType.CompileFailure,
                        new List<string>
                        {
                            resolvedOutputDirectory,
                            compileSuceeded ? "1" : "0",
                            engineVersion.ToString(),
                        },
                        true,
                        cancellationToken);
                    throw;
                }
 
                progressReporter.StageName = "Running CompileComplete event";
                await eventConsumer.HandleEvent(
                    EventType.CompileComplete,
                    new List<string>
                    {
                        resolvedOutputDirectory,
                        engineVersion.ToString(),
                    },
                    true,
                    cancellationToken);
 
                logger.LogTrace("Applying static game file symlinks...");
                progressReporter.StageName = "Symlinking GameStaticFiles";
 
                // symlink in the static data
                await configuration.SymlinkStaticFilesTo(resolvedOutputDirectory, cancellationToken);
 
                logger.LogDebug("Compile complete!");
            }
            catch (Exception ex)
            {
                progressReporter.StageName = "Cleaning output directory";
                await CleanupFailedCompile(job, remoteDeploymentManager, ex);
                throw;
            }
        }
 
        async ValueTask ProgressTask(JobProgressReporter progressReporter, TimeSpan? estimatedDuration, CancellationToken cancellationToken)
        {
            double? lastReport = estimatedDuration.HasValue ? 0 : null;
            progressReporter.ReportProgress(lastReport);
 
            var minimumSleepInterval = TimeSpan.FromMilliseconds(250);
            var sleepInterval = estimatedDuration.HasValue ? estimatedDuration.Value / 100 : minimumSleepInterval;
 
            if (estimatedDuration.HasValue)
            {
                logger.LogDebug("Compile is expected to take: {estimatedDuration}", estimatedDuration);
            }
            else
            {
                logger.LogTrace("No metric to estimate compile time.");
            }
 
            try
            {
                for (var iteration = 0; iteration < (estimatedDuration.HasValue ? 99 : Int32.MaxValue); ++iteration)
                {
                    if (estimatedDuration.HasValue)
                    {
                        var nextInterval = DateTimeOffset.UtcNow + sleepInterval;
                        do
                        {
                            var remainingSleepThisInterval = nextInterval - DateTimeOffset.UtcNow;
                            var nextSleepSpan = remainingSleepThisInterval < minimumSleepInterval ? minimumSleepInterval : remainingSleepThisInterval;
 
                            await asyncDelayer.Delay(nextSleepSpan, cancellationToken);
                            progressReporter.ReportProgress(lastReport);
                        }
                        while (DateTimeOffset.UtcNow < nextInterval);
                    }
                    else
                        await asyncDelayer.Delay(minimumSleepInterval, cancellationToken);
 
                    lastReport = estimatedDuration.HasValue ? sleepInterval * (iteration + 1) / estimatedDuration.Value : null;
                    progressReporter.ReportProgress(lastReport);
                }
            }
            catch (OperationCanceledException ex)
            {
                logger.LogTrace(ex, "ProgressTask aborted.");
            }
            catch (Exception ex)
            {
                logger.LogError(ex, "ProgressTask crashed!");
            }
        }
 
        async ValueTask VerifyApi(
            uint timeout,
            DreamDaemonSecurity securityLevel,
            Models.CompileJob job,
            JobProgressReporter progressReporter,
            IEngineExecutableLock engineLock,
            ushort portToUse,
            DMApiValidationMode validationMode,
            bool logOutput,
            CancellationToken cancellationToken)
        {
            if (validationMode == DMApiValidationMode.Skipped)
            {
                logger.LogDebug("Skipping DMAPI validation");
                job.MinimumSecurityLevel = DreamDaemonSecurity.Ultrasafe;
                return;
            }
 
            progressReporter.StageName = "Validating DMAPI";
 
            var requireValidate = validationMode == DMApiValidationMode.Required;
            logger.LogTrace("Verifying {possiblyRequired}DMAPI...", requireValidate ? "required " : String.Empty);
            var launchParameters = new DreamDaemonLaunchParameters
            {
                AllowWebClient = false,
                Port = portToUse,
                OpenDreamTopicPort = 0,
                SecurityLevel = securityLevel,
                Visibility = DreamDaemonVisibility.Invisible,
                StartupTimeout = timeout,
                TopicRequestTimeout = 0, // not used
                HealthCheckSeconds = 0, // not used
                StartProfiler = false,
                LogOutput = logOutput,
                MapThreads = 1, // lowest possible amount
            };
 
            job.MinimumSecurityLevel = securityLevel; // needed for the TempDmbProvider
 
            ApiValidationStatus validationStatus;
            await using (var provider = new TemporaryDmbProvider(
                ioManager.ResolvePath(job.DirectoryName!.Value.ToString()),
                job,
                engineLock.Version))
            await using (var controller = await sessionControllerFactory.LaunchNew(provider, engineLock, launchParameters, true, cancellationToken))
            {
                var launchResult = await controller.LaunchResult.WaitAsync(cancellationToken);
 
                if (launchResult.StartupTime.HasValue)
                    await controller.Lifetime.WaitAsync(cancellationToken);
 
                if (!controller.Lifetime.IsCompleted)
                    await controller.DisposeAsync();
 
                validationStatus = controller.ApiValidationStatus;
 
                logger.LogTrace("API validation status: {validationStatus}", validationStatus);
 
                job.DMApiVersion = controller.DMApiVersion;
            }
 
            switch (validationStatus)
            {
                case ApiValidationStatus.RequiresUltrasafe:
                    job.MinimumSecurityLevel = DreamDaemonSecurity.Ultrasafe;
                    return;
                case ApiValidationStatus.RequiresSafe:
                    job.MinimumSecurityLevel = DreamDaemonSecurity.Safe;
                    return;
                case ApiValidationStatus.RequiresTrusted:
                    job.MinimumSecurityLevel = DreamDaemonSecurity.Trusted;
                    return;
                case ApiValidationStatus.NeverValidated:
                    if (requireValidate)
                        throw new JobException(ErrorCode.DeploymentNeverValidated);
                    job.MinimumSecurityLevel = DreamDaemonSecurity.Ultrasafe;
                    break;
                case ApiValidationStatus.BadValidationRequest:
                case ApiValidationStatus.Incompatible:
                    throw new JobException(ErrorCode.DeploymentInvalidValidation);
                case ApiValidationStatus.UnaskedValidationRequest:
                default:
                    throw new InvalidOperationException(
                        $"Session controller returned unexpected ApiValidationStatus: {validationStatus}");
            }
        }
 
        async ValueTask<bool> RunDreamMaker(
            IEngineExecutableLock engineLock,
            Models.CompileJob job,
            string? additionalCompilerArguments,
            CancellationToken cancellationToken)
        {
            var environment = await engineLock.LoadEnv(logger, true, cancellationToken);
            var arguments = engineLock.FormatCompilerArguments($"{job.DmeName}.{DmeExtension}", additionalCompilerArguments);
 
            await using var dm = await processExecutor.LaunchProcess(
                engineLock.CompilerExePath,
                ioManager.ResolvePath(
                    job.DirectoryName!.Value.ToString()),
                arguments,
                cancellationToken,
                environment,
                readStandardHandles: true,
                noShellExecute: true);
 
            if (sessionConfiguration.LowPriorityDeploymentProcesses)
                dm.AdjustPriority(false);
 
            int exitCode;
            using (cancellationToken.Register(() => dm.Terminate()))
                exitCode = (await dm.Lifetime).Value;
            cancellationToken.ThrowIfCancellationRequested();
 
            logger.LogDebug("DreamMaker exit code: {exitCode}", exitCode);
            job.Output = $"{await dm.GetCombinedOutput(cancellationToken)}{Environment.NewLine}{Environment.NewLine}Exit Code: {exitCode}";
            logger.LogDebug("DreamMaker output: {newLine}{output}", Environment.NewLine, job.Output);
 
            currentDreamMakerOutput = job.Output;
            return exitCode == 0;
        }
 
        async ValueTask ModifyDme(Models.CompileJob job, CancellationToken cancellationToken)
        {
            var dmeFileName = String.Join('.', job.DmeName, DmeExtension);
            var stringDirectoryName = job.DirectoryName!.Value.ToString();
            var dmePath = ioManager.ConcatPath(stringDirectoryName, dmeFileName);
            var dmeReadTask = ioManager.ReadAllBytes(dmePath, cancellationToken);
 
            var dmeModificationsTask = configuration.CopyDMFilesTo(
                dmeFileName,
                ioManager.ResolvePath(
                    ioManager.ConcatPath(
                        stringDirectoryName,
                        ioManager.GetDirectoryName(dmeFileName))),
                cancellationToken);
 
            var dmeBytes = await dmeReadTask;
            var dme = Encoding.UTF8.GetString(dmeBytes);
 
            var dmeModifications = await dmeModificationsTask;
 
            if (dmeModifications == null || dmeModifications.TotalDmeOverwrite)
            {
                if (dmeModifications != null)
                    logger.LogDebug(".dme replacement configured!");
                else
                    logger.LogTrace("No .dme modifications required.");
                return;
            }
 
            var dmeLines = new List<string>(dme.Split('\n', StringSplitOptions.None));
            for (var dmeLineIndex = 0; dmeLineIndex < dmeLines.Count; ++dmeLineIndex)
            {
                var line = dmeLines[dmeLineIndex];
                if (line.Contains("BEGIN_INCLUDE", StringComparison.Ordinal) && dmeModifications.HeadIncludeLine != null)
                {
                    var headIncludeLineNumber = dmeLineIndex + 1;
                    logger.LogDebug(
                        "Inserting HeadInclude.dm at line {lineNumber}: {includeLine}",
                        headIncludeLineNumber,
                        dmeModifications.HeadIncludeLine);
                    dmeLines.Insert(headIncludeLineNumber, dmeModifications.HeadIncludeLine);
                    ++dmeLineIndex;
                }
                else if (line.Contains("END_INCLUDE", StringComparison.Ordinal) && dmeModifications.TailIncludeLine != null)
                {
                    logger.LogDebug(
                        "Inserting TailInclude.dm at line {lineNumber}: {includeLine}",
                        dmeLineIndex,
                        dmeModifications.TailIncludeLine);
                    dmeLines.Insert(dmeLineIndex, dmeModifications.TailIncludeLine);
                    break;
                }
            }
 
            dmeBytes = Encoding.UTF8.GetBytes(String.Join('\n', dmeLines));
            await ioManager.WriteAllBytes(dmePath, dmeBytes, cancellationToken);
        }
 
        ValueTask CleanupFailedCompile(Models.CompileJob job, IRemoteDeploymentManager remoteDeploymentManager, Exception exception)
        {
            async ValueTask CleanDir()
            {
                if (sessionConfiguration.DelayCleaningFailedDeployments)
                {
                    logger.LogDebug("Not cleaning up errored deployment directory {guid} due to config.", job.DirectoryName);
                    return;
                }
 
                logger.LogTrace("Cleaning compile directory...");
                var jobPath = job.DirectoryName!.Value.ToString();
                try
                {
                    // DCT: None available
                    await eventConsumer.HandleEvent(EventType.DeploymentCleanup, new List<string> { jobPath }, true, CancellationToken.None);
                    await ioManager.DeleteDirectory(jobPath, CancellationToken.None);
                }
                catch (Exception e)
                {
                    logger.LogWarning(e, "Error cleaning up compile directory {path}!", ioManager.ResolvePath(jobPath));
                }
            }
 
            var dirCleanTask = CleanDir();
 
            var failRemoteDeployTask = remoteDeploymentManager.FailDeployment(
                job,
                FormatExceptionForUsers(exception),
                CancellationToken.None); // DCT: None available
 
            return ValueTaskExtensions.WhenAll(
                dirCleanTask,
                failRemoteDeployTask);
        }
    }
}