from bot import Bot
from game_recovery import GameRecovery
from screen import Screen
from logger import Logger
import keyboard
import os
from config import Config
from utils.graphic_debugger import run_graphic_debugger
from version import __version__
from utils.auto_settings import adjust_settings
from utils.misc import kill_thread, send_discord
import threading
from beautifultable import BeautifulTable
import time
import logging
import cv2
import traceback


def run_bot(config: Config, screen: Screen, game_recovery: GameRecovery, pick_corpose_on_start: bool = False):
    bot = Bot(screen, pick_corpose_on_start)
    bot_thread = threading.Thread(target=bot.start)
    bot_thread.start()
    do_restart = False
    keyboard.add_hotkey(config.general["exit_key"], lambda: Logger.info(f'Force Exit') or os._exit(1))
    keyboard.add_hotkey(config.general['resume_key'], lambda: bot.toggle_pause())
    while 1:
        if bot.current_game_length() > config.general["max_game_length_s"]:
            Logger.info(f"Max game length reached. Attempting to restart {config.general['name']}!")
            if config.general["info_screenshots"]:
                cv2.imwrite("./info_screenshots/info_max_game_length_reached_" + time.strftime("%Y%m%d_%H%M%S") + ".png", bot._screen.grab())
            bot.stop()
            kill_thread(bot_thread)
            do_restart = game_recovery.go_to_hero_selection()
            break
        time.sleep(0.5)
    bot_thread.join()
    if do_restart:
        run_bot(config, screen, game_recovery, True)
    else:
        if config.general["info_screenshots"]:
            cv2.imwrite("./info_screenshots/info_could_not_recover_" + time.strftime("%Y%m%d_%H%M%S") + ".png", bot._screen.grab())
        Logger.error(f"{config.general['name']} could not recover from a max game length violation. Shutting down everything.")
        if config.general["custom_discord_hook"]:
            send_discord(f"{config.general['name']} got stuck and can not resume", config.general["custom_discord_hook"])
        os._exit(1)

def main():
    config = Config(print_warnings=True)
    if config.general["logg_lvl"] == "info":
        Logger.init(logging.INFO)
    elif config.general["logg_lvl"] == "debug":
        Logger.init(logging.DEBUG)
    else:
        print(f"ERROR: Unkown logg_lvl {config.general['logg_lvl']}. Must be one of [info, debug]")

    # create folder if they dont exist yet
    if not os.path.exists("info_screenshots"):
        os.system("mkdir info_screenshots")
    if not os.path.exists("loot_screenshots"):
        os.system("mkdir loot_screenshots")

    # If anything seems to go wrong, press f12 and the bot will force exit
    keyboard.add_hotkey(config.general["exit_key"], lambda: Logger.info(f'Force Exit') or os._exit(1))

    print(f"============ Botty {__version__} [name: {config.general['name']}] ============")
    print("\nFor gettings started and documentation\nplease read https://github.com/aeon0/botty\n")
    table = BeautifulTable()
    table.rows.append([config.general['auto_settings_key'], "Adjust D2R settings"])
    table.rows.append([config.general['graphic_debugger_key'], "Graphic debugger"])
    table.rows.append([config.general['resume_key'], "Start / Pause Botty"])
    table.rows.append([config.general['exit_key'], "Stop bot"])
    table.columns.header = ["hotkey", "action"]
    print(table)
    print("\n")

    while 1:
        if keyboard.is_pressed(config.general['resume_key']):
            screen = Screen(config.general["monitor"])
            game_recovery = GameRecovery(screen)
            run_bot(config, screen, game_recovery)
            break
        if keyboard.is_pressed(config.general['auto_settings_key']):
            adjust_settings()
            break
        elif keyboard.is_pressed(config.general['graphic_debugger_key']):
            run_graphic_debugger()
            break
        time.sleep(0.02)


if __name__ == "__main__":
    # To avoid cmd just closing down, except any errors and add a input() to the end
    try:
        main()
    except:
        print("RUNTIME ERROR:")
        traceback.print_exc()
    print("Press Enter to exit ...")
    input()