import argparse
import sys
from pathlib import Path

from source.interface import is_valid_source_path, is_valid_destination_path, get_finished_installation_message
from source.mkw.Game import Game
from source.mkw.ModConfig import ModConfig
from source.mkw.collection.Extension import Extension
from source.translation import translate as _


def cli(options, argparser: argparse.ArgumentParser):
    argparser.add_argument("-m", "--mod", required=True, help="name of the mod to install")
    argparser.add_argument("-s", "--source", required=True, help="path to the original game")
    argparser.add_argument("-d", "--dest", required=True, help="destination directory of the patched game")
    argparser.add_argument("-ot", "--output_type", required=True, help="format of the patched game")
    args = argparser.parse_args()

    mod_config_path = Path(f"./Pack/{args.mod}/mod_config.json")
    if not mod_config_path.exists():
        sys.stderr.write(_("ERROR_INVALID_MOD") % args.mod + "\n")
        return
    mod_config = ModConfig.from_file(mod_config_path)

    game_path = Path(args.source)
    if not is_valid_source_path(game_path):
        sys.stderr.write(_("ERROR_INVALID_SOURCE_GAME") % game_path + "\n")
        return
    game = Game(args.source)

    destination_path = Path(args.dest)
    if not is_valid_destination_path(destination_path):
        sys.stderr.write(_("ERROR_INVALID_GAME_DESTINATION") % destination_path + "\n")
        return

    try: output_type = Extension[args.output_type]
    except KeyError:
        sys.stderr.write(_("ERROR_INVALID_OUTPUT_TYPE") % args.output_type + "\n")
        return

    # installation and progress bar
    progressbar_max: int = 40

    title: str = ""
    description: str = ""
    current_part: int = 0
    max_part: int = 0
    current_step: int = 0
    max_step: int = 0
    determinate: bool = False

    for step in game.install_mod(
        dest=destination_path,
        mod_config=mod_config,
        output_type=output_type,
        options=options
    ):
        if step.title is not None: title = step.title
        if step.description is not None: description = step.description
        if step.max_part is not None: max_part = step.max_part
        if step.set_part is not None: current_part = step.set_part
        if step.part is not None: current_part += step.part
        if step.max_step is not None: max_step = step.max_step
        if step.set_step is not None: current_step = step.set_step
        if step.step is not None: current_step += step.step
        if step.determinate is not None: determinate = step.determinate

        progressbar_step: int = current_step * progressbar_max // max_step if max_step > 0 else 0

        sys.stdout.write(
            "\033[H\033[J"  # clear the shell
            f"{title} ({current_part} / {max_part})\n"  # print the title and the actual part
            f"{description}\n"  # print the description
        )

        if determinate: sys.stdout.write(
            f"{round((current_step / max_step if max_step > 0 else 0) * 100, 2)}% "
            f"[{'#' * progressbar_step}{' ' * (progressbar_max - progressbar_step)}]\n"
        )

    sys.stdout.write("\033[H\033[J" + get_finished_installation_message(mod_config) + "\n")