/// <summary>
/// Uses the basegame file DB to attempt to determine the source mod of this file. This method should be run on a background thread.
/// </summary>
public void DetermineSource()
{
var fullpath = Path.Combine(target.TargetPath, FilePath);
if (File.Exists(fullpath))
{
//if (FilePath.Equals(Utilities.Getth, StringComparison.InvariantCultureIgnoreCase)) return; //don't report this file
var info = BasegameFileIdentificationService.GetBasegameFileSource(target, fullpath);
if (info != null)
{
ModificationSource = info.source;
}
// TODO: MAKE LOCAL DB??
#if DEBUG
else
{
ModificationSource = Utilities.CalculateMD5(fullpath);
}
#endif
}
}
public static bool RunPlotManagerUpdate(GameTarget target)
{
Log.Information($@"Updating PlotManager for game: {target.TargetPath}");
var supercedances = M3Directories.GetFileSupercedances(target, new[] { @".pmu" });
Dictionary <string, string> funcMap = new();
List <string> combinedNames = new List <string>();
if (supercedances.TryGetValue(@"PlotManagerUpdate.pmu", out var supercedanes))
{
supercedanes.Reverse(); // list goes from highest to lowest. We want to build in lowest to highest
StringBuilder sb = null;
string currentFuncNum = null;
var metaMaps = M3Directories.GetMetaMappedInstalledDLC(target, false);
foreach (var pmuDLCName in supercedanes)
{
var uiName = metaMaps[pmuDLCName]?.ModName ?? ThirdPartyServices.GetThirdPartyModInfo(pmuDLCName, target.Game)?.modname ?? pmuDLCName;
combinedNames.Add(uiName);
var text = File.ReadAllLines(Path.Combine(M3Directories.GetDLCPath(target), pmuDLCName, target.Game.CookedDirName(), @"PlotManagerUpdate.pmu"));
foreach (var line in text)
{
if (line.StartsWith(@"public function bool F"))
{
if (sb != null)
{
funcMap[currentFuncNum] = sb.ToString();
Log.Information($@"PlotSync: Adding function {currentFuncNum} from {pmuDLCName}");
currentFuncNum = null;
}
sb = new StringBuilder();
sb.AppendLine(line);
// Method name
currentFuncNum = line.Substring(22);
currentFuncNum = currentFuncNum.Substring(0, currentFuncNum.IndexOf('('));
if (int.TryParse(currentFuncNum, out var num))
{
if (num <= 0)
{
Log.Error($@"Skipping plot manager update: Conditional {num} is not a valid number for use. Values must be greater than 0 and less than 2 billion.");
Analytics.TrackEvent(@"Bad plot manager function", new Dictionary <string, string>()
{
{ @"FunctionName", $@"F{currentFuncNum}" },
{ @"DLCName", pmuDLCName }
});
sb = null;
return(false);
}
else if (num.ToString().Length != currentFuncNum.Length)
{
Log.Error($@"Skipping plot manager update: Conditional {currentFuncNum} is not a valid number for use. Values must not contain leading zeros");
Analytics.TrackEvent(@"Bad plot manager function", new Dictionary <string, string>()
{
{ @"FunctionName", $@"F{currentFuncNum}" },
{ @"DLCName", pmuDLCName }
});
sb = null;
return(false);
}
}
else
{
Log.Error($@"Skipping plot manager update: Conditional {currentFuncNum} is not a valid number for use. Values must be greater than 0 and less than 2 billion.");
Analytics.TrackEvent(@"Bad plot manager function", new Dictionary <string, string>()
{
{ @"FunctionName", $@"F{currentFuncNum}" },
{ @"DLCName", pmuDLCName }
});
sb = null;
return(false);
}
}
else
{
sb?.AppendLine(line);
}
}
// Add final, if any was found
if (sb != null)
{
funcMap[currentFuncNum] = sb.ToString();
Log.Information($@"PlotSync: Adding function {currentFuncNum} from {pmuDLCName}");
}
}
}
var pmPath = GetPlotManagerPath(target);
var vpm = Utilities.ExtractInternalFileToStream($@"MassEffectModManagerCore.modmanager.plotmanager.{target.Game}.PlotManager.{(target.Game == MEGame.ME1 ? @"u" : @"pcc")}"); // do not localize
if (funcMap.Any())
{
var plotManager = MEPackageHandler.OpenMEPackageFromStream(vpm, $@"PlotManager.{(target.Game == MEGame.ME1 ? @"u" : @"pcc")}"); // do not localize
var clonableFunction = plotManager.Exports.FirstOrDefault(x => x.ClassName == @"Function");
// STEP 1: ADD ALL NEW FUNCTIONS BEFORE WE INITIALIZE THE FILELIB.
foreach (var v in funcMap)
{
var pmKey = $@"BioAutoConditionals.F{v.Key}";
var exp = plotManager.FindExport(pmKey);
if (exp == null)
{
// Adding a new conditional
exp = EntryCloner.CloneEntry(clonableFunction);
exp.ObjectName = new NameReference($@"F{v.Key}", 0);
exp.FileRef.InvalidateLookupTable(); // We changed the name.
// Reduces trash
UFunction uf = ObjectBinary.From <UFunction>(exp);
uf.Children = 0;
uf.ScriptBytes = Array.Empty <byte>(); // No script data
exp.WriteBinary(uf);
Log.Information($@"Generated new blank conditional function export: {exp.UIndex} {exp.InstancedFullPath}", Settings.LogModInstallation);
}
}
// Relink child chain
UClass uc = ObjectBinary.From <UClass>(plotManager.FindExport(@"BioAutoConditionals"));
uc.UpdateChildrenChain();
uc.UpdateLocalFunctions();
uc.Export.WriteBinary(uc);
// STEP 2: UPDATE FUNCTIONS
Stopwatch sw = Stopwatch.StartNew();
var fl = new FileLib(plotManager);
bool initialized = fl.Initialize(new RelativePackageCache()
{
RootPath = M3Directories.GetBioGamePath(target)
}, target.TargetPath, canUseBinaryCache: false);
if (!initialized)
{
Log.Error(@"Error initializing FileLib for plot manager sync:");
foreach (var v in fl.InitializationLog.AllErrors)
{
Log.Error(v.Message);
}
throw new Exception(M3L.GetString(M3L.string_interp_fileLibInitFailedPlotManager, string.Join(Environment.NewLine, fl.InitializationLog.AllErrors.Select(x => x.Message)))); //force localize
}
sw.Stop();
Debug.WriteLine($@"Took {sw.ElapsedMilliseconds}ms to load filelib");
bool relinkChain = false;
foreach (var v in funcMap)
{
var pmKey = $@"BioAutoConditionals.F{v.Key}";
Log.Information($@"Updating conditional entry: {pmKey}", Settings.LogModInstallation);
var exp = plotManager.FindExport(pmKey);
(_, MessageLog log) = UnrealScriptCompiler.CompileFunction(exp, v.Value, fl);
if (log.AllErrors.Any())
{
Log.Error($@"Error compiling function {exp.InstancedFullPath}:");
foreach (var l in log.AllErrors)
{
Log.Error(l.Message);
}
throw new Exception(M3L.GetString(M3L.string_interp_errorCompilingFunctionReason, exp, string.Join('\n', log.AllErrors.Select(x => x.Message))));
}
}
if (plotManager.IsModified)
{
plotManager.Save(pmPath, true);
// Update local file DB
var bgfe = new BasegameFileIdentificationService.BasegameCloudDBFile(pmPath.Substring(target.TargetPath.Length + 1), (int)new FileInfo(pmPath).Length, target.Game, M3L.GetString(M3L.string_interp_plotManagerSyncForX, string.Join(@", ", combinedNames)), Utilities.CalculateMD5(pmPath));
BasegameFileIdentificationService.AddLocalBasegameIdentificationEntries(new List <BasegameFileIdentificationService.BasegameCloudDBFile>(new[] { bgfe }));
}
}
else
{
// Just write out vanilla.
vpm.WriteToFile(pmPath);
}
return(true);
}
