zhenxun_bot/zhenxun/builtin_plugins/sign_in/utils.py

from datetime import datetime
import os
from pathlib import Path
import random

import aiofiles
import nonebot
from nonebot.drivers import Driver
from nonebot_plugin_uninfo import Uninfo
import pytz

from zhenxun import ui
from zhenxun.configs.config import BotConfig, Config
from zhenxun.models.sign_log import SignLog
from zhenxun.models.sign_user import SignUser
from zhenxun.utils.manager.priority_manager import PriorityLifecycle
from zhenxun.utils.platform import PlatformUtils

from .config import (
    SIGN_TODAY_CARD_PATH,
    level2attitude,
    lik2level,
    lik2relation,
)

assert (
    len(level2attitude) == len(lik2level) == len(lik2relation)
), "好感度态度、等级、关系长度不匹配！"

AVA_URL = "http://q1.qlogo.cn/g?b=qq&nk={}&s=160"

driver: Driver = nonebot.get_driver()

base_config = Config.get("sign_in")


MORNING_MESSAGE = [
    "早上好，希望今天是美好的一天！",
    "醒了吗，今天也要元气满满哦！",
    "早上好呀，今天也要开心哦！",
    "早安，愿你拥有美好的一天！",
]

LG_MESSAGE = [
    "今天要早点休息哦~",
    "可不要熬夜到太晚呀",
    "请尽早休息吧！",
    "不要熬夜啦！",
]


@PriorityLifecycle.on_startup(priority=5)
async def init_image():
    SIGN_TODAY_CARD_PATH.mkdir(exist_ok=True, parents=True)
    clear_sign_data_pic()


async def get_card(
    user: SignUser,
    session: Uninfo,
    nickname: str,
    add_impression: float,
    gold: int | None,
    gift: str,
    is_double: bool = False,
    is_card_view: bool = False,
) -> Path:
    """获取好感度卡片

    参数:
        user: SignUser
        session: Uninfo
        nickname: 用户昵称
        impression: 新增的好感度
        gold: 金币
        gift: 礼物
        is_double: 是否触发双倍.
        is_card_view: 是否展示好感度卡片.

    返回:
        Path: 卡片路径
    """
    user_id = user.user_id
    date = datetime.now().date()
    _type = "view" if is_card_view else "sign"
    file_name = f"{user_id}_{_type}_{date}.png"
    card_file = SIGN_TODAY_CARD_PATH / file_name

    if card_file.exists():
        return card_file

    if add_impression == -1:
        view_name = f"{user_id}_view_{date}.png"
        view_card_file = SIGN_TODAY_CARD_PATH / view_name
        if view_card_file.exists():
            return view_card_file
        is_card_view = True

    return await _generate_html_card(
        user, session, nickname, add_impression, gold, gift, is_double, is_card_view
    )


def get_level_and_next_impression(impression: float) -> tuple[int, int | float, int]:
    """获取当前好感等级与下一等级的差距

    参数:
        impression: 好感度

    返回:
        tuple[int, int, int]: 好感度等级，下一等级好感度要求，已达到的好感度要求
    """

    keys = list(lik2level.keys())
    level_int, next_impression, previous_impression = (
        int(lik2level[keys[-1]]),
        keys[-2],
        keys[-1],
    )
    for i in range(len(keys)):
        if impression >= keys[i]:
            level_int, next_impression, previous_impression = (
                int(lik2level[keys[i]]),
                keys[i - 1],
                keys[i],
            )
            if i == 0:
                next_impression = impression
            break
    return level_int, next_impression, previous_impression


def clear_sign_data_pic():
    """
    清空当前签到图片数据
    """
    date = datetime.now().date()
    for file in os.listdir(SIGN_TODAY_CARD_PATH):
        if str(date) not in file:
            os.remove(SIGN_TODAY_CARD_PATH / file)


async def _generate_html_card(
    user: SignUser,
    session: Uninfo,
    nickname: str,
    add_impression: float,
    gold: int | None,
    gift: str,
    is_double: bool = False,
    is_card_view: bool = False,
) -> Path:
    """使用渲染服务生成签到卡片

    参数:
        user: SignUser
        session: Uninfo
        nickname: 用户昵称
        add_impression: 新增的好感度
        gold: 金币
        gift: 礼物
        is_double: 是否触发双倍.
        is_card_view: 是否为卡片视图.

    返回:
        Path: 卡片路径
    """
    now = datetime.now()
    date = now.date()
    _type = "view" if is_card_view else "sign"
    file_name = f"{user.user_id}_{_type}_{date}.png"
    card_file = SIGN_TODAY_CARD_PATH / file_name

    if card_file.exists():
        return card_file

    impression = float(user.impression)
    user_console = await user.user_console
    uid_str = (
        f"{user_console.uid:08}"
        if user_console and user_console.uid is not None
        else "XXXXXXXX"
    )
    uid_formatted = f"{uid_str[:4]} {uid_str[4:]}"

    level, next_impression, previous_impression = get_level_and_next_impression(
        impression
    )

    attitude = level2attitude.get(str(level), "未知")
    interpolation_val = max(0, next_impression - impression)
    interpolation = f"{interpolation_val:.2f}"

    denominator = next_impression - previous_impression
    progress = (
        100.0
        if denominator == 0
        else min(100.0, ((impression - previous_impression) / denominator) * 100)
    )

    hour = now.hour
    if 6 < hour < 10:
        bot_message = random.choice(MORNING_MESSAGE)
    elif 0 <= hour < 6:
        bot_message = random.choice(LG_MESSAGE)
    else:
        bot_message = f"{BotConfig.self_nickname}希望你开心！"

    temperature = random.randint(1, 40)
    weather_icon_name = f"{random.randint(0, 11)}.png"
    tag_icon_name = f"{random.randint(0, 5)}.png"

    user_info = {
        "nickname": nickname,
        "uid_str": uid_formatted,
        "avatar_url": PlatformUtils.get_user_avatar_url(
            user.user_id, PlatformUtils.get_platform(session), session.self_id
        )
        or "",
        "sign_count": user.sign_count,
    }

    favorability_info = {
        "current": impression,
        "level": level,
        "next_level_at": next_impression,
        "previous_level_at": previous_impression,
    }

    reward_info = None
    rank = None
    total_gold = None
    last_sign_date_str = None

    if is_card_view:
        value_list = (
            await SignUser.annotate()
            .order_by("-impression")
            .values_list("user_id", flat=True)
        )
        rank = value_list.index(user.user_id) + 1 if user.user_id in value_list else 0
        total_gold = user_console.gold if user_console else 0

        last_log = (
            await SignLog.filter(user_id=user.user_id).order_by("-create_time").first()
        )
        last_date = "从未"
        if last_log:
            last_date = str(
                last_log.create_time.astimezone(pytz.timezone("Asia/Shanghai")).date()
            )
        last_sign_date_str = f"上次签到：{last_date}"

    else:
        reward_info = {
            "impression_added": add_impression,
            "gold_added": gold or 0,
            "gift_received": gift,
            "is_double": is_double,
        }

    page_info = {
        "date_str": str(now.replace(microsecond=0)),
        "weather_icon_name": weather_icon_name,
        "temperature": temperature,
        "tag_icon_name": tag_icon_name,
    }

    card_data = {
        "is_card_view": is_card_view,
        "user": user_info,
        "favorability": favorability_info,
        "reward": reward_info,
        "page": page_info,
        "bot_message": bot_message,
        "attitude": attitude,
        "interpolation": interpolation,
        "progress": progress,
        "rank": rank,
        "total_gold": total_gold,
        "last_sign_date_str": last_sign_date_str,
    }

    image_bytes = await ui.render_template("pages/builtin/sign", data=card_data)

    async with aiofiles.open(card_file, "wb") as f:
        await f.write(image_bytes)

    return card_file