5 files changed, 175 insertions, 2 deletions

@@ -29,6 +29,7 @@ namespace ManifoldGardenArchipelago
             }
             catch (Exception ex)
             {
+                Plugin.Logger.LogError(ex.GetBaseException().Message);
                 return new LoginFailure(ex.GetBaseException().Message);
             }
 
@@ -27,6 +27,7 @@ namespace ManifoldGardenArchipelago
         public readonly Dictionary<int, Requirement> doors = [];
         public readonly Dictionary<int, Requirement> smokeWalls = [];
         public readonly Dictionary<int, Requirement> worldGrows = [];
+        public readonly Dictionary<int, Requirement> lasers = [];
     }
 
     public class GameStateListeners
@@ -35,6 +36,7 @@ namespace ManifoldGardenArchipelago
         public readonly Dictionary<SceneItemReference, Requirement> doors = [];
         public readonly Dictionary<SceneItemReference, Requirement> smokeWalls = [];
         public readonly Dictionary<SceneItemReference, Requirement> worldGrows = [];
+        public readonly Dictionary<SceneItemReference, Requirement> lasers = [];
     }
 
     public static class GameData
@@ -419,6 +421,8 @@ namespace ManifoldGardenArchipelago
                         Requirement req = ParseRequirement(scenePair.Key, (Dictionary<object, object>)worldPair.Value);
                         sceneDescription.worldGrows[sir.index] = req;
 
+                        Plugin.Logger.LogInfo($"World {sir} requirements: {req}");
+
                         foreach (var reqScene in req.references.scenes)
                         {
                             GetOrAddListeners(listenersByScene, reqScene).worldGrows[sir] = req;
@@ -456,6 +460,53 @@ namespace ManifoldGardenArchipelago
                     }
                 }
 
+                if (sceneDetails.ContainsKey("lasers"))
+                {
+                    foreach (var laserPair in (Dictionary<object, object>)sceneDetails["lasers"])
+                    {
+                        SceneItemReference sir = new(scenePair.Key, int.Parse((string)laserPair.Key));
+                        Requirement req = ParseRequirement(scenePair.Key, (Dictionary<object, object>)laserPair.Value);
+                        sceneDescription.lasers[sir.index] = req;
+
+                        Plugin.Logger.LogInfo($"Laser {sir} requirements: {req}");
+
+                        foreach (var reqScene in req.references.scenes)
+                        {
+                            GetOrAddListeners(listenersByScene, reqScene).lasers[sir] = req;
+                        }
+
+                        foreach (var button in req.references.buttons)
+                        {
+                            GetOrAddListeners(listenersByButton, button).lasers[sir] = req;
+                        }
+
+                        foreach (var socket in req.references.sockets)
+                        {
+                            GetOrAddListeners(listenersBySocket, socket).lasers[sir] = req;
+                        }
+
+                        foreach (var pad in req.references.pads)
+                        {
+                            GetOrAddListeners(listenersByPad, pad).lasers[sir] = req;
+                        }
+
+                        foreach (var waterwheel in req.references.waterwheels)
+                        {
+                            GetOrAddListeners(listenersByWaterwheel, waterwheel).lasers[sir] = req;
+                        }
+
+                        foreach (var sphere in req.references.spheres)
+                        {
+                            GetOrAddListeners(listenersBySphere, sphere).lasers[sir] = req;
+                        }
+
+                        foreach (var item in req.references.items)
+                        {
+                            GetOrAddListeners(listenersByItem, item).lasers[sir] = req;
+                        }
+                    }
+                }
+
                 scenes[scenePair.Key] = sceneDescription;
             }
         }
@@ -33,6 +33,21 @@ namespace ManifoldGardenArchipelago
             throw new Exception("Shouldn't happen");
         }
 
+        public static SceneItemReference GetGameplayComponentSceneReference(Component component)
+        {
+            LevelSystems levelSystem = GetLevelSystems(component);
+
+            for (int i = 0; i < levelSystem.gameplayComponentsInLevel.Count(); i++)
+            {
+                if (levelSystem.gameplayComponentsInLevel[i] == component)
+                {
+                    return new(component.gameObject.scene.name, i);
+                }
+            }
+
+            throw new Exception("Shouldn't happen");
+        }
+
         public static SceneItemReference GetButtonSceneReference(ButtonLineActivator arg)
         {
             LevelSystems levelSystem = GetLevelSystems(arg);
@@ -191,6 +206,37 @@ namespace ManifoldGardenArchipelago
                     }
                 }
             }
+
+            foreach (var laser in listeners.lasers)
+            {
+                Requirement.Decision decision = laser.Value.Check();
+                if (decision == Requirement.Decision.Maybe)
+                {
+                    continue;
+                }
+
+                bool shouldOpen = (decision == Requirement.Decision.Yes);
+
+                if (SceneManager.GetSceneByName(laser.Key.scene) is Scene laserScene && laserScene.isLoaded)
+                {
+                    LevelSystems levelSystems = GetLevelSystems(laserScene);
+                    if (levelSystems.isActiveAndEnabled)
+                    {
+                        DarkModeCageController ldc = levelSystems.gameplayComponentsInLevel[laser.Key.index].GetComponent<DarkModeCageController>();
+                        
+                        if (shouldOpen)
+                        {
+                            MethodInfo methodInfo = typeof(DarkModeCageController).GetMethod("OpenLaserSource", BindingFlags.Instance | BindingFlags.NonPublic);
+                            methodInfo.Invoke(ldc, [false]);
+                        }
+                        else
+                        {
+                            MethodInfo methodInfo = typeof(DarkModeCageController).GetMethod("CloseLaserSource", BindingFlags.Instance | BindingFlags.NonPublic);
+                            methodInfo.Invoke(ldc, [false]);
+                        }
+                    }
+                }
+            }
         }
     }
 }
@@ -389,4 +389,52 @@ namespace ManifoldGardenArchipelago
             return false;
         }
     }
+
+    [HarmonyPatch(typeof(DarkModeCageController), "StartFilling")]
+    static class DarkModeCageControllerStartFillingPatch
+    {
+        static bool Prefix(DarkModeCageController __instance)
+        {
+            SceneItemReference sir = GameState.GetGameplayComponentSceneReference(__instance);
+
+            if (GameData.scenes.TryGetValue(sir.scene, out var sceneDescription))
+            {
+                if (sceneDescription.lasers.TryGetValue(sir.index, out var requirement))
+                {
+                    Requirement.Decision decision = requirement.Check();
+
+                    if (decision == Requirement.Decision.No)
+                    {
+                        return false;
+                    }
+                }
+            }
+
+            return true;
+        }
+    }
+
+    [HarmonyPatch(typeof(DarkModeCageController), "StartUnfilling")]
+    static class DarkModeCageControllerStartUnfillingPatch
+    {
+        static bool Prefix(DarkModeCageController __instance)
+        {
+            SceneItemReference sir = GameState.GetGameplayComponentSceneReference(__instance);
+
+            if (GameData.scenes.TryGetValue(sir.scene, out var sceneDescription))
+            {
+                if (sceneDescription.lasers.TryGetValue(sir.index, out var requirement))
+                {
+                    Requirement.Decision decision = requirement.Check();
+
+                    if (decision == Requirement.Decision.Yes)
+                    {
+                        return false;
+                    }
+                }
+            }
+
+            return true;
+        }
+    }
 }
@@ -75,7 +75,6 @@ Hallway_W062_W063_Optimized:
             scene: World_063_TowerX_Optimized
             index: 0
 World_001_Optimized:
-  # lasers are called DarkModeCageControllers; will handle them separately bc most worlds don't item-lock them
   locations:
     Blue - Red Laser Activated:
       socket: 5
@@ -85,6 +84,17 @@ World_001_Optimized:
       socket: 3
     Blue - Tree Purified:
       entry: AudioVisual_001_Optimized
+  lasers:
+    # These are gameplayComponentsInLevel indicies
+    127:
+      item: Blue - Green Laser
+      socket: 3
+    133:
+      item: Blue - Yellow Laser
+      socket: 4
+    149:
+      item: Blue - Red Laser
+      socket: 5
 World_044_CubicSpaceDivision_Optimized:
   locations:
     Blue God Cube Planted:
@@ -437,7 +447,7 @@ Hallway_W073_W018_Optimized:
 Hallway_W018_W063_Optimized:
   doors:
     0:
-      - button: [0, 1]
+      button: [0, 1]
     1:
       item: Orange Secret Path
       or:
@@ -758,3 +768,20 @@ World_806_Optimized:
   locations:
     Rainbow - Second Water Tetromino Puzzle Solved:
       waterwheel: 0
+World_071_AkshardhamTemple_Optimized:
+  lasers:
+    49:
+      waterwheel: 0
+      socket:
+        - scene: World_044_CubicSpaceDivision_Optimized
+          index: 0
+        - scene: World_044_CubicSpaceDivision_Optimized
+          index: 1
+        - scene: World_044_CubicSpaceDivision_Optimized
+          index: 2
+        - scene: World_044_CubicSpaceDivision_Optimized
+          index: 3
+        - scene: World_044_CubicSpaceDivision_Optimized
+          index: 4
+        - scene: World_044_CubicSpaceDivision_Optimized
+          index: 5