#!/usr/bin/env python3
import sys

import theater.caucasus
import ui.window
import ui.mainmenu
import ui.newgamemenu

from game.game import Game
from theater import start_generator
from userdata import persistency


def proceed_to_main_menu(game: Game):
    m = ui.mainmenu.MainMenu(w, None, game)
    m.display()


w = ui.window.Window()
game = persistency.restore_game()
if not game:
    new_game_menu = None  # type: NewGameMenu

    def start_new_game(player_name: str, enemy_name: str):
        conflicttheater = theater.caucasus.CaucasusTheater()
        start_generator.generate_initial(conflicttheater, enemy_name)

        proceed_to_main_menu(Game(player_name=player_name,
                                  enemy_name=enemy_name,
                                  theater=conflicttheater))

    new_game_menu = ui.newgamemenu.NewGameMenu(w, start_new_game)
    new_game_menu.display()
else:
    proceed_to_main_menu(game)

w.run()