from pathlib import Path

from littlepaimon_utils.files import load_json, save_json

RES_PATH = Path() / 'resources' / 'LittlePaimon' / 'gacha_res'
USER_INFO_PATH = Path() / 'data' / 'LittlePaimon' / 'user_data' / 'user_gacha_info.json'
ROLE_1_PATH = RES_PATH / "DIY_gacha_pool" / "role_1.json"
ROLE_2_PATH = RES_PATH / "DIY_gacha_pool" / "role_2.json"
WEAPON_PATH = RES_PATH / "DIY_gacha_pool" / "weapon.json"
All_STAR_PATH = RES_PATH / "DIY_gacha_pool" / "all_star.json"

user_info = load_json(path=USER_INFO_PATH)
role_1_pool = load_json(path=ROLE_1_PATH)
role_2_pool = load_json(path=ROLE_2_PATH)
weapon_pool = load_json(path=WEAPON_PATH)
all_star = load_json(path=All_STAR_PATH)


def save_user_info():
    save_json(data=user_info, path=USER_INFO_PATH)


def init_user_info(uid: str):
    if uid not in user_info:
        user_info[uid] = {}
        user_info[uid]["fate"] = 200
        user_info[uid]["gacha_list"] = {}
        user_info[uid]["gacha_list"]["wish_total"] = 0
        user_info[uid]["gacha_list"]["wish_4"] = 0
        user_info[uid]["gacha_list"]["wish_5"] = 0
        user_info[uid]["gacha_list"]["wish_4_up"] = 0
        user_info[uid]["gacha_list"]["wish_5_up"] = 0
        user_info[uid]["gacha_list"]["gacha_5_role"] = 0
        user_info[uid]["gacha_list"]["gacha_5_weapon"] = 0
        user_info[uid]["gacha_list"]["gacha_5_permanent"] = 0
        user_info[uid]["gacha_list"]["gacha_4_role"] = 0
        user_info[uid]["gacha_list"]["gacha_4_weapon"] = 0
        user_info[uid]["gacha_list"]["gacha_4_permanent"] = 0
        user_info[uid]["gacha_list"]["is_up_5_role"] = False
        user_info[uid]["gacha_list"]["is_up_5_weapon"] = False
        user_info[uid]["gacha_list"]["is_up_4_role"] = False
        user_info[uid]["gacha_list"]["is_up_4_weapon"] = False
        user_info[uid]["gacha_list"]["dg_name"] = ''
        user_info[uid]["gacha_list"]["dg_time"] = 0
        user_info[uid]["role_list"] = {}
        user_info[uid]["role_list"]["旅行者"] = {}
        user_info[uid]["role_list"]["旅行者"]["星级"] = '★★★★★'
        user_info[uid]["role_list"]["旅行者"]["数量"] = 6
        user_info[uid]["role_list"]["旅行者"]["出货"] = [0]
        user_info[uid]["weapon_list"] = {}
        save_user_info()