2020-05-13 01:48:57 +00:00
|
|
|
|
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<AbsolutePath> _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<Directive?> 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<InlineFile>();
|
|
|
|
|
e.SourceDataID = await _compiler.IncludeFile(data);
|
|
|
|
|
return e;
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private async Task<byte[]> ReadAndCleanModlist(AbsolutePath absolutePath)
|
2020-05-11 22:14:45 +00:00
|
|
|
|
{
|
|
|
|
|
var alwaysEnabled = _mo2Compiler.ModInis.Where(f => IgnoreDisabledMods.IsAlwaysEnabled(f.Value))
|
|
|
|
|
.Select(f => f.Key)
|
2020-05-23 21:41:41 +00:00
|
|
|
|
.Select(f => f.FileName.ToString())
|
2020-05-11 22:14:45 +00:00
|
|
|
|
.Distinct();
|
2020-05-13 01:48:57 +00:00
|
|
|
|
var lines = (await absolutePath.ReadAllLinesAsync()).Where(l =>
|
2020-05-11 22:14:45 +00:00
|
|
|
|
{
|
|
|
|
|
return l.StartsWith("+")
|
|
|
|
|
|| alwaysEnabled.Any(x => x.Equals(l.Substring(1)))
|
|
|
|
|
|| l.EndsWith("_separator");
|
|
|
|
|
}).ToArray();
|
2020-05-13 01:48:57 +00:00
|
|
|
|
return Encoding.UTF8.GetBytes(string.Join("\r\n", lines));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|