diff --git a/Wabbajack.Lib/CompilationSteps/IncludeThisProfile.cs b/Wabbajack.Lib/CompilationSteps/IncludeThisProfile.cs index 4d0141a3..665be40d 100644 --- a/Wabbajack.Lib/CompilationSteps/IncludeThisProfile.cs +++ b/Wabbajack.Lib/CompilationSteps/IncludeThisProfile.cs @@ -1,58 +1,65 @@ -using System.Collections.Generic; -using System.Linq; -using System.Text; -using System.Threading.Tasks; -using Alphaleonis.Win32.Filesystem; -using Newtonsoft.Json; -using Wabbajack.Common; - -namespace Wabbajack.Lib.CompilationSteps -{ - public class IncludeThisProfile : ACompilationStep - { - private readonly IEnumerable _correctProfiles; - private MO2Compiler _mo2Compiler; - - public IncludeThisProfile(ACompiler compiler) : base(compiler) - { - _mo2Compiler = (MO2Compiler) compiler; - _correctProfiles = _mo2Compiler.SelectedProfiles.Select(p => _mo2Compiler.MO2ProfileDir.Parent.Combine(p)).ToList(); - } - - public override async ValueTask Run(RawSourceFile source) - { - if (!_correctProfiles.Any(p => source.AbsolutePath.InFolder(p))) - return null; - - var data = source.Path.FileName == Consts.ModListTxt - ? await ReadAndCleanModlist(source.AbsolutePath) - : await source.AbsolutePath.ReadAllBytesAsync(); - - var e = source.EvolveTo(); - e.SourceDataID = await _compiler.IncludeFile(data); - return e; - - } - - public override IState GetState() - { - return new State(); - } - - private static async Task ReadAndCleanModlist(AbsolutePath absolutePath) - { - var lines = await absolutePath.ReadAllLinesAsync(); - lines = lines.Where(line => !(line.StartsWith("-") && !line.EndsWith("_separator"))).ToArray(); - return Encoding.UTF8.GetBytes(string.Join("\r\n", lines)); - } - - [JsonObject("IncludeThisProfile")] - public class State : IState - { - public ICompilationStep CreateStep(ACompiler compiler) - { - return new IncludeThisProfile(compiler); - } - } - } -} +using System.Collections.Generic; +using System.Linq; +using System.Text; +using System.Threading.Tasks; +using Alphaleonis.Win32.Filesystem; +using Newtonsoft.Json; +using Wabbajack.Common; + +namespace Wabbajack.Lib.CompilationSteps +{ + public class IncludeThisProfile : ACompilationStep + { + private readonly IEnumerable _correctProfiles; + private MO2Compiler _mo2Compiler; + + public IncludeThisProfile(ACompiler compiler) : base(compiler) + { + _mo2Compiler = (MO2Compiler) compiler; + _correctProfiles = _mo2Compiler.SelectedProfiles.Select(p => _mo2Compiler.MO2ProfileDir.Parent.Combine(p)).ToList(); + } + + public override async ValueTask Run(RawSourceFile source) + { + if (!_correctProfiles.Any(p => source.AbsolutePath.InFolder(p))) + return null; + + var data = source.Path.FileName == Consts.ModListTxt + ? await ReadAndCleanModlist(source.AbsolutePath) + : await source.AbsolutePath.ReadAllBytesAsync(); + + var e = source.EvolveTo(); + e.SourceDataID = await _compiler.IncludeFile(data); + return e; + + } + + public override IState GetState() + { + return new State(); + } + + private async Task ReadAndCleanModlist(AbsolutePath absolutePath) + { + var alwaysEnabled = _mo2Compiler.ModInis.Where(f => IgnoreDisabledMods.IsAlwaysEnabled(f.Value)) + .Select(f => f.Key) + .Distinct(); + var lines = (await absolutePath.ReadAllLinesAsync()).Where(l => + { + return l.StartsWith("+") + || alwaysEnabled.Any(x => x.Equals(l.Substring(1))) + || l.EndsWith("_separator"); + }).ToArray(); + return Encoding.UTF8.GetBytes(string.Join("\r\n", lines)); + } + + [JsonObject("IncludeThisProfile")] + public class State : IState + { + public ICompilationStep CreateStep(ACompiler compiler) + { + return new IncludeThisProfile(compiler); + } + } + } +}