nonebot-plugin-tetris-stats/nonebot_plugin_tetris_stats/games/tetrio/bind.py

from asyncio import gather
from hashlib import md5
from secrets import choice

from arclet.alconna import Arg, ArgFlag
from nonebot_plugin_alconna import Args, Subcommand
from nonebot_plugin_alconna.uniseg import UniMessage
from nonebot_plugin_orm import get_session
from nonebot_plugin_uninfo import QryItrface, Uninfo
from nonebot_plugin_uninfo import User as UninfoUser
from nonebot_plugin_uninfo.orm import get_session_persist_id
from nonebot_plugin_user import User
from yarl import URL

from ...config.config import global_config
from ...db import BindStatus, create_or_update_bind, trigger
from ...i18n import Lang
from ...utils.host import get_self_netloc
from ...utils.image import get_avatar
from ...utils.lang import get_lang
from ...utils.render import render_image
from ...utils.render.schemas.base import Avatar, People
from ...utils.render.schemas.bind import Bind
from . import alc, command, get_player
from .api import Player
from .constant import GAME_TYPE

command.add(
    Subcommand(
        'bind',
        Args(
            Arg(
                'account',
                get_player,
                notice='TETR.IO 用户名 / ID',
                flags=[ArgFlag.HIDDEN],
            )
        ),
        help_text='绑定 TETR.IO 账号',
    )
)

alc.shortcut(
    '(?i:io)(?i:绑定|绑|bind)',
    command='tstats TETR.IO bind',
    humanized='io绑定',
)

try:
    from nonebot.adapters.discord import MessageCreateEvent

    @alc.assign('TETRIO.bind')
    async def _(_: MessageCreateEvent, nb_user: User, account: Player, event_session: Uninfo, interface: QryItrface):
        async with trigger(
            session_persist_id=await get_session_persist_id(event_session),
            game_platform=GAME_TYPE,
            command_type='bind',
            command_args=[],
        ):
            user, user_info = await gather(account.user, account.get_info())
            verify = (
                user_info.data.connections.discord is not None
                and user_info.data.connections.discord.id == event_session.user.id
            )
            async with get_session() as session:
                bind_status = await create_or_update_bind(
                    session=session,
                    user=nb_user,
                    game_platform=GAME_TYPE,
                    game_account=user.unique_identifier,
                    verify=verify,
                )
            if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
                await UniMessage.image(
                    raw=await make_bind_image(
                        player=account,
                        event_session=event_session,
                        interface=interface,
                        verify=verify,
                    )
                ).finish()
except ImportError:
    pass


@alc.assign('TETRIO.bind')
async def _(nb_user: User, account: Player, event_session: Uninfo, interface: QryItrface):
    async with trigger(
        session_persist_id=await get_session_persist_id(event_session),
        game_platform=GAME_TYPE,
        command_type='bind',
        command_args=[],
    ):
        user = await account.user
        async with get_session() as session:
            bind_status = await create_or_update_bind(
                session=session,
                user=nb_user,
                game_platform=GAME_TYPE,
                game_account=user.unique_identifier,
            )
        if bind_status in (BindStatus.SUCCESS, BindStatus.UPDATE):
            await UniMessage.image(
                raw=await make_bind_image(
                    player=account,
                    event_session=event_session,
                    interface=interface,
                    verify=None,
                )
            ).finish()


async def make_bind_image(
    player: Player, event_session: Uninfo, interface: QryItrface, *, verify: bool | None = None
) -> bytes:
    (user, avatar_revision) = await gather(player.user, player.avatar_revision)
    return await render_image(
        Bind(
            platform='TETR.IO',
            type='unknown' if verify is None else 'success' if verify else 'unverified',
            user=People(
                avatar=str(
                    URL(f'http://{get_self_netloc()}/host/resource/tetrio/avatars/{user.ID}')
                    % {'revision': avatar_revision}
                )
                if avatar_revision is not None and avatar_revision != 0
                else Avatar(type='identicon', hash=md5(user.ID.encode()).hexdigest()),  # noqa: S324
                name=user.name.upper(),
            ),
            bot=People(
                avatar=await get_avatar(
                    (
                        bot_user := await interface.get_user(event_session.self_id)
                        or UninfoUser(id=event_session.self_id)
                    ),
                    'Data URI',
                    '../../static/logo/logo.svg',
                ),
                name=bot_user.nick or bot_user.name or choice(list(global_config.nickname) or ['bot']),
            ),
            prompt=Lang.prompt.io_check(),
            lang=get_lang(),
        ),
    )