Source code for mastermind.main.game_controller

from typing import Any

from mastermind.game.game import Game
from mastermind.main.game_history import GameHistoryManager
from mastermind.storage import userdata
from mastermind.validation import (
    MaximumAttempts,
    NumberOfColors,
    NumberOfDots,
    ValidationError,
    Validator,
)




class GameController:
    """Communication between MainUI and the gameboard."""



    @classmethod
    def validate_input(cls, prompt: str, validator: Validator) -> Any:
        """Get user input and validate it."""
        while True:
            try:
                user_input = input("\n" + prompt)
                return validator().validate_value(user_input)

            except ValidationError as e:
                print(f"Invalid input. {e}")




    @classmethod
    def get_game_parameters(cls) -> tuple[int, int, int]:
        """Get the number of colors, number of dots, and number of rounds from user."""
        num_of_colors = cls.validate_input(
            "Enter the number of colors (2-10): ", NumberOfColors
        )
        num_of_dots = cls.validate_input(
            "Enter the number of dots (2-10): ", NumberOfDots
        )
        max_attempts = cls.validate_input(
            "Enter the maximum number of attempts: ", MaximumAttempts
        )
        return num_of_colors, num_of_dots, max_attempts




    @classmethod
    def start_new_game(cls, game_mode: str) -> None:
        """Start a new game."""
        if game_mode not in ["HvH", "HvAI", "AIvH", "AIvAI"]:
            raise AssertionError("Unexpected invalid game mode.")

        parameters = cls.get_game_parameters()  # get user input
        game = Game(*parameters, game_mode)  # create a new game
        exit_state = game.start_game()  # start the game and retrieve exit state

        if exit_state != "d":  # "d" is discard, do not save
            GameHistoryManager.save_game(game)  # save the game




    @classmethod
    def resume_game(cls, game_index: int) -> None:
        """Resume a saved game."""
        # Retrieve game
        game = userdata.saved_games[game_index]["game"]

        # Resume game and retrieve exit state
        exit_state = game.resume_game()

        # Update saved games if not game discarded
        if exit_state == "d":  # or delete it if discarded
            userdata.saved_games.pop(game_index)
        else:
            userdata.saved_games[game_index] = GameHistoryManager().generate_meta_data(
                game
            )