from tinydb import Query, where

import config
import random
import time

from fastapi import APIRouter, Request, Response

from core_common import core_process_request, core_prepare_response, E
from core_database import get_db

router = APIRouter(prefix="/local2", tags=["local2"])
router.model_whitelist = ["KFC"]


def get_profile(cid):
    return get_db().table("sdvx_profile").get(where("card") == cid)


def get_game_profile(cid, game_version):
    profile = get_profile(cid)

    return profile["version"].get(str(game_version), None)


def get_id_from_profile(cid):
    profile = get_db().table("sdvx_profile").get(where("card") == cid)

    djid = "%08d" % profile["sdvx_id"]
    djid_split = "-".join([djid[:4], djid[4:]])

    return profile["sdvx_id"], djid_split


@router.post("/{gameinfo}/game/sv6_common")
async def game_sv6_common(request: Request):
    request_info = await core_process_request(request)

    event = [
        "DEMOGAME_PLAY",
        "MATCHING_MODE",
        "MATCHING_MODE_FREE_IP",
        "LEVEL_LIMIT_EASING",
        "ACHIEVEMENT_ENABLE",
        "APICAGACHADRAW\t30",
        "VOLFORCE_ENABLE",
        "AKANAME_ENABLE",
        "PAUSE_ONLINEUPDATE",
        "CONTINUATION",
        "TENKAICHI_MODE",
        "QC_MODE",
        "KAC_MODE",
        # "APPEAL_CARD_GEN_PRICE\t100",
        # "APPEAL_CARD_GEN_NEW_PRICE\t200",
        # "APPEAL_CARD_UNLOCK\t0,20170914,0,20171014,0,20171116,0,20180201,0,20180607,0,20181206,0,20200326,0,20200611,4,10140732,6,10150431",
        "FAVORITE_APPEALCARD_MAX\t200",
        "FAVORITE_MUSIC_MAX\t200",
        "EVENTDATE_APRILFOOL",
        "KONAMI_50TH_LOGO",
        "OMEGA_ARS_ENABLE",
        "DISABLE_MONITOR_ID_CHECK",
        "SKILL_ANALYZER_ABLE",
        "BLASTER_ABLE",
        "STANDARD_UNLOCK_ENABLE",
        "PLAYERJUDGEADJ_ENABLE",
        "MIXID_INPUT_ENABLE",
        "EVENTDATE_ONIGO",
        "EVENTDATE_GOTT",
        "GENERATOR_ABLE",
        "CREW_SELECT_ABLE",
        "PREMIUM_TIME_ENABLE",
        "OMEGA_ENABLE\t1,2,3,4,5,6,7,8,9",
        "HEXA_ENABLE\t1,2,3,4,5",
        "MEGAMIX_ENABLE",
        "VALGENE_ENABLE",
        "ARENA_ENABLE",
        "DISP_PASELI_BANNER",
    ]

    unlock = []
    for i in range(2000):
        for j in range(0, 5):
            unlock.append([i, j])

    response = E.response(
        E.game(
            E.event(
                *[
                    E.info(
                        E.event_id(s, __type="str"),
                    )
                    for s in event
                ],
            ),
            E.music_limited(
                *[
                    E.info(
                        E.music_id(s[0], __type="s32"),
                        E.music_type(s[1], __type="u8"),
                        E.limited(3, __type="u8"),
                    )
                    for s in unlock
                ],
            ),
        )
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/game/sv6_new")
async def game_sv6_new(request: Request):
    request_info = await core_process_request(request)
    game_version = request_info["game_version"]

    root = request_info["root"][0]

    dataid = root.find("dataid").text
    cardno = root.find("cardno").text
    name = root.find("name").text

    db = get_db().table("sdvx_profile")
    all_profiles_for_card = db.get(Query().card == dataid)

    if all_profiles_for_card is None:
        all_profiles_for_card = {"card": dataid, "version": {}}

    if "sdvx_id" not in all_profiles_for_card:
        sdvx_id = random.randint(10000000, 99999999)
        all_profiles_for_card["sdvx_id"] = sdvx_id

    all_profiles_for_card["version"][str(game_version)] = {
        "game_version": game_version,
        "name": name,
        "appeal_id": 0,
        "skill_level": 0,
        "skill_base_id": 0,
        "skill_name_id": 0,
        "earned_gamecoin_packet": 0,
        "earned_gamecoin_block": 0,
        "earned_blaster_energy": 0,
        "earned_extrack_energy": 0,
        "used_packet_booster": 0,
        "used_block_booster": 0,
        "hispeed": 0,
        "lanespeed": 0,
        "gauge_option": 0,
        "ars_option": 0,
        "notes_option": 0,
        "early_late_disp": 0,
        "draw_adjust": 0,
        "eff_c_left": 0,
        "eff_c_right": 1,
        "music_id": 0,
        "music_type": 0,
        "sort_type": 0,
        "narrow_down": 0,
        "headphone": 1,
        "print_count": 0,
        "start_option": 0,
        "bgm": 0,
        "submonitor": 0,
        "nemsys": 0,
        "stampA": 0,
        "stampB": 0,
        "stampC": 0,
        "stampD": 0,
        "items": [],
        "params": [],
    }

    db.upsert(all_profiles_for_card, where("card") == dataid)

    response = E.response(
        E.game(
            E.result(0, __type="u8"),
        ),
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/game/sv6_load")
async def game_sv6_load(request: Request):
    request_info = await core_process_request(request)
    game_version = request_info["game_version"]

    dataid = request_info["root"][0].find("dataid").text
    profile = get_game_profile(dataid, game_version)

    if profile:
        djid, djid_split = get_id_from_profile(dataid)

        unlock = []
        for i in range(301):
            unlock.append([i, 11, 15])
        for i in range(6001):
            unlock.append([i, 1, 1])
        unlock.append([599, 4, 10])
        for item in profile["items"]:
            unlock.append(item)

        customize = [
            [
                2,
                2,
                [
                    profile["bgm"],
                    profile["submonitor"],
                    profile["nemsys"],
                    profile["stampA"],
                    profile["stampB"],
                    profile["stampC"],
                    profile["stampD"],
                ],
            ]
        ]
        for item in profile["params"]:
            customize.append(item)

        response = E.response(
            E.game(
                E.result(0, __type="u8"),
                E.name(profile["name"], __type="str"),
                E.code(djid_split, __type="str"),
                E.sdvx_id(djid_split, __type="str"),
                E.appeal_id(profile["appeal_id"], __type="u16"),
                E.skill_level(profile["skill_level"], __type="s16"),
                E.skill_base_id(profile["skill_base_id"], __type="s16"),
                E.skill_name_id(profile["skill_name_id"], __type="s16"),
                E.gamecoin_packet(profile["earned_gamecoin_packet"], __type="u32"),
                E.gamecoin_block(profile["earned_gamecoin_block"], __type="u32"),
                E.blaster_energy(profile["earned_blaster_energy"], __type="u32"),
                E.blaster_count(9999, __type="u32"),
                E.extrack_energy(profile["earned_extrack_energy"], __type="u16"),
                E.play_count(1001, __type="u32"),
                E.day_count(301, __type="u32"),
                E.today_count(21, __type="u32"),
                E.play_chain(31, __type="u32"),
                E.max_play_chain(31, __type="u32"),
                E.week_count(9, __type="u32"),
                E.week_play_count(101, __type="u32"),
                E.week_chain(31, __type="u32"),
                E.max_week_chain(1001, __type="u32"),
                E.creator_id(1, __type="u32"),
                E.eaappli(E.relation(1, __type="s8")),
                E.ea_shop(
                    E.blaster_pass_enable(1, __type="bool"),
                    E.blaster_pass_limit_date(1605871200, __type="u64"),
                ),
                E.kac_id(profile["name"], __type="str"),
                E.block_no(0, __type="s32"),
                E.volte_factory(
                    *[
                        E.info(
                            E.goods_id(s, __type="s32"),
                            E.status(1, __type="s32"),
                        )
                        for s in range(1, 999)
                    ],
                ),
                *[
                    E.campaign(
                        E.campaign_id(s, __type="s32"),
                        E.jackpot_flg(1, __type="bool"),
                    )
                    for s in range(99)
                ],
                E.cloud(E.relation(1, __type="s8")),
                E.something(
                    *[
                        E.info(
                            E.ranking_id(s[0], __type="s32"),
                            E.value(s[1], __type="s64"),
                        )
                        for s in [[1402, 20000]]
                    ],
                ),
                E.festival(
                    E.fes_id(1, __type="s32"),
                    E.live_energy(1000000, __type="s32"),
                    *[
                        E.bonus(
                            E.energy_type(s, __type="s32"),
                            E.live_energy(1000000, __type="s32"),
                        )
                        for s in range(1, 6)
                    ],
                ),
                E.valgene_ticket(
                    E.ticket_num(0, __type="s32"),
                    E.limit_date(1605871200, __type="u64"),
                ),
                E.arena(
                    E.last_play_season(0, __type="s32"),
                    E.rank_point(0, __type="s32"),
                    E.shop_point(0, __type="s32"),
                    E.ultimate_rate(0, __type="s32"),
                    E.ultimate_rank_num(0, __type="s32"),
                    E.rank_play_cnt(0, __type="s32"),
                    E.ultimate_play_cnt(0, __type="s32"),
                ),
                E.hispeed(profile["hispeed"], __type="s32"),
                E.lanespeed(profile["lanespeed"], __type="u32"),
                E.gauge_option(profile["gauge_option"], __type="u8"),
                E.ars_option(profile["ars_option"], __type="u8"),
                E.notes_option(profile["notes_option"], __type="u8"),
                E.early_late_disp(profile["early_late_disp"], __type="u8"),
                E.draw_adjust(profile["draw_adjust"], __type="s32"),
                E.eff_c_left(profile["eff_c_left"], __type="u8"),
                E.eff_c_right(profile["eff_c_right"], __type="u8"),
                E.last_music_id(profile["music_id"], __type="s32"),
                E.last_music_type(profile["music_type"], __type="u8"),
                E.sort_type(profile["sort_type"], __type="u8"),
                E.narrow_down(profile["narrow_down"], __type="u8"),
                E.headphone(profile["headphone"], __type="u8"),
                E.item(
                    *[
                        E.info(
                            E.id(s[0], __type="u32"),
                            E.type(s[1], __type="u8"),
                            E.param(s[2], __type="u32"),
                        )
                        for s in unlock
                    ],
                ),
                E.param(
                    *[
                        E.info(
                            E.type(s[0], __type="s32"),
                            E.id(s[1], __type="s32"),
                            E.param(s[2], __type="s32", __count=len(s[2])),
                        )
                        for s in customize
                    ],
                ),
            ),
        )

    else:
        response = E.response(
            E.game(
                E.result(1, __type="u8"),
            )
        )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/game/sv6_load_m")
async def game_sv6_load_m(request: Request):
    request_info = await core_process_request(request)
    game_version = request_info["game_version"]

    dataid = request_info["root"][0].find("refid").text
    profile = get_game_profile(dataid, game_version)
    djid, djid_split = get_id_from_profile(dataid)

    best_scores = []
    db = get_db()
    for record in db.table("sdvx_scores_best").search(
        (where("game_version") == game_version) & (where("sdvx_id") == djid)
    ):
        best_scores.append(
            [
                record["music_id"],
                record["music_type"],
                record["score"],
                record["exscore"],
                record["clear_type"],
                record["score_grade"],
                0,
                0,
                record["btn_rate"],
                record["long_rate"],
                record["vol_rate"],
                0,
                0,
                0,
                0,
                0,
                0,
                0,
                0,
                0,
                0,
            ]
        )

    response = E.response(
        E.game(
            E.music(
                *[
                    E.info(
                        E.param(x, __type="u32"),
                    )
                    for x in best_scores
                ],
            ),
        ),
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/game/sv6_save")
async def game_sv6_save(request: Request):
    request_info = await core_process_request(request)
    game_version = request_info["game_version"]

    dataid = request_info["root"][0].find("refid").text

    profile = get_profile(dataid)
    game_profile = profile["version"].get(str(game_version), {})

    root = request_info["root"][0]

    game_profile["appeal_id"] = int(root.find("appeal_id").text)

    nodes = [
        "appeal_id",
        "skill_level",
        "skill_base_id",
        "skill_name_id",
        "earned_gamecoin_packet",
        "earned_gamecoin_block",
        "earned_blaster_energy",
        "earned_extrack_energy",
        "hispeed",
        "lanespeed",
        "gauge_option",
        "ars_option",
        "notes_option",
        "early_late_disp",
        "draw_adjust",
        "eff_c_left",
        "eff_c_right",
        "music_id",
        "music_type",
        "sort_type",
        "narrow_down",
        "headphone",
        "start_option",
    ]

    for node in nodes:
        game_profile[node] = int(root.find(node).text)

    game_profile["used_packet_booster"] = int(root.find("ea_shop")[0].text)
    game_profile["used_block_booster"] = int(root.find("ea_shop")[1].text)
    game_profile["print_count"] = int(root.find("print")[0].text)

    items = []
    for info in root.find("item"):
        items.append(
            [
                int(info.find("id").text),
                int(info.find("type").text),
                int(info.find("param").text),
            ]
        )
    game_profile["items"] = items

    params = []
    for info in root.find("param"):
        p = info.find("param")
        params.append(
            [
                int(info.find("type").text),
                int(info.find("id").text),
                [int(x) for x in p.text.split(" ")],
            ]
        )
    game_profile["params"] = params

    profile["version"][str(game_version)] = game_profile

    get_db().table("sdvx_profile").upsert(profile, where("card") == dataid)

    response = E.response(
        E.game(),
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/game/sv6_save_m")
async def game_sv6_save_m(request: Request):
    request_info = await core_process_request(request)
    game_version = request_info["game_version"]

    timestamp = time.time()

    root = request_info["root"][0]

    dataid = root.find("dataid").text
    profile = get_game_profile(dataid, game_version)
    djid, djid_split = get_id_from_profile(dataid)

    track = root.find("track")
    play_id = int(track.find("play_id").text)
    music_id = int(track.find("music_id").text)
    music_type = int(track.find("music_type").text)
    score = int(track.find("score").text)
    exscore = int(track.find("exscore").text)
    clear_type = int(track.find("clear_type").text)
    score_grade = int(track.find("score_grade").text)
    max_chain = int(track.find("max_chain").text)
    just = int(track.find("just").text)
    critical = int(track.find("critical").text)
    near = int(track.find("near").text)
    error = int(track.find("error").text)
    effective_rate = int(track.find("effective_rate").text)
    btn_rate = int(track.find("btn_rate").text)
    long_rate = int(track.find("long_rate").text)
    vol_rate = int(track.find("vol_rate").text)
    mode = int(track.find("mode").text)
    gauge_type = int(track.find("gauge_type").text)
    notes_option = int(track.find("notes_option").text)
    online_num = int(track.find("online_num").text)
    local_num = int(track.find("local_num").text)
    challenge_type = int(track.find("challenge_type").text)
    retry_cnt = int(track.find("retry_cnt").text)
    judge = [int(x) for x in track.find("judge").text.split(" ")]

    db = get_db()
    db.table("sdvx_scores").insert(
        {
            "timestamp": timestamp,
            "game_version": game_version,
            "sdvx_id": djid,
            "play_id": play_id,
            "music_id": music_id,
            "music_type": music_type,
            "score": score,
            "exscore": exscore,
            "clear_type": clear_type,
            "score_grade": score_grade,
            "max_chain": max_chain,
            "just": just,
            "critical": critical,
            "near": near,
            "error": error,
            "effective_rate": effective_rate,
            "btn_rate": btn_rate,
            "long_rate": long_rate,
            "vol_rate": vol_rate,
            "mode": mode,
            "gauge_type": gauge_type,
            "notes_option": notes_option,
            "online_num": online_num,
            "local_num": local_num,
            "challenge_type": challenge_type,
            "retry_cnt": retry_cnt,
            "judge": judge,
        },
    )

    best = db.table("sdvx_scores_best").get(
        (where("sdvx_id") == djid)
        & (where("game_version") == game_version)
        & (where("music_id") == music_id)
        & (where("music_type") == music_type)
    )
    best = {} if best is None else best

    best_score_data = {
        "game_version": game_version,
        "sdvx_id": djid,
        "name": profile["name"],
        "music_id": music_id,
        "music_type": music_type,
        "score": max(score, best.get("score", score)),
        "exscore": max(exscore, best.get("exscore", exscore)),
        "clear_type": max(clear_type, best.get("clear_type", clear_type)),
        "score_grade": max(score_grade, best.get("score_grade", score_grade)),
        "btn_rate": max(btn_rate, best.get("btn_rate", btn_rate)),
        "long_rate": max(long_rate, best.get("long_rate", long_rate)),
        "vol_rate": max(vol_rate, best.get("vol_rate", vol_rate)),
    }

    db.table("sdvx_scores_best").upsert(
        best_score_data,
        (where("sdvx_id") == djid)
        & (where("game_version") == game_version)
        & (where("music_id") == music_id)
        & (where("music_type") == music_type),
    )

    response = E.response(
        E.game(),
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/game/sv6_hiscore")
async def game_sv6_hiscore(request: Request):
    request_info = await core_process_request(request)
    game_version = request_info["game_version"]

    best_scores = []
    db = get_db()
    for record in db.table("sdvx_scores_best").search(
        (where("game_version") == game_version)
    ):
        best_scores.append(
            [
                record["music_id"],
                record["music_type"],
                record["sdvx_id"],
                record["name"],
                record["score"],
            ]
        )

    response = E.response(
        E.game(
            E.sc(
                *[
                    E.d(
                        E.id(s[0], __type="u32"),
                        E.ty(s[1], __type="u32"),
                        E.a_sq(s[2], __type="str"),
                        E.a_nm(s[3], __type="str"),
                        E.a_sc(s[4], __type="u32"),
                        E.l_sq(s[2], __type="str"),
                        E.l_nm(s[3], __type="str"),
                        E.l_sc(s[4], __type="u32"),
                    )
                    for s in best_scores
                ],
            ),
        ),
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/game/sv6_lounge")
async def game_sv6_lounge(request: Request):
    request_info = await core_process_request(request)

    response = E.response(
        E.game(E.interval(30, __type="u32")),
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


@router.post("/{gameinfo}/game/sv6_shop")
async def game_sv6_shop(request: Request):
    request_info = await core_process_request(request)

    response = E.response(
        E.game(E.nxt_time(1000 * 5 * 60, __type="u32")),
    )

    response_body, response_headers = await core_prepare_response(request, response)
    return Response(content=response_body, headers=response_headers)


for stub in [
    "load_r",
    "frozen",
    "save_e",
    "save_mega",
    "play_e",
    "play_s",
    "entry_s",
    "entry_e",
    "log",
]:

    @router.post(f"/{{gameinfo}}/game/sv6_{stub}")
    async def game_sv6_stub(request: Request):
        request_info = await core_process_request(request)

        response = E.response(
            E.game(),
        )

        response_body, response_headers = await core_prepare_response(request, response)
        return Response(content=response_body, headers=response_headers)