from tinydb import TinyDB, Query

db = TinyDB('./db.json')
material_table = db.table('material')
medicine_table = db.table('medicine')


def get_medicines(type="ALL"):
    assert type in ["ALL", "回复状态", "突破概率", "加攻击力"], f"type:{type} 不是有效的类别"
    if type in ["ALL"]:
        a = medicine_table.all()
    else:
        medicine = Query()
        a = medicine_table.search(medicine.type == type)
    return list(map(lambda x: x["name"], a))


def _get_medicine_elixir_config(medicine_select: str):
    medicine = Query()
    return medicine_table.search(medicine.name == medicine_select)[0]

def _get_material_elixir_config(material_select: str):
    medicine = Query()
    return material_table.search(medicine.name == material_select)[0]

def get_first_material(medicine_select, medicine_level_select="ALL",material_max_num=5) ->list:
    material = Query()
    m = _get_medicine_elixir_config(medicine_select)
    func1_type = m["func1_type"]
    func1_power = m["func1_power"]
    func2_type = m["func2_type"]
    func2_power = m["func2_power"]
    if medicine_level_select == "ALL":
        a = material_table.search((material.main_func_t == func1_type) | (material.auxi_func_t == func1_type) | (
                    material.main_func_t == func2_type) | (material.auxi_func_t == func2_type))
    else:
        a = material_table.search((material.level == medicine_level_select) & (
                    (material.main_func_t == func1_type) | (material.auxi_func_t == func1_type) | (
                        material.main_func_t == func2_type) | (material.auxi_func_t == func2_type)))

    def get_num(material0):
        global material_second_f
        name = material0["name"]
        if material0["main_func_t"] == func1_type:
            material_second_f = (func2_type,False)
            num = func1_power / material0["main_func_p"]
        elif material0["auxi_func_t"] == func1_type:
            material_second_f = (func2_type,True)
            num = func1_power / material0["auxi_func_p"]
        elif material0["main_func_t"] == func2_type:
            material_second_f = (func1_type,False)
            num = func2_power / material0["main_func_p"]
        elif material0["auxi_func_t"] == func2_type:
            material_second_f = (func1_type,True)
            num = func2_power / material0["auxi_func_p"]
        num = int(num) + 1 if num > int(num) else int(num)
        return (name,num,material_second_f)
    rtn = list(map(get_num, a))
    rtn = list(filter(lambda x:x[1]<=material_max_num, rtn))

    def check_material(material0):
        if material0[1] > material_max_num:
            return False
        material_t = material.main_func_t if material0[2][1] else material.auxi_func_t
        a = material_table.search(material_t == material0[2][0])
        if a == []:
            return False
        return True

    rtn = list(filter(check_material, rtn))
    rtn = list(map(lambda x: f"{x[0]}*{x[1]}", rtn))
    return rtn

def get_second_material(medicine_select, first_material:str, medicine_level_select="ALL",material_max_num=5) ->list:
    m = _get_medicine_elixir_config(medicine_select)
    first_material_name, _ = first_material.split("*")
    first_material = _get_material_elixir_config(first_material_name)
    func1_type = m["func1_type"]
    func1_power = m["func1_power"]
    func2_type = m["func2_type"]
    func2_power = m["func2_power"]

    if first_material["main_func_t"] == func1_type:
        second_material_func_need,second_material_main = (func2_type,func2_power),False
    elif first_material["auxi_func_t"] == func1_type:
        second_material_func_need, second_material_main = (func2_type,func2_power), True
    elif first_material["main_func_t"] == func2_type:
        second_material_func_need, second_material_main = (func1_type,func1_power), False
    elif first_material["auxi_func_t"] == func2_type:
        second_material_func_need, second_material_main = (func1_type,func1_power), True

    material = Query()
    material_t = material.main_func_t if second_material_main else material.auxi_func_t
    if medicine_level_select == "ALL":
        a = material_table.search((material_t == second_material_func_need[0]))
    else:
        a = material_table.search((material.level == medicine_level_select) & (material_t == second_material_func_need[0]))

    def get_num(material0):
        name = material0["name"]
        material0_p = material0["main_func_p"] if second_material_main else material0["auxi_func_p"]
        num = second_material_func_need[1]/material0_p
        num = int(num) + 1 if num > int(num) else int(num)
        return (name,num)

    rtn = list(map(get_num, a))
    rtn = list(filter(lambda x:x[1]<=material_max_num, rtn))
    rtn = list(map(lambda x: f"{x[0]}*{x[1]}", rtn))
    return rtn

def get_possible_material(medicine_select, first_material:str="无", second_material:str="无",material_max_num=5):
    possible_choice = set()
    if first_material == "无":
        for first_material in get_first_material(medicine_select):
            for second_material in get_second_material(medicine_select, first_material):
                possible_choice.add((first_material, second_material))
    elif second_material == "无":
        for second_material in get_second_material(medicine_select,first_material):
            possible_choice.add((first_material, second_material))
    else:
        possible_choice.add((first_material,second_material))

    m = _get_medicine_elixir_config(medicine_select)
    func1_type = m["func1_type"]
    func2_type = m["func2_type"]

    rtn = []
    for first_material,second_material in possible_choice:
        first_material_name,first_material_num = first_material.split("*")
        second_material_name,second_material_num = second_material.split("*")
        first_material = _get_material_elixir_config(first_material_name)
        second_material = _get_material_elixir_config(second_material_name)
        if first_material["main_func_t"] in [func1_type,func2_type]:
            main_temp = first_material["main_temp"]
            main_material = f"{first_material_name}*{first_material_num}"
            auxi_material = f"{second_material_name}*{second_material_num}"
        else:
            main_temp = second_material["main_temp"]
            auxi_material = f"{first_material_name}*{first_material_num}"
            main_material = f"{second_material_name}*{second_material_num}"

        if main_temp==0:
            material_third_list=['恒心草(一品)*1', '红绫草(一品)*1', '五柳根(二品)*1', '天元果(二品)*1', '紫猴花(三品)*1', '九叶芝(三品)*1', '血莲精(四品)*1', '鸡冠草(四品)*1', '地心火芝(五品)*1', '天蝉灵叶(五品)*1', '三叶青芝(六品)*1', '七彩月兰(六品)*1', '地心淬灵乳(七品)*1', '天麻翡石精(七品)*1', '木灵三针花(八品)*1', '鎏鑫天晶草(八品)*1', '离火梧桐芝(九品)*1', '尘磊岩麟果(九品)*1', '宁心草(一品)*1', '凝血草(一品)*1', '流莹草(二品)*1', '蛇涎果(二品)*1', '轻灵草(三品)*1', '龙葵(三品)*1', '菩提花(四品)*1', '乌稠木(四品)*1', '天灵果(五品)*1', '灯心草(五品)*1', '白沉脂(六品)*1', '苦曼藤(六品)*1', '天问花(七品)*1', '渊血冥花(七品)*1', '阴阳黄泉花(八品)*1', '厉魂血珀(八品)*1', '太乙碧莹花(九品)*1', '森檀木(九品)*1', '地黄参(一品)*1', '火精枣(一品)*1', '风灵花(二品)*1', '伏龙参(二品)*1', '枫香脂(三品)*1', '炼魂珠(三品)*1', '石龙芮(四品)*1', '锦地罗(四品)*1', '伴妖草(五品)*1', '剑心竹(五品)*1', '混元果(六品)*1', '皇龙花(六品)*1', '血玉竹(七品)*1', '肠蚀草(七品)*1', '狼桃(八品)*1', '霸王花(八品)*1', '地龙干(九品)*1', '龙须藤(九品)*1']

        else:
            material0 = Query()
            material0 = material0.phar_temp > 0 if main_temp<0 else material0.phar_temp <0
            a = material_table.search(material0)

            def get_num(x):
                name = x["name"]
                phar_temp = x["phar_temp"]
                num = -main_temp//phar_temp
                num = 1 if num==0 else num
                return (name,num)

            a = list(map(get_num,a))
            a = list(filter(lambda x:x[1]<=material_max_num, a))
            material_third_list = list(map(lambda x:f'{x[0]}*{x[1]}',a))
        rtn.append((main_material,auxi_material,material_third_list))
    return rtn



def init():
    medicine_list = get_medicines()
    return medicine_list