import sys
from pathlib import Path

import numpy as np
import pygame

from overcooked_simulator import ROOT_DIR
from overcooked_simulator.player import Player
from overcooked_simulator.pygame_gui.pygame_gui import PyGameGUI
from overcooked_simulator.simulation_runner import Simulator


def main():
    simulator = Simulator(Path(ROOT_DIR, "game_content/layouts", "basic.layout"), 600)
    player_one_name = "p1"
    player_two_name = "p2"
    simulator.register_player(Player(player_one_name, np.array([350.0, 200.0])))
    simulator.register_player(Player(player_two_name, np.array([100.0, 200.0])))

    # TODO maybe read the player names and keyboard keys from config file?
    keys1 = [
        pygame.K_LEFT,
        pygame.K_RIGHT,
        pygame.K_UP,
        pygame.K_DOWN,
        pygame.K_SPACE,
        pygame.K_i,
    ]
    keys2 = [pygame.K_a, pygame.K_d, pygame.K_w, pygame.K_s, pygame.K_f, pygame.K_e]
    gui = PyGameGUI(simulator, [player_one_name, player_two_name], [keys1, keys2])

    simulator.start()
    gui.start_pygame()

    simulator.stop()
    sys.exit()


if __name__ == "__main__":
    main()