zhenxun_bot/plugins/genshin/query_user/query_memo/data_source.py

import asyncio
from asyncio.exceptions import TimeoutError
from io import BytesIO
from typing import Optional, Tuple, Union

import nonebot
from nonebot import Driver
from nonebot.adapters.onebot.v11 import MessageSegment

from configs.config import Config
from configs.path_config import IMAGE_PATH
from services.log import logger
from utils.http_utils import AsyncHttpx
from utils.image_utils import BuildImage
from utils.message_builder import image
from utils.utils import get_user_avatar

from .._models import Genshin
from .._utils import get_ds

driver: Driver = nonebot.get_driver()


memo_path = IMAGE_PATH / "genshin" / "genshin_memo"
memo_path.mkdir(exist_ok=True, parents=True)


@driver.on_startup
async def _():
    for name, url in zip(
        ["resin.png", "task.png", "resin_discount.png", "chengehu.png", "zhibian.png"],
        [
            "https://upload-bbs.mihoyo.com/upload/2021/09/29/8819732/54266243c7d15ba31690c8f5d63cc3c6_71491376413333325"
            "20.png?x-oss-process=image//resize,s_600/quality,q_80/auto-orient,0/interlace,1/format,png",
            "https://patchwiki.biligame.com/images/ys/thumb/c/cc/6k6kuj1kte6m1n7hexqfrn92z6h4yhh.png/60px-委托任务logo.png",
            "https://patchwiki.biligame.com/images/ys/d/d9/t1hv6wpucbwucgkhjntmzroh90nmcdv.png",
            "https://s3.bmp.ovh/imgs/2022/08/21/3a3b2e6c22e305ff.png",
            "https://s3.bmp.ovh/imgs/2022/08/21/c2d7ace21e1d46cf.png",
        ],
    ):
        file = memo_path / name
        if not file.exists():
            await AsyncHttpx.download_file(url, file)
            logger.info(f"已下载原神便签资源 -> {file}...")


async def get_user_memo(
    user_id: int, uid: int, uname: str
) -> Optional[Union[str, MessageSegment]]:
    uid = str(uid)
    if uid[0] in ["1", "2"]:
        server_id = "cn_gf01"
    elif uid[0] == "5":
        server_id = "cn_qd01"
    else:
        return None
    return await parse_data_and_draw(user_id, uid, server_id, uname)


async def get_memo(uid: str, server_id: str) -> Tuple[Union[str, dict], int]:
    try:
        req = await AsyncHttpx.get(
            url=f"https://api-takumi-record.mihoyo.com/game_record/app/genshin/api/dailyNote?server={server_id}&role_id={uid}",
            headers={
                "DS": get_ds(f"role_id={uid}&server={server_id}"),
                "x-rpc-app_version": Config.get_config("genshin", "mhyVersion"),
                "User-Agent": "Mozilla/5.0 (iPhone; CPU iPhone OS 13_2_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) miHoYoBBS/2.11.1",
                "x-rpc-client_type": Config.get_config("genshin", "client_type"),
                "Referer": "https://webstatic.mihoyo.com/",
                "Cookie": await Genshin.random_cookie(uid),
            },
        )
        data = req.json()
        if data["message"] == "OK":
            return data["data"], 200
        return data["message"], 999
    except TimeoutError:
        return "访问超时，请稍后再试", 997
    except Exception as e:
        logger.error(f"便签查询获取失败未知错误 {e}：{e}")
        return "发生了一些错误，请稍后再试", 998


def create_border(
    image_name: str, content: str, notice_text: str, value: str
) -> BuildImage:
    border = BuildImage(500, 75, color="#E0D9D1", font="HYWenHei-85W.ttf", font_size=20)
    text_bk = BuildImage(
        350, 75, color="#F5F1EB", font_size=23, font="HYWenHei-85W.ttf"
    )
    _x = 70 if image_name == "resin.png" else 50
    _px = 10 if image_name == "resin.png" else 20
    text_bk.paste(
        BuildImage(_x, _x, background=memo_path / image_name),
        (_px, 0),
        True,
        center_type="by_height",
    )
    text_bk.text((87, 15), content)
    text_bk.paste(
        BuildImage(
            0,
            0,
            plain_text=notice_text,
            font_color=(203, 189, 175),
            font="HYWenHei-85W.ttf",
            font_size=17,
        ),
        (87, 45),
        True,
    )
    font_width, _ = border.getsize(value)
    border.text((350 + 76 - int(font_width / 2), 0), value, center_type="by_height")
    border.paste(text_bk, (2, 0), center_type="by_height")
    return border


async def parse_data_and_draw(
    user_id: int, uid: str, server_id: str, uname: str
) -> Union[str, MessageSegment]:
    data, code = await get_memo(uid, server_id)
    if code != 200:
        return data
    user_avatar = BytesIO(await get_user_avatar(user_id))
    for x in data["expeditions"]:
        file_name = x["avatar_side_icon"].split("_")[-1]
        role_avatar = memo_path / "role_avatar" / file_name
        if not role_avatar.exists():
            await AsyncHttpx.download_file(x["avatar_side_icon"], role_avatar)
    return await asyncio.get_event_loop().run_in_executor(
        None, _parse_data_and_draw, data, user_avatar, uid, uname
    )


def _parse_data_and_draw(
    data: dict, user_avatar: BytesIO, uid: int, uname: str
) -> Union[str, MessageSegment]:
    current_resin = data["current_resin"]  # 当前树脂
    max_resin = data["max_resin"]  # 最大树脂
    resin_recovery_time = data["resin_recovery_time"]  # 树脂全部回复时间
    finished_task_num = data["finished_task_num"]  # 完成的每日任务
    total_task_num = data["total_task_num"]  # 每日任务总数
    remain_resin_discount_num = data["remain_resin_discount_num"]  # 值得铭记的强敌总数
    resin_discount_num_limit = data["resin_discount_num_limit"]  # 剩余值得铭记的强敌
    current_expedition_num = data["current_expedition_num"]  # 当前挖矿人数
    max_expedition_num = data["max_expedition_num"]  # 每日挖矿最大人数
    expeditions = data["expeditions"]  # 挖矿详情
    current_coin = data["current_home_coin"]  # 当前宝钱
    max_coin = data["max_home_coin"]  # 最大宝钱
    coin_recovery_time = data["home_coin_recovery_time"]  # 宝钱全部回复时间
    transformer_available = data["transformer"]["obtained"]  # 参量质变仪可获取
    transformer_state = data["transformer"]["recovery_time"]["reached"]  # 参量质变仪状态
    transformer_recovery_time = data["transformer"]["recovery_time"]["Day"]  # 参量质变仪回复时间
    transformer_recovery_hour = data["transformer"]["recovery_time"][
        "Hour"
    ]  # 参量质变仪回复时间
    coin_minute, coin_second = divmod(int(coin_recovery_time), 60)
    coin_hour, coin_minute = divmod(coin_minute, 60)
    # print(data)
    minute, second = divmod(int(resin_recovery_time), 60)
    hour, minute = divmod(minute, 60)

    A = BuildImage(1030, 570, color="#f1e9e1", font_size=15, font="HYWenHei-85W.ttf")
    A.text((10, 15), "原神便笺 | Create By ZhenXun", (198, 186, 177))
    ava = BuildImage(100, 100, background=user_avatar)
    ava.circle()
    A.paste(ava, (40, 40), True)
    A.paste(
        BuildImage(0, 0, plain_text=uname, font_size=20, font="HYWenHei-85W.ttf"),
        (160, 62),
        True,
    )
    A.paste(
        BuildImage(
            0,
            0,
            plain_text=f"UID：{uid}",
            font_size=15,
            font="HYWenHei-85W.ttf",
            font_color=(21, 167, 89),
        ),
        (160, 92),
        True,
    )
    border = create_border(
        "resin.png",
        "原粹树脂",
        "将在{:0>2d}:{:0>2d}:{:0>2d}秒后全部恢复".format(hour, minute, second),
        f"{current_resin}/{max_resin}",
    )

    A.paste(border, (10, 155))
    border = create_border(
        "task.png",
        "每日委托",
        "今日委托已全部完成" if finished_task_num == total_task_num else "今日委托完成数量不足",
        f"{finished_task_num}/{total_task_num}",
    )
    A.paste(border, (10, 235))
    border = create_border(
        "resin_discount.png",
        "值得铭记的强敌",
        "本周剩余消耗减半次数",
        f"{remain_resin_discount_num}/{resin_discount_num_limit}",
    )
    A.paste(border, (10, 315))
    border = create_border(
        "chengehu.png",
        "洞天财翁-洞天宝钱",
        "洞天财翁已达到存储上限"
        if current_coin == max_coin
        else f"{coin_hour}小时{coin_minute}分钟后存满",
        f"{current_coin}/{max_coin}",
    )
    A.paste(border, (10, 395))
    border = create_border(
        "zhibian.png",
        "参量质变仪",
        "不存在"
        if not transformer_available
        else "已准备完成 "
        if transformer_state
        else f"{transformer_recovery_hour}小时后可使用"
        if not transformer_recovery_time
        else f"{transformer_recovery_time}天后可使用",
        "不存在" if not transformer_available else "可使用" if transformer_state else "冷却中",
    )
    A.paste(border, (10, 475))

    expeditions_border = BuildImage(
        470, 510, color="#E0D9D1", font="HYWenHei-85W.ttf", font_size=20
    )
    expeditions_text = BuildImage(
        466, 506, color="#F5F1EB", font_size=23, font="HYWenHei-85W.ttf"
    )
    expeditions_text.text(
        (5, 5), f"探索派遣限制{current_expedition_num}/{max_expedition_num}", (100, 100, 98)
    )
    h = 45
    for x in expeditions:
        _bk = BuildImage(
            400, 82, color="#ECE3D8", font="HYWenHei-85W.ttf", font_size=21
        )
        file_name = x["avatar_side_icon"].split("_")[-1]
        role_avatar = memo_path / "role_avatar" / file_name
        _ava_img = BuildImage(75, 75, background=role_avatar)
        # _ava_img.circle()
        if x["status"] == "Finished":
            msg = "探索完成"
            font_color = (146, 188, 63)
            _circle_color = (146, 188, 63)
        else:
            minute, second = divmod(int(x["remained_time"]), 60)
            hour, minute = divmod(minute, 60)
            font_color = (193, 180, 167)
            msg = "还剩{:0>2d}小时{:0>2d}分钟{:0>2d}秒".format(hour, minute, second)
            _circle_color = "#DE9C58"

        _circle_bk = BuildImage(60, 60)
        _circle_bk.circle()
        a_circle = BuildImage(55, 55, color=_circle_color)
        a_circle.circle()
        b_circle = BuildImage(47, 47)
        b_circle.circle()
        a_circle.paste(b_circle, (4, 4), True)
        _circle_bk.paste(a_circle, (4, 4), True)

        _bk.paste(_circle_bk, (25, 0), True, "by_height")
        _bk.paste(_ava_img, (19, -13), True)
        _bk.text((100, 0), msg, font_color, "by_height")
        _bk.circle_corner(20)

        expeditions_text.paste(_bk, (25, h), True)
        h += 75 + 16

    expeditions_border.paste(expeditions_text, center_type="center")

    A.paste(expeditions_border, (550, 45))

    return image(b64=A.pic2bs4())