internal byte[] SearchInAllPaksForPath(string searchingPath, PakExtractor fullExtractor, bool checkMainPakFirst = true)
{
if (checkMainPakFirst && fullExtractor.HasPath(searchingPath))
{
return(fullExtractor.ReadRaw(searchingPath));
}
if (SearchLookup.ContainsKey(searchingPath))
{
try
{
using (FileStream f = new FileStream(SearchLookup[searchingPath], FileMode.Open, FileAccess.Read))
{
try
{
PakExtractor modPakExtractor = new PakExtractor(new BinaryReader(f));
if (modPakExtractor.HasPath(searchingPath))
{
return(modPakExtractor.ReadRaw(searchingPath));
}
}
catch { }
}
}
catch (IOException)
{
}
}
if (!checkMainPakFirst && fullExtractor.HasPath(searchingPath))
{
return(fullExtractor.ReadRaw(searchingPath));
}
return(null);
}
internal byte[] FindFile(string target, PakExtractor ourExtractor)
{
if (CreatedPakData.ContainsKey(target))
{
return(CreatedPakData[target]);
}
return(SearchInAllPaksForPath(target, ourExtractor, false));
}
private readonly CategoryReference refData2B; // SceneComponent
public LevelBaker(PakExtractor extractor, ModIntegrator integrator)
{
Extractor = extractor;
ParentIntegrator = integrator;
AssetReader y = new AssetReader(new BinaryReader(new MemoryStream(Properties.Resources.LevelTemplate)));
refData1B = y.categories[2].ReferenceData;
refData2B = y.categories[11].ReferenceData;
}
public ModIntegrator()
{
OptionalModIDs = new List <string>();
// Include static assets
PakExtractor staticAssetsExtractor = new PakExtractor(new BinaryReader(new MemoryStream(Properties.Resources.IntegratorStaticAssets)));
foreach (KeyValuePair <string, long> entry in staticAssetsExtractor.PathToOffset)
{
StarterPakData[entry.Key] = staticAssetsExtractor.ReadRaw(entry.Value);
}
}
internal Dictionary <string, string> SearchLookup; // file to path --> pak you can find it in
internal void InitializeSearch(string installPath)
{
SearchLookup = new Dictionary <string, string>();
string[] realPakPaths = Directory.GetFiles(installPath, "*_P.pak", SearchOption.TopDirectoryOnly);
foreach (string realPakPath in realPakPaths)
{
using (FileStream f = new FileStream(realPakPath, FileMode.Open, FileAccess.Read))
{
try
{
PakExtractor ourExtractor = new PakExtractor(new BinaryReader(f));
Metadata us = null;
try
{
us = ourExtractor.ReadMetadata();
}
catch { }
if (us == null || IntegratorUtils.IgnoredModIDs.Contains(us.ModID))
{
continue;
}
foreach (KeyValuePair <string, long> entry in ourExtractor.PathToOffset)
{
SearchLookup[entry.Key] = realPakPath;
}
}
catch
{
continue;
}
}
}
}
public void IntegrateMods(string paksPath, string installPath) // @"C:\Users\<CLIENT USERNAME>\AppData\Local\Astro\Saved\Paks", @"C:\Program Files (x86)\Steam\steamapps\common\ASTRONEER\Astro\Content\Paks"
{
Directory.CreateDirectory(paksPath);
string[] files = Directory.GetFiles(paksPath, "*_P.pak", SearchOption.TopDirectoryOnly);
InitializeSearch(paksPath);
int modCount = 0;
Dictionary <string, List <string> > newComponents = new Dictionary <string, List <string> >();
Dictionary <string, Dictionary <string, List <string> > > newItems = new Dictionary <string, Dictionary <string, List <string> > >();
List <string> newPersistentActors = new List <string>();
List <string> newTrailheads = new List <string>();
List <Metadata> allMods = new List <Metadata>();
foreach (string file in files)
{
using (FileStream f = new FileStream(file, FileMode.Open, FileAccess.Read))
{
Metadata us = null;
try
{
us = new PakExtractor(new BinaryReader(f)).ReadMetadata();
}
catch
{
continue;
}
if (us == null || IntegratorUtils.IgnoredModIDs.Contains(us.ModID))
{
continue;
}
modCount++;
allMods.Add(us);
Dictionary <string, List <string> > theseComponents = us.LinkedActorComponents;
if (theseComponents != null)
{
foreach (KeyValuePair <string, List <string> > entry in theseComponents)
{
if (newComponents.ContainsKey(entry.Key))
{
newComponents[entry.Key].AddRange(entry.Value);
}
else
{
newComponents.Add(entry.Key, entry.Value);
}
}
}
Dictionary <string, Dictionary <string, List <string> > > theseItems = us.ItemListEntries;
if (theseItems != null)
{
foreach (KeyValuePair <string, Dictionary <string, List <string> > > entry in theseItems)
{
if (newItems.ContainsKey(entry.Key))
{
foreach (KeyValuePair <string, List <string> > entry2 in entry.Value)
{
if (newItems[entry.Key].ContainsKey(entry2.Key))
{
newItems[entry.Key][entry2.Key].AddRange(entry2.Value);
}
else
{
newItems[entry.Key].Add(entry2.Key, entry2.Value);
}
}
}
else
{
newItems.Add(entry.Key, entry.Value);
}
}
}
List <string> thesePersistentActors = us.PersistentActors;
if (thesePersistentActors != null)
{
newPersistentActors.AddRange(thesePersistentActors);
}
List <string> theseTrailheads = us.MissionTrailheads;
if (theseTrailheads != null)
{
newTrailheads.AddRange(theseTrailheads);
}
}
}
CreatedPakData = new Dictionary <string, byte[]>
{
{ "metadata.json", StarterPakData["metadata.json"] }
};
if (modCount > 0)
{
// Apply static files
CreatedPakData = StarterPakData.ToDictionary(entry => entry.Key, entry => (byte[])entry.Value.Clone());
if (!newComponents.ContainsKey("/Game/Globals/PlayControllerInstance"))
{
newComponents.Add("/Game/Globals/PlayControllerInstance", new List <string>());
}
newComponents["/Game/Globals/PlayControllerInstance"].Add("/Game/Integrator/ServerModComponent");
// Generate mods data table
var dtb = new DataTableBaker(this);
CreatedPakData["Astro/Content/Integrator/ListOfMods.uasset"] = dtb.Bake(allMods.ToArray(), OptionalModIDs, CreatedPakData["Astro/Content/Integrator/ListOfMods.uasset"]).ToArray();
CreatedPakData["Astro/Content/Integrator/IntegratorStatics.uasset"] = dtb.Bake2(CreatedPakData["Astro/Content/Integrator/IntegratorStatics.uasset"]).ToArray();
}
string[] realPakPaths = Directory.GetFiles(installPath, "*.pak", SearchOption.TopDirectoryOnly);
foreach (string realPakPath in realPakPaths)
{
using (FileStream f = new FileStream(realPakPath, FileMode.Open, FileAccess.Read))
{
PakExtractor ourExtractor;
try
{
ourExtractor = new PakExtractor(new BinaryReader(f));
}
catch
{
continue;
}
var actorBaker = new ActorBaker();
var itemListBaker = new ItemListBaker();
var levelBaker = new LevelBaker(ourExtractor, this);
// Patch level for persistent actors and missions
if (newPersistentActors.Count > 0 || newTrailheads.Count > 0)
{
foreach (string mapPath in MapPaths)
{
byte[] mapPathData = FindFile(mapPath, ourExtractor);
if (mapPathData != null)
{
CreatedPakData[mapPath] = levelBaker.Bake(newPersistentActors.ToArray(), newTrailheads.ToArray(), mapPathData).ToArray();
}
}
}
// Add components
foreach (KeyValuePair <string, List <string> > entry in newComponents)
{
string establishedPath = entry.Key.ConvertGamePathToAbsolutePath();
byte[] actorData = FindFile(establishedPath, ourExtractor);
if (actorData == null)
{
continue;
}
try
{
CreatedPakData[establishedPath] = actorBaker.Bake(entry.Value.ToArray(), actorData).ToArray();
}
catch (Exception ex)
{
Debug.WriteLine(ex.ToString());
}
}
// Add new item entries
foreach (KeyValuePair <string, Dictionary <string, List <string> > > entry in newItems)
{
string establishedPath = entry.Key.ConvertGamePathToAbsolutePath();
byte[] actorData = FindFile(establishedPath, ourExtractor);
if (actorData == null)
{
continue;
}
try
{
CreatedPakData[establishedPath] = itemListBaker.Bake(entry.Value, actorData).ToArray();
}
catch (Exception ex)
{
Debug.WriteLine(ex.ToString());
}
}
}
}
byte[] pakData = PakBaker.Bake(CreatedPakData);
using (FileStream f = new FileStream(Path.Combine(paksPath, @"999-AstroModIntegrator_P.pak"), FileMode.Create, FileAccess.Write))
{
f.Write(pakData, 0, pakData.Length);
}
}