From cdd527a7fd47e6b398b69cd717410c553b61474d Mon Sep 17 00:00:00 2001
From: Yohann D'ANELLO <yohann.danello@gmail.com>
Date: Fri, 4 Dec 2020 14:57:53 +0100
Subject: [PATCH] Close the inventory using the same key

---
 squirrelbattle/game.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/squirrelbattle/game.py b/squirrelbattle/game.py
index d6045ce..b4b3147 100644
--- a/squirrelbattle/game.py
+++ b/squirrelbattle/game.py
@@ -83,6 +83,8 @@ class Game:
 
         if self.state == GameMode.PLAY:
             self.handle_key_pressed_play(key)
+        elif self.state == GameMode.INVENTORY:
+            self.handle_key_pressed_inventory(key)
         elif self.state == GameMode.MAINMENU:
             self.handle_key_pressed_main_menu(key)
         elif self.state == GameMode.SETTINGS:
@@ -110,6 +112,13 @@ class Game:
         elif key == KeyValues.SPACE:
             self.state = GameMode.MAINMENU
 
+    def handle_key_pressed_inventory(self, key: KeyValues) -> None:
+        """
+        In the inventory menu, we can interact with items or close the menu.
+        """
+        if key == KeyValues.SPACE or key == KeyValues.INVENTORY:
+            self.state = GameMode.PLAY
+
     def handle_key_pressed_main_menu(self, key: KeyValues) -> None:
         """
         In the main menu, we can navigate through options.