#!/usr/bin/env python3
import os
import sys
import dcs
import logging

import theater.caucasus
import theater.persiangulf
import theater.nevada

import ui.window
import ui.mainmenu
import ui.newgamemenu
import ui.corruptedsavemenu

from game.game import Game
from theater import start_generator
from userdata import persistency, logging as logging_module

persistency.setup(sys.argv[1])
dcs.planes.FlyingType.payload_dirs = [os.path.join(os.path.dirname(os.path.realpath(__file__)), "resources\\payloads")]

logging_module.setup_version_string(sys.argv[2])
logging.info("Using {} as userdata folder".format(persistency.base_path()))


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


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

        def start_new_game(player_name: str, enemy_name: str, terrain: str, sams: bool, midgame: bool, multiplier: float):
            if terrain == "persiangulf":
                conflicttheater = theater.persiangulf.PersianGulfTheater()
            elif terrain == "nevada":
                conflicttheater = theater.nevada.NevadaTheater()
            else:
                conflicttheater = theater.caucasus.CaucasusTheater()

            if midgame:
                for i in range(0, int(len(conflicttheater.controlpoints) / 2)):
                    conflicttheater.controlpoints[i].captured = True

            start_generator.generate_initial(conflicttheater, enemy_name, sams, multiplier)
            game = Game(player_name=player_name,
                        enemy_name=enemy_name,
                        theater=conflicttheater)
            game.budget = int(game.budget * multiplier)
            game.settings.multiplier = multiplier
            game.settings.sams = sams

            if midgame:
                game.budget = game.budget * 4 * len(list(conflicttheater.conflicts()))

            proceed_to_main_menu(game)

        new_game_menu = ui.newgamemenu.NewGameMenu(w, start_new_game)
        new_game_menu.display()
    else:
        proceed_to_main_menu(game)
except Exception as e:
    print(e)
    ui.corruptedsavemenu.CorruptedSaveMenu(w).display()

w.run()