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

from typing import Optional, Union
from nonebot.adapters.onebot.v11 import MessageSegment
from configs.config import Config
from asyncio.exceptions import TimeoutError
from services.log import logger
from configs.path_config import IMAGE_PATH
from utils.image_utils import BuildImage
from utils.http_utils import AsyncHttpx
from utils.utils import get_user_avatar
from utils.message_builder import image
from .._utils import get_ds
from .._models import Genshin
from io import BytesIO
from nonebot import Driver
import asyncio
import nonebot


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/07/24/28d9338c7da4bcb2.png",
            "https://genshin.honeyhunterworld.com/img/gadget/i_3016.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) -> "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.get_user_cookie(int(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())