mirror of
https://github.com/wabbajack-tools/wabbajack.git
synced 2024-08-30 18:42:17 +00:00
1025 lines
38 KiB
C#
1025 lines
38 KiB
C#
using Compression.BSA;
|
|
using Newtonsoft.Json;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Reflection;
|
|
using System.Security.Cryptography;
|
|
using System.Text;
|
|
using System.Text.RegularExpressions;
|
|
using System.Web;
|
|
using Wabbajack.Common;
|
|
using static Wabbajack.NexusAPI;
|
|
using VFS;
|
|
|
|
namespace Wabbajack
|
|
{
|
|
public class Compiler
|
|
{
|
|
|
|
|
|
|
|
public string MO2Folder;
|
|
|
|
public dynamic MO2Ini { get; }
|
|
public string GamePath { get; }
|
|
|
|
public bool IgnoreMissingFiles { get; set; }
|
|
|
|
public string MO2DownloadsFolder
|
|
{
|
|
get
|
|
{
|
|
return Path.Combine(MO2Folder, "downloads");
|
|
}
|
|
}
|
|
|
|
|
|
|
|
public string MO2Profile;
|
|
|
|
public string MO2ProfileDir
|
|
{
|
|
get
|
|
{
|
|
return Path.Combine(MO2Folder, "profiles", MO2Profile);
|
|
}
|
|
}
|
|
|
|
public Action<string> Log_Fn { get; }
|
|
public List<Directive> InstallDirectives { get; private set; }
|
|
public string NexusKey { get; private set; }
|
|
internal UserStatus User { get; private set; }
|
|
public List<Archive> SelectedArchives { get; private set; }
|
|
public List<RawSourceFile> AllFiles { get; private set; }
|
|
public ModList ModList { get; private set; }
|
|
public ConcurrentBag<Directive> ExtraFiles { get; private set; }
|
|
public Dictionary<string, dynamic> ModInis { get; private set; }
|
|
|
|
public VirtualFileSystem VFS
|
|
{
|
|
get
|
|
{
|
|
return VirtualFileSystem.VFS;
|
|
}
|
|
|
|
}
|
|
|
|
public List<IndexedArchive> IndexedArchives { get; private set; }
|
|
public Dictionary<string, IEnumerable<VirtualFile>> IndexedFiles { get; private set; }
|
|
|
|
public class IndexedFileMatch
|
|
{
|
|
public IndexedArchive Archive;
|
|
public IndexedArchiveEntry Entry;
|
|
public DateTime LastModified;
|
|
}
|
|
|
|
public Dictionary<string, IEnumerable<IndexedFileMatch>> DirectMatchIndex;
|
|
|
|
public void Info(string msg, params object[] args)
|
|
{
|
|
if (args.Length > 0)
|
|
msg = String.Format(msg, args);
|
|
Log_Fn(msg);
|
|
}
|
|
|
|
public void Status(string msg, params object[] args)
|
|
{
|
|
if (args.Length > 0)
|
|
msg = String.Format(msg, args);
|
|
WorkQueue.Report(msg, 0);
|
|
}
|
|
|
|
|
|
private void Error(string msg, params object[] args)
|
|
{
|
|
if (args.Length > 0)
|
|
msg = String.Format(msg, args);
|
|
Log_Fn(msg);
|
|
throw new Exception(msg);
|
|
}
|
|
|
|
public Compiler(string mo2_folder, Action<string> log_fn)
|
|
{
|
|
MO2Folder = mo2_folder;
|
|
Log_Fn = log_fn;
|
|
MO2Ini = Path.Combine(MO2Folder, "ModOrganizer.ini").LoadIniFile();
|
|
GamePath = ((string)MO2Ini.General.gamePath).Replace("\\\\", "\\");
|
|
}
|
|
|
|
private IndexedArchive LoadArchive(string file)
|
|
{
|
|
var info = new IndexedArchive();
|
|
info.File = VFS.Lookup(file);
|
|
|
|
info.Name = Path.GetFileName(file);
|
|
|
|
var ini_name = file + ".meta";
|
|
if (ini_name.FileExists())
|
|
{
|
|
info.IniData = ini_name.LoadIniFile();
|
|
info.Meta = File.ReadAllText(ini_name);
|
|
}
|
|
|
|
return info;
|
|
}
|
|
|
|
private bool IsArchiveFile(string name)
|
|
{
|
|
var ext = Path.GetExtension(name);
|
|
if (ext == ".bsa" || Consts.SupportedArchives.Contains(ext))
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
public void Compile()
|
|
{
|
|
Info($"Indexing {MO2Folder}");
|
|
VFS.AddRoot(MO2Folder);
|
|
Info($"Indexing {GamePath}");
|
|
VFS.AddRoot(GamePath);
|
|
|
|
var mo2_files = Directory.EnumerateFiles(MO2Folder, "*", SearchOption.AllDirectories)
|
|
.Where(p => p.FileExists())
|
|
.Select(p => new RawSourceFile(VFS.Lookup(p)) { Path = p.RelativeTo(MO2Folder)});
|
|
|
|
var game_files = Directory.EnumerateFiles(GamePath, "*", SearchOption.AllDirectories)
|
|
.Where(p => p.FileExists())
|
|
.Select(p => new RawSourceFile(VFS.Lookup(p)) { Path = Path.Combine(Consts.GameFolderFilesDir, p.RelativeTo(GamePath))});
|
|
|
|
var loot_path = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData), "LOOT");
|
|
|
|
Info($"Indexing {loot_path}");
|
|
VFS.AddRoot(loot_path);
|
|
|
|
var loot_files = Directory.EnumerateFiles(loot_path, "userlist.yaml", SearchOption.AllDirectories)
|
|
.Where(p => p.FileExists())
|
|
.Select(p => new RawSourceFile(VFS.Lookup(p)) { Path = Path.Combine(Consts.LOOTFolderFilesDir, p.RelativeTo(loot_path)) });
|
|
|
|
|
|
Info($"Indexing Archives");
|
|
IndexedArchives = Directory.EnumerateFiles(MO2DownloadsFolder)
|
|
.Where(f => Consts.SupportedArchives.Contains(Path.GetExtension(f)))
|
|
.Where(f => File.Exists(f + ".meta"))
|
|
.Select(f => new IndexedArchive()
|
|
{
|
|
File = VFS.Lookup(f),
|
|
Name = Path.GetFileName(f),
|
|
IniData = (f + ".meta").LoadIniFile(),
|
|
Meta = File.ReadAllText(f + ".meta")
|
|
})
|
|
.ToList();
|
|
|
|
Info($"Indexing Files");
|
|
IndexedFiles = IndexedArchives.PMap(f => { Status($"Finding files in {Path.GetFileName(f.File.FullPath)}");
|
|
return VFS.FilesInArchive(f.File); })
|
|
.SelectMany(fs => fs)
|
|
.OrderByDescending(f => f.TopLevelArchive.LastModified)
|
|
.GroupBy(f => f.Hash)
|
|
.ToDictionary(f => f.Key, f => f.AsEnumerable());
|
|
|
|
Info("Searching for mod files");
|
|
|
|
AllFiles = mo2_files.Concat(game_files)
|
|
.Concat(loot_files)
|
|
.ToList();
|
|
|
|
Info("Found {0} files to build into mod list", AllFiles.Count);
|
|
|
|
ExtraFiles = new ConcurrentBag<Directive>();
|
|
|
|
ModInis = Directory.EnumerateDirectories(Path.Combine(MO2Folder, "mods"))
|
|
.Select(f =>
|
|
{
|
|
var mod_name = Path.GetFileName(f);
|
|
var meta_path = Path.Combine(f, "meta.ini");
|
|
if (File.Exists(meta_path))
|
|
return (mod_name, meta_path.LoadIniFile());
|
|
return (null, null);
|
|
})
|
|
.Where(f => f.Item2 != null)
|
|
.ToDictionary(f => f.Item1, f => f.Item2);
|
|
|
|
var stack = MakeStack();
|
|
|
|
|
|
Info("Running Compilation Stack");
|
|
var results = AllFiles.PMap(f => RunStack(stack, f)).ToList();
|
|
|
|
// Add the extra files that were generated by the stack
|
|
Info($"Adding {ExtraFiles.Count} that were generated by the stack");
|
|
results = results.Concat(ExtraFiles).ToList();
|
|
|
|
var nomatch = results.OfType<NoMatch>();
|
|
Info("No match for {0} files", nomatch.Count());
|
|
foreach (var file in nomatch)
|
|
Info(" {0}", file.To);
|
|
if (nomatch.Count() > 0)
|
|
{
|
|
if (IgnoreMissingFiles)
|
|
{
|
|
Info("Continuing even though files were missing at the request of the user.");
|
|
}
|
|
else {
|
|
Info("Exiting due to no way to compile these files");
|
|
return;
|
|
}
|
|
}
|
|
|
|
InstallDirectives = results.Where(i => !(i is IgnoredDirectly)).ToList();
|
|
|
|
Info("Getting nexus api_key please click authorize if a browser window appears");
|
|
|
|
NexusKey = NexusAPI.GetNexusAPIKey();
|
|
User = NexusAPI.GetUserStatus(NexusKey);
|
|
|
|
if (!User.is_premium)
|
|
{
|
|
Info($"User {User.name} is not a premium Nexus user, cannot continue");
|
|
}
|
|
|
|
|
|
GatherArchives();
|
|
BuildPatches();
|
|
|
|
ModList = new ModList()
|
|
{
|
|
Archives = SelectedArchives,
|
|
Directives = InstallDirectives,
|
|
Name = MO2Profile
|
|
};
|
|
|
|
PatchExecutable();
|
|
|
|
ResetMembers();
|
|
|
|
Info("Done Building Modpack");
|
|
}
|
|
|
|
/// <summary>
|
|
/// Clear references to lists that hold a lot of data.
|
|
/// </summary>
|
|
private void ResetMembers()
|
|
{
|
|
AllFiles = null;
|
|
InstallDirectives = null;
|
|
SelectedArchives = null;
|
|
ExtraFiles = null;
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// Fills in the Patch fields in files that require them
|
|
/// </summary>
|
|
private void BuildPatches()
|
|
{
|
|
var groups = InstallDirectives.OfType<PatchedFromArchive>()
|
|
.GroupBy(p => p.ArchiveHashPath[0])
|
|
.ToList();
|
|
|
|
Info("Patching building patches from {0} archives", groups.Count);
|
|
var absolute_paths = AllFiles.ToDictionary(e => e.Path, e => e.AbsolutePath);
|
|
groups.PMap(group => BuildArchivePatches(group.Key, group, absolute_paths));
|
|
|
|
if (InstallDirectives.OfType<PatchedFromArchive>().FirstOrDefault(f => f.Patch == null) != null)
|
|
{
|
|
Error("Missing patches after generation, this should not happen");
|
|
}
|
|
|
|
}
|
|
|
|
private void BuildArchivePatches(string archive_sha, IEnumerable<PatchedFromArchive> group, Dictionary<string, string> absolute_paths)
|
|
{
|
|
var archive = VFS.HashIndex[archive_sha];
|
|
using (var files = VFS.StageWith(group.Select(g => VFS.FileForArchiveHashPath(g.ArchiveHashPath))))
|
|
{
|
|
var by_path = files.GroupBy(f => string.Join("|", f.Paths.Skip(1))).ToDictionary(f => f.Key, f => f.First());
|
|
// Now Create the patches
|
|
group.PMap(entry =>
|
|
{
|
|
Info("Patching {0}", entry.To);
|
|
using (var origin = by_path[string.Join("|", entry.ArchiveHashPath.Skip(1))].OpenRead())
|
|
using (var output = new MemoryStream())
|
|
{
|
|
var a = origin.ReadAll();
|
|
var b = LoadDataForTo(entry.To, absolute_paths);
|
|
BSDiff.Create(a, b, output);
|
|
entry.Patch = output.ToArray().ToBase64();
|
|
Info($"Patch size {entry.Patch.Length} for {entry.To}");
|
|
}
|
|
});
|
|
}
|
|
|
|
}
|
|
|
|
private byte[] LoadDataForTo(string to, Dictionary<string, string> absolute_paths)
|
|
{
|
|
if (absolute_paths.TryGetValue(to, out var absolute))
|
|
return File.ReadAllBytes(absolute);
|
|
|
|
if (to.StartsWith(Consts.BSACreationDir))
|
|
{
|
|
var bsa_id = to.Split('\\')[1];
|
|
var bsa = InstallDirectives.OfType<CreateBSA>().First(b => b.TempID == bsa_id);
|
|
|
|
using (var a = new BSAReader(Path.Combine(MO2Folder, bsa.To)))
|
|
{
|
|
var file = a.Files.First(e => e.Path == Path.Combine(to.Split('\\').Skip(2).ToArray()));
|
|
return file.GetData();
|
|
}
|
|
|
|
}
|
|
Error($"Couldn't load data for {to}");
|
|
return null;
|
|
}
|
|
|
|
private void GatherArchives()
|
|
{
|
|
Info($"Building a list of archives based on the files required");
|
|
|
|
var shas = InstallDirectives.OfType<FromArchive>()
|
|
.Select(a => a.ArchiveHashPath[0])
|
|
.Distinct();
|
|
|
|
var archives = IndexedArchives.OrderByDescending(f => f.File.LastModified)
|
|
.GroupBy(f => f.File.Hash)
|
|
.ToDictionary(f => f.Key, f => f.First());
|
|
|
|
SelectedArchives = shas.PMap(sha => ResolveArchive(sha, archives));
|
|
|
|
}
|
|
|
|
private Archive ResolveArchive(string sha, IDictionary<string, IndexedArchive> archives)
|
|
{
|
|
if (archives.TryGetValue(sha, out var found))
|
|
{
|
|
if (found.IniData == null)
|
|
Error("No download metadata found for {0}, please use MO2 to query info or add a .meta file and try again.", found.Name);
|
|
var general = found.IniData.General;
|
|
if (general == null)
|
|
Error("No General section in mod metadata found for {0}, please use MO2 to query info or add the info and try again.", found.Name);
|
|
|
|
Archive result;
|
|
|
|
if (general.directURL != null && general.directURL.StartsWith("https://drive.google.com"))
|
|
{
|
|
var regex = new Regex("((?<=id=)[a-zA-Z0-9_-]*)|(?<=\\/file\\/d\\/)[a-zA-Z0-9_-]*");
|
|
var match = regex.Match(general.directURL);
|
|
result = new GoogleDriveMod()
|
|
{
|
|
Id = match.ToString()
|
|
};
|
|
}
|
|
else if (general.directURL != null && general.directURL.StartsWith(Consts.MegaPrefix))
|
|
{
|
|
result = new MEGAArchive()
|
|
{
|
|
URL = general.directURL
|
|
};
|
|
}
|
|
else if (general.directURL != null && general.directURL.StartsWith("https://www.dropbox.com/"))
|
|
{
|
|
var uri = new UriBuilder((string)general.directURL);
|
|
var query = HttpUtility.ParseQueryString(uri.Query);
|
|
|
|
if (query.GetValues("dl").Count() > 0)
|
|
query.Remove("dl");
|
|
|
|
query.Set("dl", "1");
|
|
|
|
uri.Query = query.ToString();
|
|
|
|
result = new DirectURLArchive()
|
|
{
|
|
URL = uri.ToString()
|
|
};
|
|
}
|
|
else if (general.directURL != null && general.directURL.StartsWith("https://www.moddb.com/downloads/start"))
|
|
{
|
|
result = new MODDBArchive()
|
|
{
|
|
URL = general.directURL
|
|
};
|
|
}
|
|
else if (general.directURL != null && general.directURL.StartsWith("http://www.mediafire.com/file/"))
|
|
{
|
|
Error("Mediafire links are not currently supported");
|
|
return null;
|
|
/*result = new MediaFireArchive()
|
|
{
|
|
URL = general.directURL
|
|
};*/
|
|
}
|
|
else if (general.directURL != null)
|
|
{
|
|
|
|
var tmp = new DirectURLArchive()
|
|
{
|
|
URL = general.directURL
|
|
};
|
|
if (general.directURLHeaders != null)
|
|
{
|
|
tmp.Headers = new List<string>();
|
|
tmp.Headers.AddRange(general.directURLHeaders.Split('|'));
|
|
}
|
|
result = tmp;
|
|
}
|
|
else if (general.manualURL != null)
|
|
{
|
|
result = new ManualURLArchive()
|
|
{
|
|
URL = general.manualURL.ToString()
|
|
};
|
|
}
|
|
else if (general.modID != null && general.fileID != null && general.gameName != null)
|
|
{
|
|
result = new NexusMod()
|
|
{
|
|
GameName = general.gameName,
|
|
FileID = general.fileID,
|
|
ModID = general.modID
|
|
};
|
|
Status($"Getting Nexus info for {found.Name}");
|
|
try
|
|
{
|
|
var link = NexusAPI.GetNexusDownloadLink((NexusMod)result, NexusKey, true);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
Error($"Unable to resolve {found.Name} on the Nexus was the file removed?");
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
Error("No way to handle archive {0} but it's required by the modpack", found.Name);
|
|
return null;
|
|
}
|
|
|
|
result.Name = found.Name;
|
|
result.Hash = found.File.Hash;
|
|
result.Meta = found.Meta;
|
|
|
|
return result;
|
|
}
|
|
Error("No match found for Archive sha: {0} this shouldn't happen", sha);
|
|
return null;
|
|
}
|
|
|
|
|
|
private Directive RunStack(IEnumerable<Func<RawSourceFile, Directive>> stack, RawSourceFile source)
|
|
{
|
|
Status("Compiling {0}", source.Path);
|
|
foreach (var f in stack)
|
|
{
|
|
var result = f(source);
|
|
if (result != null) return result;
|
|
}
|
|
throw new InvalidDataException("Data fell out of the compilation stack");
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// Creates a execution stack. The stack should be passed into Run stack. Each function
|
|
/// in this stack will be run in-order and the first to return a non-null result will have its
|
|
/// result included into the pack
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
private IEnumerable<Func<RawSourceFile, Directive>> MakeStack()
|
|
{
|
|
Info("Generating compilation stack");
|
|
return new List<Func<RawSourceFile, Directive>>()
|
|
{
|
|
IgnoreStartsWith("logs\\"),
|
|
IncludeRegex("^downloads\\\\.*\\.meta"),
|
|
IgnoreStartsWith("downloads\\"),
|
|
IgnoreStartsWith("webcache\\"),
|
|
IgnoreStartsWith("overwrite\\"),
|
|
IgnorePathContains("temporary_logs"),
|
|
IgnorePathContains("GPUCache"),
|
|
IgnorePathContains("SSEEdit Cache"),
|
|
IgnoreEndsWith(".pyc"),
|
|
IgnoreEndsWith(".log"),
|
|
IgnoreOtherProfiles(),
|
|
IgnoreDisabledMods(),
|
|
IncludeThisProfile(),
|
|
// Ignore the ModOrganizer.ini file it contains info created by MO2 on startup
|
|
IncludeStubbedMO2Ini(),
|
|
IncludeLootFiles(),
|
|
IgnoreStartsWith(Path.Combine(Consts.GameFolderFilesDir, "Data")),
|
|
IgnoreStartsWith(Path.Combine(Consts.GameFolderFilesDir, "Papyrus Compiler")),
|
|
IgnoreStartsWith(Path.Combine(Consts.GameFolderFilesDir, "Skyrim")),
|
|
IgnoreRegex(Consts.GameFolderFilesDir + "\\\\.*\\.bsa"),
|
|
IncludeModIniData(),
|
|
DirectMatch(),
|
|
DeconstructBSAs(), // Deconstruct BSAs before building patches so we don't generate massive patch files
|
|
IncludePatches(),
|
|
IncludeDummyESPs(),
|
|
IncludeTaggedFiles(),
|
|
|
|
|
|
// If we have no match at this point for a game folder file, skip them, we can't do anything about them
|
|
IgnoreGameFiles(),
|
|
|
|
// There are some types of files that will error the compilation, because tehy're created on-the-fly via tools
|
|
// so if we don't have a match by this point, just drop them.
|
|
IgnoreEndsWith(".ini"),
|
|
IgnoreEndsWith(".html"),
|
|
IgnoreEndsWith(".txt"),
|
|
// Don't know why, but this seems to get copied around a bit
|
|
IgnoreEndsWith("HavokBehaviorPostProcess.exe"),
|
|
// Theme file MO2 downloads somehow
|
|
IgnoreEndsWith("splash.png"),
|
|
|
|
PatchStockESMs(),
|
|
|
|
DropAll()
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> PatchStockESMs()
|
|
{
|
|
return source =>
|
|
{
|
|
string filename = Path.GetFileName(source.Path);
|
|
string game_file = Path.Combine(GamePath, "Data", filename);
|
|
if (Consts.GameESMs.Contains(filename) && source.Path.StartsWith("mods\\") && File.Exists(game_file))
|
|
{
|
|
Info($"A ESM named {filename} was found in a mod that shares a name with a core game ESMs, it is assumed this is a cleaned ESM and it will be binary patched.");
|
|
var result = source.EvolveTo<CleanedESM>();
|
|
result.SourceESMHash = VFS.Lookup(game_file).Hash;
|
|
|
|
Status($"Generating patch of {filename}");
|
|
using (var ms = new MemoryStream()) {
|
|
BSDiff.Create(File.ReadAllBytes(game_file), File.ReadAllBytes(source.AbsolutePath), ms);
|
|
result.SourceData = ms.ToArray().ToBase64();
|
|
}
|
|
Info($"Generated a {result.SourceData.Length} byte patch for {filename}");
|
|
|
|
return result;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
|
|
|
|
private Func<RawSourceFile, Directive> IncludeLootFiles()
|
|
{
|
|
var prefix = Consts.LOOTFolderFilesDir + "\\";
|
|
return source =>
|
|
{
|
|
|
|
if (source.Path.StartsWith(prefix))
|
|
{
|
|
var result = source.EvolveTo<InlineFile>();
|
|
result.SourceData = File.ReadAllBytes(source.AbsolutePath).ToBase64();
|
|
return result;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IncludeStubbedMO2Ini()
|
|
{
|
|
return source =>
|
|
{
|
|
if (source.Path == "ModOrganizer.ini")
|
|
{
|
|
return RemapIni(source, GamePath);
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Directive RemapIni(RawSourceFile source, string gamePath)
|
|
{
|
|
var data = File.ReadAllText(source.AbsolutePath);
|
|
|
|
data = data.Replace(GamePath, Consts.GAME_PATH_MAGIC_BACK);
|
|
data = data.Replace(GamePath.Replace("\\", "\\\\"), Consts.GAME_PATH_MAGIC_DOUBLE_BACK);
|
|
data = data.Replace(GamePath.Replace("\\", "/"), Consts.GAME_PATH_MAGIC_FORWARD);
|
|
|
|
data = data.Replace(MO2Folder, Consts.MO2_PATH_MAGIC_BACK);
|
|
data = data.Replace(MO2Folder.Replace("\\", "\\\\"), Consts.MO2_PATH_MAGIC_DOUBLE_BACK);
|
|
data = data.Replace(MO2Folder.Replace("\\", "/"), Consts.MO2_PATH_MAGIC_FORWARD);
|
|
var result = source.EvolveTo<RemappedInlineFile>();
|
|
result.SourceData = Encoding.UTF8.GetBytes(data).ToBase64();
|
|
return result;
|
|
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IgnorePathContains(string v)
|
|
{
|
|
v = $"\\{v.Trim('\\')}\\";
|
|
var reason = $"Ignored because path contains {v}";
|
|
return source =>
|
|
{
|
|
if (source.Path.Contains(v))
|
|
{
|
|
var result = source.EvolveTo<IgnoredDirectly>();
|
|
result.Reason = reason;
|
|
return result;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// If a user includes WABBAJACK_INCLUDE directly in the notes or comments of a mod, the contents of that
|
|
/// mod will be inlined into the installer. USE WISELY.
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
private Func<RawSourceFile, Directive> IncludeTaggedFiles()
|
|
{
|
|
var include_directly = ModInis.Where(kv => {
|
|
var general = kv.Value.General;
|
|
if (general.notes != null && general.notes.Contains(Consts.WABBAJACK_INCLUDE))
|
|
return true;
|
|
if (general.comments != null && general.comments.Contains(Consts.WABBAJACK_INCLUDE))
|
|
return true;
|
|
return false;
|
|
}).Select(kv => $"mods\\{kv.Key}\\");
|
|
|
|
|
|
return source =>
|
|
{
|
|
|
|
if (source.Path.StartsWith("mods"))
|
|
{
|
|
foreach (var modpath in include_directly)
|
|
{
|
|
if (source.Path.StartsWith(modpath))
|
|
{
|
|
var result = source.EvolveTo<InlineFile>();
|
|
result.SourceData = File.ReadAllBytes(source.AbsolutePath).ToBase64();
|
|
return result;
|
|
}
|
|
}
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// Some tools like the Cathedral Asset Optimizer will create dummy ESPs whos only existance is to make
|
|
/// sure a BSA with the same name is loaded. We don't have a good way to detect these, but if an ESP is
|
|
/// less than 100 bytes in size and shares a name with a BSA it's a pretty good chance that it's a dummy
|
|
/// and the contents are generated.
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
private Func<RawSourceFile, Directive> IncludeDummyESPs()
|
|
{
|
|
return source =>
|
|
{
|
|
if (Path.GetExtension(source.AbsolutePath) == ".esp")
|
|
{
|
|
var bsa = Path.Combine(Path.GetDirectoryName(source.AbsolutePath), Path.GetFileNameWithoutExtension(source.AbsolutePath) + ".bsa");
|
|
var bsa_textures = Path.Combine(Path.GetDirectoryName(source.AbsolutePath), Path.GetFileNameWithoutExtension(source.AbsolutePath) + " - Textures.bsa");
|
|
var esp_size = new FileInfo(source.AbsolutePath).Length;
|
|
if (esp_size <= 100 && (File.Exists(bsa) || File.Exists(bsa_textures)))
|
|
{
|
|
var inline = source.EvolveTo<InlineFile>();
|
|
inline.SourceData = File.ReadAllBytes(source.AbsolutePath).ToBase64();
|
|
return inline;
|
|
}
|
|
}
|
|
|
|
return null;
|
|
};
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// This function will search for a way to create a BSA in the installed mod list by assembling it from files
|
|
/// found in archives. To do this we hash all the files in side the BSA then try to find matches and patches for
|
|
/// all of the files.
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
private Func<RawSourceFile, Directive> DeconstructBSAs()
|
|
{
|
|
var include_directly = ModInis.Where(kv => {
|
|
var general = kv.Value.General;
|
|
if (general.notes != null && general.notes.Contains(Consts.WABBAJACK_INCLUDE))
|
|
return true;
|
|
if (general.comments != null && general.comments.Contains(Consts.WABBAJACK_INCLUDE))
|
|
return true;
|
|
return false;
|
|
}).Select(kv => $"mods\\{kv.Key}\\");
|
|
|
|
var microstack = new List<Func<RawSourceFile, Directive>>()
|
|
{
|
|
DirectMatch(),
|
|
IncludePatches(),
|
|
DropAll()
|
|
};
|
|
|
|
var microstack_with_include = new List<Func<RawSourceFile, Directive>>()
|
|
{
|
|
DirectMatch(),
|
|
IncludePatches(),
|
|
IncludeALL()
|
|
};
|
|
|
|
|
|
return source =>
|
|
{
|
|
if (!Consts.SupportedBSAs.Contains(Path.GetExtension(source.Path))) return null;
|
|
|
|
bool default_include = false;
|
|
if (source.Path.StartsWith("mods"))
|
|
{
|
|
foreach (var modpath in include_directly)
|
|
{
|
|
if (source.Path.StartsWith(modpath))
|
|
{
|
|
default_include = true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
var source_files = source.File.FileInArchive;
|
|
|
|
var stack = default_include ? microstack_with_include : microstack;
|
|
|
|
var matches = source_files.PMap(e => RunStack(stack, new RawSourceFile(e)));
|
|
|
|
var id = Guid.NewGuid().ToString();
|
|
|
|
foreach (var match in matches)
|
|
{
|
|
if (match is IgnoredDirectly)
|
|
{
|
|
Error($"File required for BSA creation doesn't exist: {match.To}");
|
|
}
|
|
match.To = Path.Combine(Consts.BSACreationDir, id, match.To);
|
|
ExtraFiles.Add(match);
|
|
};
|
|
|
|
CreateBSA directive;
|
|
using (var bsa = new BSAReader(source.AbsolutePath))
|
|
{
|
|
directive = new CreateBSA()
|
|
{
|
|
To = source.Path,
|
|
TempID = id,
|
|
Type = (uint)bsa.HeaderType,
|
|
FileFlags = (uint)bsa.FileFlags,
|
|
ArchiveFlags = (uint)bsa.ArchiveFlags,
|
|
};
|
|
};
|
|
|
|
return directive;
|
|
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IncludeALL()
|
|
{
|
|
return source =>
|
|
{
|
|
var inline = source.EvolveTo<InlineFile>();
|
|
inline.SourceData = File.ReadAllBytes(source.AbsolutePath).ToBase64();
|
|
return inline;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IgnoreDisabledMods()
|
|
{
|
|
var disabled_mods = File.ReadAllLines(Path.Combine(MO2ProfileDir, "modlist.txt"))
|
|
.Where(line => line.StartsWith("-") && !line.EndsWith("_separator"))
|
|
.Select(line => Path.Combine("mods", line.Substring(1)) + "\\")
|
|
.ToList();
|
|
return source =>
|
|
{
|
|
if (disabled_mods.FirstOrDefault(mod => source.Path.StartsWith(mod)) != null)
|
|
{
|
|
var r = source.EvolveTo<IgnoredDirectly>();
|
|
r.Reason = "Disabled Mod";
|
|
return r;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IncludePatches()
|
|
{
|
|
var indexed = IndexedFiles.Values
|
|
.SelectMany(f => f)
|
|
.GroupBy(f => Path.GetFileName(f.Paths.Last()).ToLower())
|
|
.ToDictionary(f => f.Key);
|
|
|
|
return source =>
|
|
{
|
|
if (indexed.TryGetValue(Path.GetFileName(source.Path.ToLower()), out var value))
|
|
{
|
|
// TODO: Improve this
|
|
var found = value.First();
|
|
|
|
var e = source.EvolveTo<PatchedFromArchive>();
|
|
e.ArchiveHashPath = found.MakeRelativePaths();
|
|
e.To = source.Path;
|
|
return e;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IncludeModIniData()
|
|
{
|
|
return source =>
|
|
{
|
|
if (source.Path.StartsWith("mods\\") && source.Path.EndsWith("\\meta.ini"))
|
|
{
|
|
var e = source.EvolveTo<InlineFile>();
|
|
e.SourceData = File.ReadAllBytes(source.AbsolutePath).ToBase64();
|
|
return e;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IgnoreGameFiles()
|
|
{
|
|
var start_dir = Consts.GameFolderFilesDir + "\\";
|
|
return source =>
|
|
{
|
|
if (source.Path.StartsWith(start_dir))
|
|
{
|
|
var i = source.EvolveTo<IgnoredDirectly>();
|
|
i.Reason = "Default game file";
|
|
return i;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IncludeThisProfile()
|
|
{
|
|
var correct_profile = Path.Combine("profiles", MO2Profile) + "\\";
|
|
return source =>
|
|
{
|
|
if (source.Path.StartsWith(correct_profile))
|
|
{
|
|
byte[] data;
|
|
if (source.Path.EndsWith("\\modlist.txt"))
|
|
data = ReadAndCleanModlist(source.AbsolutePath);
|
|
else
|
|
data = File.ReadAllBytes(source.AbsolutePath);
|
|
|
|
var e = source.EvolveTo<InlineFile>();
|
|
e.SourceData = data.ToBase64();
|
|
return e;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private 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));
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IgnoreOtherProfiles()
|
|
{
|
|
var correct_profile = Path.Combine("profiles", MO2Profile) + "\\";
|
|
return source =>
|
|
{
|
|
if (source.Path.StartsWith("profiles\\") && !source.Path.StartsWith(correct_profile))
|
|
{
|
|
var c = source.EvolveTo<IgnoredDirectly>();
|
|
c.Reason = "File not for this profile";
|
|
return c;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IgnoreEndsWith(string v)
|
|
{
|
|
var reason = String.Format("Ignored because path ends with {0}", v);
|
|
return source =>
|
|
{
|
|
if (source.Path.EndsWith(v))
|
|
{
|
|
var result = source.EvolveTo<IgnoredDirectly>();
|
|
result.Reason = reason;
|
|
return result;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IgnoreRegex(string p)
|
|
{
|
|
var reason = String.Format("Ignored because path matches regex {0}", p);
|
|
var regex = new Regex(p);
|
|
return source =>
|
|
{
|
|
if (regex.IsMatch(source.Path))
|
|
{
|
|
var result = source.EvolveTo<IgnoredDirectly>();
|
|
result.Reason = reason;
|
|
return result;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
|
|
private Func<RawSourceFile, Directive> IncludeRegex(string pattern)
|
|
{
|
|
var regex = new Regex(pattern);
|
|
return source =>
|
|
{
|
|
if (regex.IsMatch(source.Path))
|
|
{
|
|
var result = source.EvolveTo<InlineFile>();
|
|
result.SourceData = File.ReadAllBytes(source.AbsolutePath).ToBase64();
|
|
return result;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
|
|
private Func<RawSourceFile, Directive> DropAll()
|
|
{
|
|
return source => {
|
|
var result = source.EvolveTo<NoMatch>();
|
|
result.Reason = "No Match in Stack";
|
|
Info($"No match for: {source.Path}");
|
|
return result;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> DirectMatch()
|
|
{
|
|
|
|
return source =>
|
|
{
|
|
if (IndexedFiles.TryGetValue(source.Hash, out var found))
|
|
{
|
|
var result = source.EvolveTo<FromArchive>();
|
|
|
|
var match = found.Where(f => Path.GetFileName(f.Paths[0]) == Path.GetFileName(source.Path))
|
|
.FirstOrDefault();
|
|
|
|
if (match == null)
|
|
match = found.FirstOrDefault();
|
|
|
|
result.ArchiveHashPath = match.MakeRelativePaths();
|
|
|
|
return result;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
private Func<RawSourceFile, Directive> IgnoreStartsWith(string v)
|
|
{
|
|
var reason = String.Format("Ignored because path starts with {0}", v);
|
|
return source =>
|
|
{
|
|
if (source.Path.StartsWith(v))
|
|
{
|
|
var result = source.EvolveTo<IgnoredDirectly>();
|
|
result.Reason = reason;
|
|
return result;
|
|
}
|
|
return null;
|
|
};
|
|
}
|
|
|
|
internal void PatchExecutable()
|
|
{
|
|
var settings = new JsonSerializerSettings() { TypeNameHandling = TypeNameHandling.Auto };
|
|
var data = JsonConvert.SerializeObject(ModList, settings).BZip2String();
|
|
var executable = Assembly.GetExecutingAssembly().Location;
|
|
var out_path = Path.Combine(Path.GetDirectoryName(executable), MO2Profile + ".exe");
|
|
Info("Patching Executable {0}", Path.GetFileName(out_path));
|
|
File.Copy(executable, out_path, true);
|
|
using (var os = File.OpenWrite(out_path))
|
|
using (var bw = new BinaryWriter(os))
|
|
{
|
|
long orig_pos = os.Length;
|
|
os.Position = os.Length;
|
|
bw.Write(data.LongLength);
|
|
bw.Write(data);
|
|
bw.Write(orig_pos);
|
|
bw.Write(Encoding.ASCII.GetBytes(Consts.ModPackMagic));
|
|
}
|
|
}
|
|
}
|
|
}
|