public async Task ModifiedIniFilesArePatchedAgainstFileWithSameName()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var ini = utils.AddModFile(mod, @"foo.ini", 10);
var meta = utils.AddModFile(mod, "meta.ini");
utils.Configure();
var archive = utils.AddManualDownload(
new Dictionary <string, byte[]> {
{ "/baz/foo.ini", File.ReadAllBytes(ini) }
});
File.WriteAllLines(meta, new[]
{
"[General]",
$"installationFile={archive}",
});
// Modify after creating mod archive in the downloads folder
File.WriteAllText(ini, "Wabbajack, Wabbajack, Wabbajack!");
var modlist = await CompileAndInstall(profile);
var directive = modlist.Directives.Where(m => m.To == $"mods\\{mod}\\foo.ini").FirstOrDefault();
Assert.IsNotNull(directive);
Assert.IsInstanceOfType(directive, typeof(PatchedFromArchive));
}
public override async ValueTask <Directive> Run(RawSourceFile source)
{
var filename = Path.GetFileName(source.Path);
var gameFile = Path.Combine(_mo2Compiler.GamePath, "Data", filename);
if (!Consts.GameESMs.Contains(filename) || !source.Path.StartsWith("mods\\") ||
!File.Exists(gameFile))
{
return(null);
}
Utils.Log(
$"An ESM named {filename} was found in a mod that shares a name with one of the core game ESMs, it is assumed this is a cleaned ESM and it will be binary patched");
var result = source.EvolveTo <CleanedESM>();
result.SourceESMHash = _compiler.VFS.Index.ByRootPath[gameFile].Hash;
Utils.Status($"Generating patch of {filename}");
await using (var ms = new MemoryStream())
{
await Utils.CreatePatch(File.ReadAllBytes(gameFile), File.ReadAllBytes(source.AbsolutePath), ms);
var data = ms.ToArray();
result.SourceDataID = _compiler.IncludeFile(data);
Utils.Log($"Generated a {data.Length} byte patch for {filename}");
}
return(result);
}
public async Task TestUpdating()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var unchanged = utils.AddModFile(mod, @"Data\scripts\unchanged.pex", 10);
var deleted = utils.AddModFile(mod, @"Data\scripts\deleted.pex", 10);
var modified = utils.AddModFile(mod, @"Data\scripts\modified.pex", 10);
utils.Configure();
utils.AddManualDownload(
new Dictionary <string, byte[]>
{
{ "/baz/unchanged.pex", File.ReadAllBytes(unchanged) },
{ "/baz/deleted.pex", File.ReadAllBytes(deleted) },
{ "/baz/modified.pex", File.ReadAllBytes(modified) },
});
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\unchanged.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\deleted.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\modified.pex");
var unchanged_path = utils.PathOfInstalledFile(mod, @"Data\scripts\unchanged.pex");
var deleted_path = utils.PathOfInstalledFile(mod, @"Data\scripts\deleted.pex");
var modified_path = utils.PathOfInstalledFile(mod, @"Data\scripts\modified.pex");
var extra_path = utils.PathOfInstalledFile(mod, @"something_i_made.foo");
File.WriteAllText(extra_path, "bleh");
var extra_folder = Path.Combine(Path.GetDirectoryName(utils.PathOfInstalledFile(mod, @"something_i_made.foo")), "folder_i_made");
Directory.CreateDirectory(extra_folder);
Assert.IsTrue(Directory.Exists(extra_folder));
var unchanged_modified = File.GetLastWriteTime(unchanged_path);
var modified_modified = File.GetLastWriteTime(modified_path);
File.WriteAllText(modified_path, "random data");
File.Delete(deleted_path);
Assert.IsTrue(File.Exists(extra_path));
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\unchanged.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\deleted.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\modified.pex");
Assert.AreEqual(unchanged_modified, File.GetLastWriteTime(unchanged_path));
Assert.AreNotEqual(modified_modified, File.GetLastWriteTime(modified_path));
Assert.IsFalse(File.Exists(extra_path));
Assert.IsFalse(Directory.Exists(extra_folder));
}
public async Task TestDirectMatchFromGameFolder()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var test_pex = utils.AddGameFile(@"enbstuff\test.pex", 10);
utils.Configure();
utils.AddManualDownload(
new Dictionary <string, byte[]> {
{ "/baz/biz.pex", File.ReadAllBytes(test_pex) }
});
await CompileAndInstall(profile);
utils.VerifyInstalledGameFile(@"enbstuff\test.pex");
}
public async Task TestDirectMatch()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var test_pex = utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
utils.Configure();
utils.AddManualDownload(
new Dictionary <string, byte[]> {
{ "/baz/biz.pex", File.ReadAllBytes(test_pex) }
});
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
}
public async Task TestDirectMatchIsIgnoredWhenGameFolderFilesOverrideExists()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var test_pex = utils.AddGameFile(@"enbstuff\test.pex", 10);
utils.Configure();
Directory.CreateDirectory(Path.Combine(utils.MO2Folder, Consts.GameFolderFilesDir));
utils.AddManualDownload(
new Dictionary <string, byte[]> {
{ "/baz/biz.pex", File.ReadAllBytes(test_pex) }
});
await CompileAndInstall(profile);
Assert.IsFalse(File.Exists(Path.Combine(utils.InstallFolder, Consts.GameFolderFilesDir, @"enbstuff\test.pex")));
}
public void UnmodifiedInlinedFilesArePulledFromArchives()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var ini = utils.AddModFile(mod, @"foo.ini", 10);
utils.Configure();
utils.AddManualDownload(
new Dictionary <string, byte[]> {
{ "/baz/biz.pex", File.ReadAllBytes(ini) }
});
var modlist = CompileAndInstall(profile);
var directive = modlist.Directives.Where(m => m.To == $"mods\\{mod}\\foo.ini").FirstOrDefault();
Assert.IsNotNull(directive);
Assert.IsInstanceOfType(directive, typeof(FromArchive));
}
public async Task TestDuplicateFilesAreCopied()
{
var profile = utils.AddProfile();
var mod = utils.AddMod();
var test_pex = utils.AddModFile(mod, @"Data\scripts\test.pex", 10);
// Make a copy to make sure it gets picked up and moved around.
File.Copy(test_pex, test_pex + ".copy");
utils.Configure();
utils.AddManualDownload(
new Dictionary <string, byte[]> {
{ "/baz/biz.pex", File.ReadAllBytes(test_pex) }
});
await CompileAndInstall(profile);
utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex");
utils.VerifyInstalledFile(mod, @"Data\scripts\test.pex.copy");
}
public void VerifyInstalledFile(string mod, string file)
{
var src = Path.Combine(MO2Folder, "mods", mod, file);
Assert.IsTrue(File.Exists(src), src);
var dest = Path.Combine(InstallFolder, "mods", mod, file);
Assert.IsTrue(File.Exists(dest), dest);
var src_data = File.ReadAllBytes(src);
var dest_data = File.ReadAllBytes(dest);
Assert.AreEqual(src_data.Length, dest_data.Length);
for (int x = 0; x < src_data.Length; x++)
{
if (src_data[x] != dest_data[x])
{
Assert.Fail($"Index {x} of {mod}\\{file} are not the same");
}
}
}
public void ImageNodeContentFromPath(string path)
{
image_data = File.ReadAllBytes(path);
bitmap_source = null;
}
internal static void RunUpgrade()
{
Logging.Info("Upgrading from 037 to 038");
string base_directory_path = BaseDirectoryForQiqqa;
if (Directory.Exists(base_directory_path))
{
int info_library_count, info_item_count;
string[] library_directories = Directory.GetDirectories(base_directory_path);
info_library_count = 0;
foreach (string library_directory in library_directories)
{
++info_library_count;
Logging.Info("Inspecting directory {0}", library_directory);
string documents_directory = Path.GetFullPath(Path.Combine(library_directory, @"documents"));
string database_file = LibraryDB.GetLibraryDBPath(library_directory);
string database_syncref_file = IntranetLibraryTools.GetLibraryMetadataPath(library_directory);
// make sure we skip S3DB internet DB sync directories and only 'go through the upgrade process
// when this looks like a viable (local) Qiqqa library:
if (!File.Exists(database_file) && Directory.Exists(documents_directory) && !File.Exists(database_syncref_file))
{
Logging.Warn("We have to upgrade {0}", library_directory);
SQLiteUpgrade_LibraryDB library_db = new SQLiteUpgrade_LibraryDB(library_directory);
using (var connection = library_db.GetConnection())
{
connection.Open();
using (var transaction = connection.BeginTransaction())
{
// Get a list of ALL the files in the documents directory...
string[] full_filenames = Directory.GetFiles(documents_directory, "*.*", SearchOption.AllDirectories);
info_item_count = 0;
foreach (string full_filename in full_filenames)
{
++info_item_count;
StatusManager.Instance.UpdateStatus("DBUpgrade", String.Format("Upgrading library {0}/{1}", info_library_count, library_directories.Length), info_item_count, full_filenames.Length);
string fingerprint = Path.GetFileNameWithoutExtension(full_filename);
string extension = Path.GetExtension(full_filename).Trim('.');
if (EXTENSIONS.Contains(extension))
{
Logging.Info("Upgrading {0}--{1}", fingerprint, extension);
byte[] data = File.ReadAllBytes(full_filename);
library_db.PutBlob(connection, transaction, fingerprint, extension, data);
}
else
{
Logging.Info("NOT upgrading {0}--{1}", fingerprint, extension);
}
}
transaction.Commit();
}
}
}
}
}
StatusManager.Instance.UpdateStatus("DBUpgrade", "Finished migrating libraries.");
}
internal static void RunUpgrade(SplashScreenWindow splashscreen_window)
{
Logging.Info("Upgrading from 037 to 038");
string base_directory_path = BaseDirectoryForQiqqa;
if (Directory.Exists(base_directory_path))
{
int info_library_count, info_item_count;
string[] library_directories = Directory.GetDirectories(base_directory_path);
info_library_count = 0;
foreach (string library_directory in library_directories)
{
++info_library_count;
Logging.Info("Inspecting directory {0}", library_directory);
string documents_directory = Path.GetFullPath(Path.Combine(library_directory, @"documents"));
string database_file = Path.GetFullPath(Path.Combine(library_directory, @"Qiqqa.library"));
if (!File.Exists(database_file) && Directory.Exists(documents_directory))
{
Logging.Warn("We have to upgrade {0}", library_directory);
SQLiteUpgrade_LibraryDB library_db = new SQLiteUpgrade_LibraryDB(library_directory);
using (var connection = library_db.GetConnection())
{
connection.Open();
using (var transaction = connection.BeginTransaction())
{
// Get a list of ALL the files in the documents directory...
string[] full_filenames = Directory.GetFiles(documents_directory, "*.*", SearchOption.AllDirectories);
info_item_count = 0;
foreach (string full_filename in full_filenames)
{
++info_item_count;
splashscreen_window.UpdateMessage("Upgrading library {0}/{1}: {2:P0}", info_library_count, library_directories.Length, info_item_count / (double)full_filenames.Length);
string fingerprint = Path.GetFileNameWithoutExtension(full_filename);
string extension = Path.GetExtension(full_filename).Trim('.');
if (EXTENSIONS.Contains(extension))
{
Logging.Info("Upgrading {0}--{1}", fingerprint, extension);
byte[] data = File.ReadAllBytes(full_filename);
library_db.PutBlob(connection, transaction, fingerprint, extension, data);
}
else
{
Logging.Info("NOT upgrading {0}--{1}", fingerprint, extension);
}
}
transaction.Commit();
}
}
}
}
}
splashscreen_window.UpdateMessage("Finished migrating libraries.");
}
public override byte[] ReadAllBytes(string path)
{
return(AfsFile.ReadAllBytes(path));
}