nonebot-plugin-tetris-stats/nonebot_plugin_tetris_stats/games/tetrio/query/v2.py

from asyncio import gather
from datetime import datetime, timedelta
from hashlib import md5

from yarl import URL

from ....utils.host import HostPage, get_self_netloc
from ....utils.metrics import get_metrics
from ....utils.render import render
from ....utils.render.schemas.base import Avatar
from ....utils.render.schemas.tetrio.user.info_v2 import (
    Badge,
    Blitz,
    Info,
    Sprint,
    Statistic,
    TetraLeague,
    TetraLeagueStatistic,
    User,
    Zen,
)
from ....utils.screenshot import screenshot
from ..api import Player
from ..api.schemas.summaries.league import NeverPlayedData, NeverRatedData
from .tools import flow_to_history, handling_special_value


async def make_query_image_v2(player: Player) -> bytes:
    (
        (user, user_info, league, sprint, blitz, zen),
        (avatar_revision, banner_revision, leagueflow),
    ) = await gather(
        gather(player.user, player.get_info(), player.league, player.sprint, player.blitz, player.zen),
        gather(player.avatar_revision, player.banner_revision, player.get_leagueflow()),
    )
    if sprint.data.record is not None:
        duration = timedelta(milliseconds=sprint.data.record.results.stats.finaltime).total_seconds()
        sprint_value = f'{duration:.3f}s' if duration < 60 else f'{duration // 60:.0f}m {duration % 60:.3f}s'  # noqa: PLR2004
    else:
        sprint_value = 'N/A'

    play_time: str | None
    if (game_time := handling_special_value(user_info.data.gametime)) is not None:
        if game_time // 3600 > 0:
            play_time = f'{game_time//3600:.0f}h {game_time % 3600 // 60:.0f}m {game_time % 60:.0f}s'
        elif game_time // 60 > 0:
            play_time = f'{game_time//60:.0f}m {game_time % 60:.0f}s'
        else:
            play_time = f'{game_time:.0f}s'
    else:
        play_time = game_time
    netloc = get_self_netloc()
    async with (
        HostPage(
            await render(
                'v2/tetrio/user/info',
                Info(
                    user=User(
                        id=user.ID,
                        name=user.name.upper(),
                        bio=user_info.data.bio,
                        banner=str(
                            URL(f'http://{netloc}/host/resource/tetrio/banners/{user.ID}')
                            % {'revision': banner_revision}
                        )
                        if banner_revision is not None and banner_revision != 0
                        else None,
                        avatar=str(
                            URL(f'http://{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
                        ),
                        badges=[
                            Badge(
                                id=i.id,
                                description=i.label,
                                group=i.group,
                                receive_at=i.ts if isinstance(i.ts, datetime) else None,
                            )
                            for i in user_info.data.badges
                        ],
                        country=user_info.data.country,
                        role=user_info.data.role,
                        xp=user_info.data.xp,
                        friend_count=user_info.data.friend_count,
                        supporter_tier=user_info.data.supporter_tier,
                        bad_standing=user_info.data.badstanding or False,
                        playtime=play_time,
                        join_at=user_info.data.ts,
                    ),
                    tetra_league=TetraLeague(
                        rank=league.data.rank,
                        highest_rank='z' if isinstance(league.data, NeverRatedData) else league.data.bestrank,
                        tr=round(league.data.tr, 2),
                        glicko=round(league.data.glicko, 2),
                        rd=round(league.data.rd, 2),
                        global_rank=league.data.standing,
                        country_rank=league.data.standing_local,
                        pps=(metrics := get_metrics(pps=league.data.pps, apm=league.data.apm, vs=league.data.vs)).pps,
                        apm=metrics.apm,
                        apl=metrics.apl,
                        vs=metrics.vs,
                        adpl=metrics.adpl,
                        statistic=TetraLeagueStatistic(total=league.data.gamesplayed, wins=league.data.gameswon),
                        decaying=league.data.decaying,
                        history=flow_to_history(leagueflow, lambda x: x[-100:]),
                    )
                    if not isinstance(league.data, NeverPlayedData)
                    else None,
                    statistic=Statistic(
                        total=handling_special_value(user_info.data.gamesplayed),
                        wins=handling_special_value(user_info.data.gameswon),
                    ),
                    sprint=Sprint(
                        time=sprint_value,
                        global_rank=sprint.data.rank,
                        play_at=sprint.data.record.ts,
                    )
                    if sprint.data.record is not None
                    else None,
                    blitz=Blitz(
                        score=blitz.data.record.results.stats.score,
                        global_rank=blitz.data.rank,
                        play_at=blitz.data.record.ts,
                    )
                    if blitz.data.record is not None
                    else None,
                    zen=Zen(level=zen.data.level, score=zen.data.score),
                ),
            ),
        ) as page_hash
    ):
        return await screenshot(f'http://{netloc}/host/{page_hash}.html')