#! /usr/bin/python3

import argparse

from check_game import check_game
from download_data import detailed_export_game
from compress import link

"""
init db + populate tables
download games of variant
download single game
analyze single game
"""


def add_init_subparser(subparsers):
    parser = subparsers.add_parser(
        'init',
        help='Init database tables, retrieve variant and suit information from hanab.live'
    )


def add_download_subparser(subparsers):
    parser = subparsers.add_parser('download', help='Download games from hanab.live')
    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument('--var', '-v', type=int)
    group.add_argument('--id', '-i', type=int)


def add_analyze_subparser(subparsers):
    parser = subparsers.add_parser('analyze', help='Analyze a game and find the last winning state')
    parser.add_argument('game_id', type=int)
    parser.add_argument('--download', '-d', help='Download game if not in database', action='store_true')


def analyze_game(game_id: int, download: bool = False):
    if download:
        detailed_export_game(game_id)
    print('Analyzing game {}'.format(game_id))
    turn, sol = check_game(game_id)
    if turn == 0:
        print('Instance is unfeasible')
    else:
        print(
            'Last winning turn was {}.\n'
            'A replay achieving perfect score is: {}'.format(turn, link(sol))
        )


def main_parser() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser(
        prog='hanabi_suite',
        description='High-level interface for analysis of hanabi instances.'
    )
    subparsers = parser.add_subparsers(dest='command', required=True, help='select subcommand')

    add_init_subparser(subparsers)
    add_analyze_subparser(subparsers)
    add_download_subparser(subparsers)

    return parser


if __name__ == "__main__":
    args = main_parser().parse_args()
    switcher = {
        'analyze': analyze_game
    }
    method_args = dict(vars(args))
    method_args.pop('command')
    switcher[args.command](**method_args)