Added chests, they are immortal and contain objects the player can take for free.

squirrelbattle/display/menudisplay.py

@@ -5,7 +5,8 @@ import curses
 from random import randint
 from typing import List
 
-from squirrelbattle.menus import Menu, MainMenu, SettingsMenu, StoreMenu
+from squirrelbattle.menus import Menu, MainMenu, SettingsMenu, StoreMenu,\
+     ChestMenu
 from .display import Box, Display
 from ..entities.player import Player
 from ..enums import KeyValues, GameMode
@@ -156,13 +157,16 @@ class PlayerInventoryDisplay(MenuDisplay):
     player: Player = None
     selected: bool = True
     store_mode: bool = False
+    chest_mode: bool = False
 
     def update(self, game: Game) -> None:
         self.player = game.player
         self.update_menu(game.inventory_menu)
         self.store_mode = game.state == GameMode.STORE
+        self.chest_mode = game.state == GameMode.CHEST
         self.selected = game.state == GameMode.INVENTORY \
-            or (self.store_mode and not game.is_in_store_menu)
+            or (self.store_mode and not game.is_in_store_menu)\
+            or (self.chest_mode and not game.is_in_chest_menu)
 
     def update_pad(self) -> None:
         self.menubox.update_title(_("INVENTORY"))
@@ -239,3 +243,39 @@ class StoreInventoryDisplay(MenuDisplay):
         self.menu.position = max(0, min(len(self.menu.values) - 1, y - 2))
         game.is_in_store_menu = True
         game.handle_key_pressed(KeyValues.ENTER)
+
+class ChestInventoryDisplay(MenuDisplay):
+    """
+    A class to handle the display of a merchant's inventory.
+    """
+    menu: ChestMenu
+    selected: bool = False
+
+    def update(self, game: Game) -> None:
+        self.update_menu(game.chest_menu)
+        self.selected = game.is_in_chest_menu
+
+    def update_pad(self) -> None:
+        self.menubox.update_title(_("CHEST"))
+        for i, item in enumerate(self.menu.values):
+            rep = self.pack[item.name.upper()]
+            selection = f"[{rep}]" if i == self.menu.position \
+                and self.selected else f" {rep} "
+            self.addstr(self.pad, i + 1, 0, selection
+                        + " " + item.translated_name.capitalize())
+
+    @property
+    def truewidth(self) -> int:
+        return max(1, self.height if hasattr(self, "height") else 10)
+
+    @property
+    def trueheight(self) -> int:
+        return 2 + super().trueheight
+
+    def handle_click(self, y: int, x: int, game: Game) -> None:
+        """
+        We can select a menu item with the mouse.
+        """
+        self.menu.position = max(0, min(len(self.menu.values) - 1, y - 2))
+        game.is_in_chest_menu = True
+        game.handle_key_pressed(KeyValues.ENTER)