feat: Add Pygame menu

3 files changed, 130 insertions, 0 deletions

diff --git a/src/splitscreen_duo/__init__.py b/src/splitscreen_duo/__init__.py
index eea3683..4be2b6f 100644
--- a/src/splitscreen_duo/__init__.py
+++ b/src/splitscreen_duo/__init__.py
@@ -1,4 +1,19 @@
+from . import menu
+import os
+import logging
+
+
+logger = logging.getLogger(__name__)
+
+
 def main() -> int:
+    logging.basicConfig(
+        level=(
+            logging.DEBUG if os.environ.get("DEVELOPMENT", "").lower() else logging.INFO
+        )
+    )
     print("The Dual Screen Console")
+    logger.info(f"Running as {os.environ.get("INSTANCE", "unknown")}")
+    menu.main_loop()
 
     return 0
diff --git a/src/splitscreen_duo/input.py b/src/splitscreen_duo/input.py
new file mode 100644
index 0000000..1a1e5d0
--- /dev/null
+++ b/src/splitscreen_duo/input.py
@@ -0,0 +1,51 @@
+import pygame
+
+
+class Input:
+    def __init__(self, debug=True):
+        self.debug = debug
+        self.controller = None
+
+        if not self.debug:
+            pygame.joystick.init()
+
+            if pygame.joystick.get_count() > 0:
+                self.controller = pygame.joystick.Joystick(0)
+                self.controller.init()
+
+    def get_input(self, event):
+        if self.debug:
+            return self.get_keyboard_input(event)
+        else:
+            return self.get_controller_input(event)
+
+    def get_keyboard_input(self, event):
+        if event.type == pygame.KEYDOWN:
+            if event.key == pygame.K_UP:
+                return "UP"
+            if event.key == pygame.K_DOWN:
+                return "DOWN"
+            if event.key == pygame.K_RETURN:
+                return "SELECT"
+            if event.key == pygame.K_ESCAPE:
+                return "QUIT"
+
+        return None
+
+    def get_controller_input(self, event):
+        if not self.controller:
+            return None
+
+        if event.type == pygame.JOYAXISMOTION:
+            if event.axis == 1:  # Y-axis
+                if event.value < -0.5:  # Joystick up
+                    return "UP"
+                if event.value > 0.5:  # Joystick down
+                    return "DOWN"
+        elif event.type == pygame.JOYBUTTONDOWN:
+            if event.button == 0:  # Button A or select
+                return "SELECT"
+            if event.button == 1:  # Button B
+                return "QUIT"
+
+        return None
diff --git a/src/splitscreen_duo/menu.py b/src/splitscreen_duo/menu.py
new file mode 100644
index 0000000..489a75c
--- /dev/null
+++ b/src/splitscreen_duo/menu.py
@@ -0,0 +1,64 @@
+import sys
+import pygame
+import os
+from .input import Input
+
+pygame.init()
+pygame.display.set_caption("SplitScreen Duo Menu")
+
+WIDTH, HEIGHT = (
+    pygame.display.Info().current_w // 2,
+    pygame.display.Info().current_h // 2,
+)
+WHITE = (255, 255, 255)
+BLACK = (0, 0, 0)
+FONT = pygame.font.Font(None, 36)
+IS_DEVELOPMENT_MODE = os.environ.get("DEVELOPMENT", "").lower()
+OPTIONS = ["Breakout", "Pong v.s. Computer", "Snake", "Quit"]
+selected_index = 0
+screen = pygame.display.set_mode(
+    (WIDTH, HEIGHT) if IS_DEVELOPMENT_MODE else (0, 0),
+    0 if IS_DEVELOPMENT_MODE else pygame.FULLSCREEN,
+)
+
+
+def draw_menu():
+    screen.fill(WHITE)
+
+    for i, option in enumerate(OPTIONS):
+        text = FONT.render(option, True, BLACK if i != selected_index else (200, 0, 0))
+
+        screen.blit(text, (WIDTH // 2 - text.get_width() // 2, 150 + i * 50))
+
+    pygame.display.flip()
+
+
+def main_loop():
+    global selected_index
+
+    is_running = True
+    input_handler = Input(debug=IS_DEVELOPMENT_MODE)
+
+    while is_running:
+        draw_menu()
+
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                pygame.quit()
+                sys.exit()
+
+            action = input_handler.get_input(event)
+
+            if action == "UP":
+                selected_index = (selected_index - 1) % len(OPTIONS)
+            elif action == "DOWN":
+                selected_index = (selected_index + 1) % len(OPTIONS)
+            elif action == "SELECT":
+                if OPTIONS[selected_index] == "Quit":
+                    pygame.quit()
+                    sys.exit()
+                else:
+                    print(OPTIONS[selected_index])
+            elif action == "QUIT":
+                pygame.quit()
+                sys.exit()