Populate ComputerLookup on add to netMap list

Arkhist · Jul 22, 2021 · 88ba02b · 88ba02b
1 parent 3b3d745
commit 88ba02b
Showing 1 changed file with 8 additions and 17 deletions.
diff --git a/PathfinderAPI/BaseGameFixes/Performance/NodeLookup.cs b/PathfinderAPI/BaseGameFixes/Performance/NodeLookup.cs
@@ -1,4 +1,5 @@
-using Hacknet;
+using System.Collections.Generic;
+using Hacknet;
 using HarmonyLib;
 using Pathfinder.Event;
 using Pathfinder.Event.Loading;
@@ -9,24 +10,14 @@ namespace Pathfinder.BaseGameFixes.Performance
     [HarmonyPatch]
     internal static class NodeLookup
     {
-        [Util.Initialize]
-        internal static void Initialize()
-        {
-            EventManager<SaveComputerLoadedEvent>.AddHandler(PopulateOnComputerLoad);
-        }
-
         [HarmonyPostfix]
-        [HarmonyPatch(typeof(ComputerLoader), nameof(ComputerLoader.loadComputer))]
-        internal static void PopulateOnComputerCreation(object __result)
-        {
-            if (__result == null)
-                return;
-            ComputerLookup.PopulateLookups((Computer) __result);
-        }
-
-        internal static void PopulateOnComputerLoad(SaveComputerLoadedEvent args)
+        [HarmonyPatch(typeof(List<Computer>), nameof(List<Computer>.Add))]
+        internal static void AddComputerReference(List<Computer> __instance, Computer item)
         {
-            ComputerLookup.PopulateLookups(args.Comp);
+            if (object.ReferenceEquals(__instance, OS.currentInstance?.netMap?.nodes))
+            {
+                ComputerLookup.PopulateLookups(item);
+            }
         }
 
         [HarmonyPrefix]