wabbajack/Wabbajack.Lib/CompilationSteps/IncludeStubbedConfigfiles.cs

using System;
using System.Text;
using System.Threading.Tasks;
using Alphaleonis.Win32.Filesystem;
using Newtonsoft.Json;
using Wabbajack.Common;

namespace Wabbajack.Lib.CompilationSteps
{
    public class IncludeStubbedConfigFiles : ACompilationStep
    {
        private readonly MO2Compiler _mo2Compiler;

        public IncludeStubbedConfigFiles(ACompiler compiler) : base(compiler)
        {
            _mo2Compiler = (MO2Compiler) compiler;
        }

        public override async ValueTask<Directive?> Run(RawSourceFile source)
        {
            return Consts.ConfigFileExtensions.Contains(source.Path.Extension) ? await RemapFile(source) : null;
        }
        private async Task<Directive?> RemapFile(RawSourceFile source)
        {
            var data = await source.AbsolutePath.ReadAllTextAsync();
            var originalData = data;

            data = RemapData(_mo2Compiler, data);

            if (data == originalData)
                return null;
            var result = source.EvolveTo<RemappedInlineFile>();
            result.SourceDataID = await _compiler.IncludeFile(Encoding.UTF8.GetBytes(data));
            return result;
        }

        public static string RemapData(ACompiler compiler, string data)
        {
            var gamePath = compiler.GamePath.Normalize();
            data = data.Replace(gamePath, Consts.GAME_PATH_MAGIC_BACK, StringComparison.InvariantCultureIgnoreCase);
            data = data.Replace(gamePath.Replace("\\", "\\\\"), Consts.GAME_PATH_MAGIC_DOUBLE_BACK, StringComparison.InvariantCultureIgnoreCase);
            data = data.Replace(gamePath.Replace("\\", "/"), Consts.GAME_PATH_MAGIC_FORWARD, StringComparison.InvariantCultureIgnoreCase);

            var sourcePath = compiler.SourcePath.Normalize();
            data = data.Replace(sourcePath, Consts.MO2_PATH_MAGIC_BACK, StringComparison.InvariantCultureIgnoreCase);
            data = data.Replace(sourcePath.Replace("\\", "\\\\"), Consts.MO2_PATH_MAGIC_DOUBLE_BACK, StringComparison.InvariantCultureIgnoreCase);
            data = data.Replace(sourcePath.Replace("\\", "/"), Consts.MO2_PATH_MAGIC_FORWARD, StringComparison.InvariantCultureIgnoreCase);

            var downloadsPath = compiler.DownloadsPath.Normalize();
            data = data.Replace(downloadsPath, Consts.DOWNLOAD_PATH_MAGIC_BACK, StringComparison.InvariantCultureIgnoreCase);
            data = data.Replace(downloadsPath.Replace("\\", "\\\\"), Consts.DOWNLOAD_PATH_MAGIC_DOUBLE_BACK, StringComparison.InvariantCultureIgnoreCase);
            data = data.Replace(downloadsPath.Replace("\\", "/"), Consts.DOWNLOAD_PATH_MAGIC_FORWARD, StringComparison.InvariantCultureIgnoreCase);
            return data;
        }
    }
}