add one-way energy link by hitting the mushroom with the hammer

worlds/zork_grand_inquisitor/client.py

@@ -285,6 +285,24 @@ async def controller(self):
                         }
                     ])
 
+                # Handle Energy Link
+                while len(self.game_controller.energy_link_queue) > 0:
+                    energy_to_add: int = self.game_controller.energy_link_queue.popleft()
+
+                    await self.send_msgs([
+                        {
+                            "cmd": "Set",
+                            "key": f"EnergyLink{self.team}",
+                            "slot": self.slot,
+                            "operations":
+                                [
+                                    {"operation": "add", "value": energy_to_add},
+                                ],
+                        },
+                    ])
+
+                    CommonClient.logger.info(f"Added {energy_to_add} J to the Energy Link pool")
+
                 # Handle Death Link
                 await self.update_death_link(self.death_link_status)
 

worlds/zork_grand_inquisitor/game_controller.py

@@ -92,6 +92,9 @@ class GameController:
     active_trap: Optional[ZorkGrandInquisitorItems]
     active_trap_until: Optional[datetime.datetime]
 
+    energy_link_queue: collections.deque
+    pause_energy_link_monitoring: bool
+
     pending_death_link: Tuple[bool, Optional[str], Optional[str]]
     outgoing_death_link: Tuple[bool, Optional[str]]
     pause_death_monitoring: bool
@@ -171,6 +174,9 @@ def __init__(self, logger=None) -> None:
         self.active_trap = None
         self.active_trap_until = None
 
+        self.energy_link_queue = collections.deque()
+        self.pause_energy_link_monitoring = False
+
         self.pending_death_link = (False, None, None)
         self.outgoing_death_link = (False, None)
         self.pause_death_monitoring = False
@@ -422,6 +428,9 @@ def update(self) -> None:
                 if self.option_trap_percentage:
                     self._manage_traps()
 
+                if self._player_is_at("dg3e"):
+                    self._manage_energy_link()
+
                 if self.option_death_link:
                     self._handle_death_link()
 
@@ -472,6 +481,9 @@ def reset(self) -> None:
         self.active_trap = None
         self.active_trap_until = None
 
+        self.energy_link_queue = collections.deque()
+        self.pause_energy_link_monitoring = False
+
         self.pending_death_link = (False, None, None)
         self.outgoing_death_link = (False, None)
         self.pause_death_monitoring = False
@@ -1485,6 +1497,36 @@ def _activate_trap_zvision(self) -> None:
     def _deactivate_trap_zvision(self) -> None:
         self.game_state_manager.set_zvision(False)
 
+    def _manage_energy_link(self) -> None:
+        mushroom_hammered: bool = self._read_game_state_value_for(4217) == 1
+        mushroom_hammered_throck: bool = self._read_game_state_value_for(4219) == 1
+        mushroom_hammered_snapdragon: bool = self._read_game_state_value_for(4220) == 1
+        mushroom_hammered_snapdragon_throck: bool = self._read_game_state_value_for(4222) == 1
+
+        any_mushroom_hammered: bool = (
+            mushroom_hammered
+            or mushroom_hammered_throck
+            or mushroom_hammered_snapdragon
+            or mushroom_hammered_snapdragon_throck
+        )
+
+        # Pause Monitoring Flag
+        if self.pause_energy_link_monitoring and not any_mushroom_hammered:
+            self.pause_energy_link_monitoring = False
+
+        if not self.pause_energy_link_monitoring and any_mushroom_hammered:
+            # Contribute Energy
+            if mushroom_hammered:
+                self.energy_link_queue.append(750)
+            elif mushroom_hammered_throck:
+                self.energy_link_queue.append(1500)
+            elif mushroom_hammered_snapdragon:
+                self.energy_link_queue.append(750)
+            elif mushroom_hammered_snapdragon_throck:
+                self.energy_link_queue.append(1500)
+
+            self.pause_energy_link_monitoring = True
+
     def _handle_death_link(self) -> None:
         # Pause Monitoring Flag
         if self.pause_death_monitoring and not self._player_is_at("gjde"):