wabbajack/Wabbajack.Test/ACompilerTest.cs

using System;
using System.Threading.Tasks;
using Wabbajack.Common;
using Wabbajack.Lib;
using Wabbajack.Lib.LibCefHelpers;
using Xunit;
using Xunit.Abstractions;

namespace Wabbajack.Test
{
    public abstract class ACompilerTest : XunitContextBase, IAsyncDisposable
    {
        private IDisposable _unsub;
        protected TestUtils utils { get; set; }

        public ACompilerTest(ITestOutputHelper helper) : base (helper)
        {
            Helpers.Init();
            Consts.TestMode = true;

            utils = new TestUtils();
            utils.Game = Game.SkyrimSpecialEdition;

            DateTime startTime = DateTime.Now;
            _unsub = Utils.LogMessages.Subscribe(f => XunitContext.WriteLine($"{DateTime.Now - startTime} -  {f.ShortDescription}"));

        }

        public async ValueTask DisposeAsync()
        {
            await utils.DisposeAsync();
            _unsub.Dispose();
            base.Dispose();
        }

        protected async Task<MO2Compiler> ConfigureAndRunCompiler(string profile, bool useGameFiles= false)
        {
            var compiler = new MO2Compiler(
                sourcePath: utils.SourcePath,
                downloadsPath: utils.DownloadsPath,
                mo2Profile: profile,
                outputFile: OutputFile(profile));
            compiler.UseGamePaths = useGameFiles;
            Assert.True(await compiler.Begin());
            return compiler;
        }

        protected async Task<ModList> CompileAndInstall(string profile, bool useGameFiles = false)
        {
            var compiler = await ConfigureAndRunCompiler(profile, useGameFiles: useGameFiles);
            Utils.Log("Finished Compiling");
            await Install(compiler);
            return compiler.ModList;
        }
        
        protected async Task<NativeCompiler> ConfigureAndRunCompiler(AbsolutePath configPath, bool useGameFiles= false)
        {
            var settings = configPath.FromJson<NativeCompilerSettings>();
            var profile = utils.AddProfile();

            var compiler = new NativeCompiler(
                settings: settings,
                sourcePath: utils.SourcePath,
                downloadsPath: utils.DownloadsPath,
                outputModListPath: OutputFile(profile)) 
                {UseGamePaths = useGameFiles};
            Assert.True(await compiler.Begin());
            return compiler;
        }
        protected async Task<ModList> CompileAndInstall(AbsolutePath settingsPath, bool useGameFiles = false)
        {
            var compiler = await ConfigureAndRunCompiler(settingsPath, useGameFiles: useGameFiles);
            Utils.Log("Finished Compiling");
            await Install(compiler);
            return compiler.ModList;
        }

        private static AbsolutePath OutputFile(string profile)
        {
            return ((RelativePath)profile).RelativeToEntryPoint().WithExtension(Consts.ModListExtension);
        }

        protected async Task Install(ACompiler compiler)
        {
            Utils.Log("Loading Modlist");
            var modlist = AInstaller.LoadFromFile(compiler.ModListOutputFile);
            Utils.Log("Constructing Installer");
            var installer = new MO2Installer(
                archive: compiler.ModListOutputFile,
                modList: modlist,
                outputFolder: utils.InstallPath,
                downloadFolder: utils.DownloadsPath,
                parameters: CreateDummySystemParameters());
            installer.WarnOnOverwrite = false;
            installer.GameFolder = utils.GameFolder;
            Utils.Log("Starting Install");
            await installer.Begin();
        }

        public static SystemParameters CreateDummySystemParameters()
        {
            return new SystemParameters
            {
                WindowsVersion = new Version("6.2.4.0"),
                ScreenWidth = 1920,
                ScreenHeight = 1080,
                SystemMemorySize = 16 * 1024 * 1040,
                VideoMemorySize = 4 * 1024 * 1024
            };
        }
    }
}