__init__.py

from nonebot import on_command, on_startswith, require, get_driver
from nonebot.adapters.onebot.v11 import Message, Bot, MessageEvent, MessageSegment
from nonebot.params import CommandArg


from .utils.Artifact import (
    artifact_obtain,
    ARTIFACT_LIST,
    Artifact,
    calculate_strengthen_points,
)
from .config.config import STAMINA_RESTORE, MAX_STAMINA, Config
from .utils.json_rw import init_user_info, updata_uid_stamina, user_info, save_user_info
from .utils.artifact_eval import *
from pathlib import Path
from base64 import b64encode
from io import BytesIO

import random
import re
import requests

get_obtain = on_command("原神副本", aliases={"圣遗物副本", "查看原神副本", "查看圣遗物副本"})
get_artifact = on_startswith("刷副本")
get_warehouse = on_startswith("查看圣遗物仓库")
strengthen_artifact = on_startswith("强化圣遗物")
artifact_info = on_startswith("圣遗物详情")
artifact_re_init = on_startswith("圣遗物洗点")
transform = on_startswith(("转换狗粮", "转化狗粮"))
transform_all = on_command(
    "转换全部0级圣遗物",
    aliases={"转化全部0级圣遗物"},
)
get_user_stamina = on_command("查看体力值", aliases={"查看体力", "查看当前体力"})
recharge = on_command("氪体力")
artifact_rate = on_command("圣遗物评分")


@get_obtain.handle()
async def get_obtain_(bot: Bot):
    use_pic = Config.parse_obj(get_driver().config.dict()).use_pic
    if use_pic:
        pic = Path(__file__).parent / "resources" / "table.png"
        await get_obtain.send(MessageSegment.image(pic))
    else:
        mes = "当前副本如下\n"
        for name in artifact_obtain.keys():
            suits = " ".join(artifact_obtain[name])
            mes += f"{name}  掉落  {suits}\n"
        await get_obtain.finish(mes, at_sender=True)


@get_artifact.handle()
async def get_artifact_(bot: Bot, event: MessageEvent):
    msg = str(event.get_message())[3:].strip().split(" ")
    if not msg:
        return
    obtain = msg[0]
    ns = 2 if len(msg) > 1 and msg[1] in ["浓缩", "ns"] else 1
    uid = str(event.user_id)
    init_user_info(uid)

    if obtain == "":
        return
    artifact_obtain_map = {
        r"^魔女|渡火": "火本",
        r"^防御|毒奶": "华馆",
        r"^绝缘|旗印|追忆|注连": "充能",
        r"^物理|千岩": "苍白",
        r"^逆飞|流星|磐岩": "岩本",
        r"^沉沦|水套|冰套|雪山": "冰本",
        r"^风套|翠绿|少女": "风本",
        r"^骑士|染血|宗室": "宗室本",
        r"^如雷|平雷": "雷本",
        r"^乐团|角斗|周": "龙狼",
        r"^普攻|掉血|流血": "余响",
        r"^草套|饰金|湿巾": "草本",
        r"^散兵|花神|绽放": "绽放",
        r"^花海|水仙|甘露": "水仙",
        r"^剧团|黄金|猎人|逐影": "黄金",
        r"^昔时|回声|夜话": "昔时"
    }
    for pattern, name in artifact_obtain_map.items():
        if re.match(f"{pattern}本", obtain):
            obtain = name
            break
    if obtain not in artifact_obtain.keys():
        mes = f"没有副本名叫 {obtain} ,发送 原神副本 可查看所有副本"
        await get_artifact.finish(mes, at_sender=True)

    if user_info[uid]["stamina"] < 20 * ns:
        await get_artifact.finish("体力值不足，请等待体力恢复.\n发送 查看体力值 可查看当前体力", at_sender=True)

    user_info[uid]["stamina"] -= 20 * ns
    # 随机掉了几个圣遗物
    if ns == 1:
        r = 2 if random.randint(0, 100000) > 75000 else 1
    else:
        ran = random.randint(0, 100000)
        if ran <= 60000:
            r = 2
        elif 60000 < ran <= 90000:
            r = 3
        else:
            r = 4
    # 随机获得的狗粮点数
    strengthen_points = random.randint(70000, 100000)
    user_info[uid]["strengthen_points"] += strengthen_points * ns

    mes = f"本次刷取副本为 {obtain} \n掉落圣遗物 {r} 个\n获得狗粮点数 {strengthen_points}\n\n"

    for _ in range(r):
        # 随机一个副本掉落的套装名字,然后随机部件的名字
        r_suit_name = random.choice(artifact_obtain[obtain])
        r_artifact_name = random.choice(ARTIFACT_LIST[r_suit_name]["element"])

        artifact = Artifact(r_artifact_name)

        number = int(len(user_info[uid]["warehouse"])) + 1

        # mes += f"当前仓库编号 {number}\n"
        mes += artifact.get_artifact_CQ_code(number)
        mes += "\n"

        user_info[uid]["warehouse"].append(artifact.get_artifact_dict())

    save_user_info()
    await get_artifact.finish(Message(mes), at_sender=True)


@get_warehouse.handle()
async def get_warehouse_(bot: Bot, event: MessageEvent):
    page = str(event.get_message())[7:].strip()
    uid = str(event.user_id)
    init_user_info(uid)
    if page == "":
        page = "1"

    if not page.isdigit():
        await get_warehouse.finish("你需要输入一个数字", at_sender=True)

    page = int(page)

    mes = "仓库如下\n"
    txt = ""

    for i in range(5):
        try:
            ar = user_info[uid]["warehouse"][i + (page - 1) * 5]
            artifact = Artifact(ar)
            number = i + (page - 1) * 5 + 1
            # txt += f"\n\n仓库圣遗物编号 {i+(page-1)*5+1}"
            txt += artifact.get_artifact_CQ_code(number)

        except IndexError:
            pass

    if txt == "":
        txt = "当前页数没有圣遗物"

    mes += txt
    mes += f"\n\n当前为仓库第 {page} 页，你的仓库共有 {(len(user_info[uid]['warehouse']) // 5) + 1} 页"

    await get_warehouse.finish(Message(mes), at_sender=True)


@strengthen_artifact.handle()
async def strengthen_artifact_(bot: Bot, event: MessageEvent):
    uid = str(event.user_id)
    init_user_info(uid)
    try:
        txt = str(event.get_message())[5:].replace(" ", "")
        strengthen_level, number = txt.split("级")
    except Exception:
        await strengthen_artifact.finish("指令格式错误", at_sender=True)

    try:
        # noinspection PyUnboundLocalVariable
        artifact = user_info[uid]["warehouse"][int(number) - 1]
    except IndexError:
        await strengthen_artifact.finish("圣遗物编号错误", at_sender=True)

    # noinspection PyUnboundLocalVariable
    artifact = Artifact(artifact)
    # noinspection PyUnboundLocalVariable
    strengthen_point = calculate_strengthen_points(
        artifact.level + 1, artifact.level + int(strengthen_level)
    )

    if strengthen_point > user_info[uid]["strengthen_points"]:
        await strengthen_artifact.finish(
            "狗粮点数不足\n你可以发送 刷副本 副本名称 获取狗粮点数\n或者发送 转换狗粮 圣遗物编号 销毁仓库里不需要的圣遗物获取狗粮点数\n发送 转换全部0级圣遗物 可将全部0级圣遗物销毁",
            at_sender=True,
        )

    user_info[uid]["strengthen_points"] -= strengthen_point

    for _ in range(int(strengthen_level)):
        artifact.strengthen()

    mes = "强化成功，当前圣遗物属性为：\n"
    mes += artifact.get_artifact_detail()

    user_info[uid]["warehouse"][int(number) - 1] = artifact.get_artifact_dict()
    save_user_info()
    await strengthen_artifact.finish(Message(mes), at_sender=True)


@artifact_info.handle()
async def artifact_info_(bot: Bot, event: MessageEvent):
    number = str(event.get_message())[5:].strip()
    uid = str(event.user_id)
    init_user_info(uid)

    try:
        artifact = user_info[uid]["warehouse"][int(number) - 1]
    except IndexError:
        await artifact_info.finish("编号错误", at_sender=True)
        return

    artifact = Artifact(artifact)
    await artifact_info.finish(Message(artifact.get_artifact_detail()), at_sender=True)


@artifact_re_init.handle()
async def artifact_re_init_(bot: Bot, event: MessageEvent):
    number = str(event.get_message())[5:].strip()
    uid = str(event.user_id)
    init_user_info(uid)

    try:
        artifact = user_info[uid]["warehouse"][int(number) - 1]
    except IndexError:
        await artifact_re_init.finish("编号错误", at_sender=True)

    # noinspection PyUnboundLocalVariable
    artifact = Artifact(artifact)

    if artifact.level < 20:
        await artifact_re_init.finish("没有强化满的圣遗物不能洗点", at_sender=True)

    strengthen_points = calculate_strengthen_points(1, artifact.level)
    strengthen_points = int(strengthen_points * 0.5)

    artifact.re_init()
    user_info[uid]["warehouse"][int(number) - 1] = artifact.get_artifact_dict()

    user_info[uid]["strengthen_points"] += strengthen_points

    mes = f"洗点完成，获得返还狗粮 {strengthen_points} \n当前圣遗物属性如下：\n"
    mes += artifact.get_artifact_detail()
    save_user_info()

    await artifact_re_init.finish(Message(mes), at_sender=True)


@transform.handle()
async def transform_(bot: Bot, event: MessageEvent):
    number = str(event.get_message())[4:].strip().split(" ")
    number = [int(i) for i in number]
    uid = str(event.user_id)
    init_user_info(uid)
    strengthen_points_sum = 0
    ln = 0
    number = sorted(set(number))
    for n in number:
        try:
            artifact = user_info[uid]["warehouse"][n - 1 - ln]
        except IndexError:
            await transform.finish("编号错误", at_sender=True)
        # noinspection PyUnboundLocalVariable
        artifact = Artifact(artifact)

        strengthen_points = calculate_strengthen_points(0, artifact.level)
        strengthen_points = int(strengthen_points * 0.8)

        del user_info[uid]["warehouse"][n - 1 - ln]

        user_info[uid]["strengthen_points"] += strengthen_points
        strengthen_points_sum += strengthen_points
        if ln < len(number):
            ln += 1

    save_user_info()

    mes = f"转化完成，圣遗物已转化为 {int(strengthen_points_sum)} 狗粮点数\n你当前狗粮点数为 {int(user_info[uid]['strengthen_points'])} "
    await transform.finish(Message(mes), at_sender=True)


@get_user_stamina.handle()
async def get_user_stamina_(bot: Bot, event: MessageEvent):
    uid = str(event.user_id)
    init_user_info(uid)
    mes = f"你当前的体力值为 {int(user_info[uid]['stamina'])} ,体力值每 {STAMINA_RESTORE} 分钟恢复1点，自动恢复上限为 {MAX_STAMINA}\n"
    mes += f"你当前的狗粮点数为 {int(user_info[uid]['strengthen_points'])}"
    await get_user_stamina.finish(Message(mes), at_sender=True)


@recharge.handle()
async def recharge_(bot: Bot, event: MessageEvent, arg: Message = CommandArg()):
    # if not (str(event.user_id) in bot.config.superusers):
    #     await recharge.finish(f"这个指令仅限超级管理员使用")
    #     return
    msg = arg.extract_plain_text().strip()
    user = (
        event.message["at"][0].data["qq"] if event.message.get("at") else event.user_id
    )
    if not msg:
        obtain = 60
    else:
        obtain = int(msg)
    init_user_info(str(user))
    user_info[str(user)]["stamina"] += obtain
    save_user_info()
    await recharge.finish(f"充值完毕！谢谢惠顾～")


@transform_all.handle()
async def transform_all_(bot: Bot, event: MessageEvent):
    uid = str(event.user_id)
    init_user_info(uid)

    _0_level_artifact = 0
    temp_list = []

    for artifact in user_info[uid]["warehouse"]:
        if artifact["level"] == 0:
            _0_level_artifact += 1
        else:
            temp_list.append(artifact)

    strengthen_points = _0_level_artifact * 3024

    user_info[uid]["warehouse"] = temp_list
    user_info[uid]["strengthen_points"] += strengthen_points
    save_user_info()

    await transform_all.finish(
        f"0级圣遗物已全部转化为狗粮，共转化 {_0_level_artifact} 个圣遗物，获得狗粮点数 {strengthen_points}"
    )


updata_stamina = require("nonebot_plugin_apscheduler").scheduler


async def get_format_sub_item(artifact_attr):
    msg = ""
    for i in artifact_attr["sub_item"]:
        msg += f'{i["name"]:\u3000<6} | {i["value"]}\n'
    return msg


@artifact_rate.handle()
async def artifact_rate_(bot: Bot, event: MessageEvent):
    if "[CQ:image" not in event.raw_message:
        await artifact_rate.finish("图呢？\n*请将指令与截图一起发送", at_sender=True)
    if len(event.message) > 2:
        await artifact_rate.finish("只能上传一张截图哦", at_sender=True)
    for i in event.message:
        if i.type == "image":
            image_url = i.data["url"]
            break
        continue
    # await bot.send(ev, f"图片收到啦~\n正在识图中...")
    # noinspection PyUnboundLocalVariable
    image_content = BytesIO(requests.get(image_url).content)
    image_b64 = b64encode(image_content.read()).decode()
    try:
        artifact_attr = await get_artifact_attr(image_b64)
    except (requests.exceptions.ConnectionError, requests.exceptions.ReadTimeout):
        await artifact_rate.finish(f"连接超时", at_sender=True)
    # noinspection PyUnboundLocalVariable
    if "err" in artifact_attr.keys():
        err_msg = artifact_attr["full"]["message"]
        await artifact_rate.finish(f"发生了点小错误：\n{err_msg}", at_sender=True)
    # await bot.send(ev, f"识图成功！\n正在评分中...", at_sender=True)
    rate_result = await rate_artifact(artifact_attr)
    if "err" in rate_result.keys():
        err_msg = rate_result["full"]["message"]
        await artifact_rate.finish(
            f"发生了点小错误：\n{err_msg}\n*注：将在下版本加入属性修改", at_sender=True
        )
    format_result = (
        f'圣遗物评分结果：\n主属性：{artifact_attr["main_item"]["name"]}\n{await get_format_sub_item(artifact_attr)}'
        f'------------------------------\n总分：{rate_result["total_percent"]}\n'
        f'主词条：{rate_result["main_percent"]}\n副词条：{rate_result["sub_percent"]}\n评分、识图均来自genshin.pub'
    )
    await artifact_rate.finish(Message(format_result), at_sender=True)


@updata_stamina.scheduled_job("interval", minutes=STAMINA_RESTORE)
async def _call():
    updata_uid_stamina()