squirrel-battle/dungeonbattle/menus.py

import sys
from enum import Enum
from typing import Any, Optional

from .enums import GameMode, KeyValues
from .settings import Settings


class Menu:
    values: list

    def __init__(self):
        self.position = 0

    def go_up(self) -> None:
        self.position = max(0, self.position - 1)

    def go_down(self) -> None:
        self.position = min(len(self.values) - 1, self.position + 1)

    def validate(self) -> Any:
        return self.values[self.position]


class MainMenuValues(Enum):
    START = 'Jouer'
    SETTINGS = 'Paramètres'
    EXIT = 'Quitter'

    def __str__(self):
        return self.value


class MainMenu(Menu):
    values = [e for e in MainMenuValues]

    def handle_key_pressed(self, key: KeyValues, game: Any) -> None:
        """
        In the main menu, we can navigate through options.
        """
        if key == KeyValues.DOWN:
            self.go_down()
        if key == KeyValues.UP:
            self.go_up()
        if key == KeyValues.ENTER:
            option = self.validate()
            if option == MainMenuValues.START:
                game.state = GameMode.PLAY
            elif option == MainMenuValues.SETTINGS:
                game.state = GameMode.SETTINGS
            elif option == MainMenuValues.EXIT:
                sys.exit(0)


class SettingsMenu(Menu):
    waiting_for_key: bool = False

    def update_values(self, settings: Settings) -> None:
        s = settings.dumps_to_string()
        self.values = s[6:-2].replace("\"", "").split(",\n    ")

    def handle_key_pressed(self, key: Optional[KeyValues], raw_key: str,
                           game: Any) -> None:
        """
        For now, in the settings mode, we can only go backwards.
        """
        if not self.waiting_for_key:
            if key == KeyValues.SPACE:
                game.state = GameMode.MAINMENU
            if key == KeyValues.DOWN:
                self.go_down()
            if key == KeyValues.UP:
                self.go_up()
            if key == KeyValues.ENTER:
                option = self.validate().split(": ")[0]
                if option != "TEXTURE_PACK":
                    self.waiting_for_key = True
        else:
            option = self.validate().split(": ")[0]
            setattr(game.settings, option, raw_key)
            game.settings.write_settings()
            self.update_values(game.settings)
            self.waiting_for_key = False


class ArbitraryMenu(Menu):
    def __init__(self, values: list):
        super().__init__()
        self.values = values