#!/usr/bin/env python3

""" Entry point to play the game """

import argparse

import pygame

from .controller import Controller, KeyboardController, JoystickController
from .screens.base_screen import Screen
from .screens.in_game import InGame as InGameScreen, Player
from .screens.pause import Pause as PauseScreen
from .misc import Pause, UnPause


def main() -> None:
    """ Main entry point. """
    parser = argparse.ArgumentParser()

    parser.add_argument("--starting-level", type=int, choices=list(range(1, 30)), default=1)
    parser.add_argument("--joystick", type=int, dest="joystick_id", metavar="Joystick ID")

    args = parser.parse_args()

    pygame.init()

    screen = pygame.display.set_mode((801, 1000))

    controller: Controller
    if args.joystick_id is not None:
        joystick = pygame.joystick.Joystick(args.joystick_id)
        joystick.init()
        controller = JoystickController(joystick)
    else:
        controller = KeyboardController(pygame.key)

    # Just one player
    players = [Player(controller, args.starting_level)]

    for player in players:
        player.refresh_piece_preview_canvas()

    clock = pygame.time.Clock()

    current_screen: Screen
    current_screen = ingame_screen = InGameScreen(players, screen)

    while True:
        try:
            current_screen.oneframe()
        except Pause:
            current_screen = PauseScreen(screen)
        except UnPause:
            current_screen = ingame_screen

        current_screen.refresh()
        pygame.display.update()

        clock.tick(60)


if __name__ == "__main__":
    main()