mirror of
https://github.com/wabbajack-tools/wabbajack.git
synced 2024-08-30 18:42:17 +00:00
61 lines
1.9 KiB
C#
61 lines
1.9 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using Alphaleonis.Win32.Filesystem;
|
|
using Newtonsoft.Json;
|
|
|
|
namespace Wabbajack.Lib.CompilationSteps
|
|
{
|
|
public class IncludeThisProfile : ACompilationStep
|
|
{
|
|
private readonly IEnumerable<string> _correctProfiles;
|
|
private readonly MO2Compiler _mo2Compiler;
|
|
|
|
public IncludeThisProfile(ACompiler compiler) : base(compiler)
|
|
{
|
|
_mo2Compiler = (MO2Compiler) compiler;
|
|
_correctProfiles = _mo2Compiler.SelectedProfiles.Select(p => Path.Combine("profiles", p) + "\\").ToList();
|
|
}
|
|
|
|
public override async ValueTask<Directive> Run(RawSourceFile source)
|
|
{
|
|
if (_correctProfiles.Any(p => source.Path.StartsWith(p)))
|
|
{
|
|
var data = source.Path.EndsWith("\\modlist.txt")
|
|
? ReadAndCleanModlist(source.AbsolutePath)
|
|
: File.ReadAllBytes(source.AbsolutePath);
|
|
|
|
var e = source.EvolveTo<InlineFile>();
|
|
e.SourceDataID = _compiler.IncludeFile(data);
|
|
return e;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
public override IState GetState()
|
|
{
|
|
return new State();
|
|
}
|
|
|
|
private static byte[] ReadAndCleanModlist(string absolutePath)
|
|
{
|
|
var lines = File.ReadAllLines(absolutePath);
|
|
lines = (from line in lines
|
|
where !(line.StartsWith("-") && !line.EndsWith("_separator"))
|
|
select line).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);
|
|
}
|
|
}
|
|
}
|
|
}
|