Created IncludeGenericGamePlugin

This commit is contained in:
erri120 2020-06-11 18:18:21 +02:00
parent 3a2a5918b6
commit 4ed7ab1aa1
No known key found for this signature in database
GPG Key ID: A8C0A18D8D4D3135
2 changed files with 89 additions and 25 deletions

View File

@ -0,0 +1,60 @@
using System;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using Newtonsoft.Json;
using Wabbajack.Common;
namespace Wabbajack.Lib.CompilationSteps
{
public class IncludeGenericGamePlugin : ACompilationStep
{
private readonly bool _validGame;
private readonly string _pluginsFolder = string.Empty;
private readonly string _gameName = string.Empty;
public IncludeGenericGamePlugin(ACompiler compiler) : base(compiler)
{
if (!(compiler is MO2Compiler mo2Compiler))
return;
if (mo2Compiler.CompilingGame.NexusName == null)
return;
_validGame = mo2Compiler.CompilingGame.IsGenericMO2Plugin;
_pluginsFolder = mo2Compiler.MO2Folder.Combine("plugins").ToString();
_gameName = $"game_{mo2Compiler.CompilingGame.NexusName}.py";
}
private static Regex regex = new Regex(@"^game_$");
public override async ValueTask<Directive?> Run(RawSourceFile source)
{
if (!_validGame)
return null;
if (!source.AbsolutePath.ToString().StartsWith(_pluginsFolder))
return null;
if(!source.AbsolutePath.FileName.ToString().Equals(_gameName, StringComparison.InvariantCultureIgnoreCase))
return null;
var res = source.EvolveTo<InlineFile>();
res.SourceDataID = await _compiler.IncludeFile(await source.AbsolutePath.ReadAllBytesAsync());
return res;
}
public override IState GetState()
{
return new State();
}
[JsonObject("IncludeGenericGamePlugin")]
public class State : IState
{
public ICompilationStep CreateStep(ACompiler compiler)
{
return new IncludeGenericGamePlugin(compiler);
}
}
}
}

View File

@ -175,7 +175,6 @@ namespace Wabbajack.Lib
UpdateTracker.NextStep("Reindexing downloads after meta inferring");
await VFS.AddRoot(MO2DownloadsFolder);
await VFS.WriteToFile(VFSCacheName);
if (cancel.IsCancellationRequested) return false;
UpdateTracker.NextStep("Pre-validating Archives");
@ -193,34 +192,38 @@ namespace Wabbajack.Lib
var stockGameFolder = CompilingGame.GameLocation();
foreach (var (relativePath, hash) in await ClientAPI.GetGameFiles(CompilingGame.Game, Version.Parse(CompilingGame.InstalledVersion)))
var installedVersion = CompilingGame.InstalledVersion;
if (installedVersion != null)
{
if (!VFS.Index.ByRootPath.TryGetValue(relativePath.RelativeTo(stockGameFolder), out var virtualFile))
continue;
if (virtualFile.Hash != hash)
foreach (var (relativePath, hash) in await ClientAPI.GetGameFiles(CompilingGame.Game, Version.Parse(installedVersion)))
{
Utils.Log(
$"File {relativePath} int the game folder appears to be modified, it will not be used during compilation");
continue;
if (!VFS.Index.ByRootPath.TryGetValue(relativePath.RelativeTo(stockGameFolder), out var virtualFile))
continue;
if (virtualFile.Hash != hash)
{
Utils.Log(
$"File {relativePath} int the game folder appears to be modified, it will not be used during compilation");
continue;
}
var state = new GameFileSourceDownloader.State
{
Game = CompilingGame.Game,
GameVersion = CompilingGame.InstalledVersion,
GameFile = relativePath
};
Utils.Log($"Adding Game file: {relativePath}");
IndexedArchives.Add(new IndexedArchive(virtualFile)
{
Name = (string)relativePath.FileName,
IniData = state.GetMetaIniString().LoadIniString(),
Meta = state.GetMetaIniString()
});
}
var state = new GameFileSourceDownloader.State
{
Game = CompilingGame.Game, GameVersion = CompilingGame.InstalledVersion, GameFile = relativePath
};
Utils.Log($"Adding Game file: {relativePath}");
IndexedArchives.Add(new IndexedArchive(virtualFile)
{
Name = (string)relativePath.FileName,
IniData = state.GetMetaIniString().LoadIniString(),
Meta = state.GetMetaIniString()
});
}
await CleanInvalidArchives();
UpdateTracker.NextStep("Finding Install Files");
@ -550,6 +553,7 @@ namespace Wabbajack.Lib
{
new IgnoreGameFilesIfGameFolderFilesExist(this),
new IncludePropertyFiles(this),
new IncludeGenericGamePlugin(this),
new IgnoreStartsWith(this,"logs\\"),
new IgnoreStartsWith(this, "downloads\\"),
new IgnoreStartsWith(this,"webcache\\"),