mirror of
https://github.com/A-Minos/nonebot-plugin-tetris-stats.git
synced 2026-03-05 05:36:54 +08:00
212 lines
9.4 KiB
Python
212 lines
9.4 KiB
Python
from nonebot import on_regex
|
||
from nonebot.adapters.onebot.v11 import GROUP, MessageEvent
|
||
from nonebot.matcher import Matcher
|
||
from nonebot.log import logger
|
||
|
||
from typing import Any, Mapping
|
||
from asyncio import gather
|
||
from re import I
|
||
|
||
from ..Utils.Request import request
|
||
|
||
from ..Utils.MessageAnalyzer import handleBindMessage, handleStatsQueryMessage
|
||
from ..Utils.SQL import queryBindInfo, writeBindInfo
|
||
|
||
ioBind = on_regex(pattern=r'^io绑定|^iobind', flags=I, permission=GROUP)
|
||
ioStats = on_regex(pattern=r'^io查|^iostats', flags=I, permission=GROUP)
|
||
|
||
|
||
@ioBind.handle()
|
||
async def _(event: MessageEvent, matcher: Matcher):
|
||
decodedMessage = await handleBindMessage(message=event.raw_message, gameType='IO')
|
||
if decodedMessage[0] is None:
|
||
await matcher.finish(decodedMessage[1][0])
|
||
if decodedMessage[0] == 'ID':
|
||
userIDStats = await checkUserID(userID=decodedMessage[1][1])
|
||
if userIDStats[0] is False:
|
||
await matcher.finish(userIDStats[1])
|
||
else:
|
||
userID = decodedMessage[1][1]
|
||
elif decodedMessage[0] == 'Name':
|
||
userData = await getUserData(userName=decodedMessage[1][1])
|
||
if userData[0] is False:
|
||
await matcher.finish('用户信息请求失败')
|
||
elif userData[1] is False:
|
||
await matcher.finish(f'用户信息请求错误:\n{userData[2]["error"]}')
|
||
else:
|
||
userID = await getUserID(userData=userData[2])
|
||
if event.sender.user_id is None: # 理论上是不会有None出现的，ide快乐行属于是（
|
||
logger.error('获取QQ号失败')
|
||
await matcher.finish('获取QQ号失败')
|
||
await matcher.finish(await writeBindInfo(QQNumber=event.sender.user_id, user=userID, gameType='IO'))
|
||
|
||
|
||
@ioStats.handle()
|
||
async def _(event: MessageEvent, matcher: Matcher):
|
||
decodedMessage = await handleStatsQueryMessage(message=event.raw_message, gameType='IO')
|
||
if decodedMessage[0] is None:
|
||
await matcher.finish(decodedMessage[1][0])
|
||
elif decodedMessage[0] == 'AT':
|
||
if event.is_tome() is True:
|
||
await matcher.finish(message='不能查询bot的信息')
|
||
bindInfo = await queryBindInfo(QQNumber=decodedMessage[1][1], gameType='IO')
|
||
if bindInfo is None:
|
||
message = '未查询到绑定信息'
|
||
else:
|
||
message = (f'* 由于无法验证绑定信息，不能保证查询到的用户为本人\n{await generateMessage(userID=bindInfo)}')
|
||
elif decodedMessage[0] == 'ME':
|
||
if event.sender.user_id is None:
|
||
logger.error('获取QQ号失败')
|
||
await matcher.finish('获取QQ号失败，请联系bot主人')
|
||
bindInfo = await queryBindInfo(QQNumber=event.sender.user_id, gameType='IO')
|
||
if bindInfo is None:
|
||
message = '未查询到绑定信息'
|
||
else:
|
||
message = (f'* 由于无法验证绑定信息，不能保证查询到的用户为本人\n{await generateMessage(userID=bindInfo)}')
|
||
elif decodedMessage[0] == 'ID':
|
||
message = await generateMessage(userID=decodedMessage[1][1])
|
||
elif decodedMessage[0] == 'Name':
|
||
message = await generateMessage(userName=decodedMessage[1][1])
|
||
await matcher.finish(message=message)
|
||
|
||
|
||
async def getUserData(userName: str = None, userID: str = None) -> tuple[bool, bool, dict[str, Any]]:
|
||
# 获取用户数据
|
||
if userName is not None and userID is None:
|
||
userDataUrl = f'https://ch.tetr.io/api/users/{userName}'
|
||
elif userName is None and userID is not None:
|
||
userDataUrl = f'https://ch.tetr.io/api/users/{userID}'
|
||
else:
|
||
raise ValueError(
|
||
'[TETRIS STATS] IODataProcessing.getUserData: 预期外行为，请上报GitHub')
|
||
return await request(Url=userDataUrl)
|
||
|
||
|
||
async def getSoloData(userName: str = None, userID: str = None) -> tuple[bool, bool, dict[str, Any]]:
|
||
# 获取Solo数据
|
||
if userName is not None and userID is None:
|
||
userSoloUrl = f'https://ch.tetr.io/api/users/{userName}/records'
|
||
elif userName is None and userID is not None:
|
||
userSoloUrl = f'https://ch.tetr.io/api/users/{userID}/records'
|
||
else:
|
||
raise ValueError(
|
||
'[TETRIS STATS] IODataProcessing.getSoloData: 预期外行为，请上报GitHub')
|
||
return await request(Url=userSoloUrl)
|
||
|
||
|
||
async def getUserID(userData: dict) -> str:
|
||
return userData['data']['user']['_id']
|
||
|
||
|
||
async def checkUserID(userID: str) -> tuple[bool, str]:
|
||
userData = await getUserData(userID=userID)
|
||
if userData[0] is False:
|
||
return (False, '用户信息请求失败')
|
||
elif userData[1] is False:
|
||
return (False, f'用户信息请求错误:\n{userData[2]["error"]}')
|
||
elif userID == userData[2]['data']['user']['_id']:
|
||
return (True, '')
|
||
else:
|
||
raise ValueError(
|
||
'[TETRIS STATS] IODataProcessing.checkUserID: 服务器返回的userID和用户提供的不一致，这种情况理论上不应该发生，以防万一还是写一下（x')
|
||
|
||
|
||
async def getLeagueStats(userData: dict) -> dict[str, Any]:
|
||
# 获取排位统计数据
|
||
league = userData['data']['user']['league']
|
||
leagueStats: dict[str, Any] = {}
|
||
if league['gamesplayed'] == 0:
|
||
leagueStats['Played'] = False
|
||
else:
|
||
leagueStats['Played'] = True
|
||
leagueStats['PPS'] = league['pps']
|
||
leagueStats['APM'] = league['apm']
|
||
leagueStats['VS'] = 0 if league['vs'] is None else league['vs']
|
||
leagueStats['Rank'] = False if league['rank'] == 'z' else league['rank'].upper()
|
||
if league['rating'] != -1:
|
||
leagueStats['Ranked'] = True
|
||
leagueStats['Rating'] = round(league['rating'], 2)
|
||
leagueStats['Glicko'] = round(league['glicko'], 2)
|
||
leagueStats['RD'] = round(league['rd'], 2)
|
||
else:
|
||
leagueStats['Ranked'] = False
|
||
leagueStats['Standing'] = league['standing']
|
||
leagueStats['LPM'] = round((league['pps'] * 24), 2)
|
||
leagueStats['APL'] = round(
|
||
(leagueStats['APM'] / leagueStats['LPM']), 2)
|
||
leagueStats['ADPM'] = round((leagueStats['VS'] * 0.6), 2)
|
||
leagueStats['ADPL'] = round(
|
||
(leagueStats['ADPM'] / leagueStats['LPM']), 2)
|
||
return leagueStats
|
||
|
||
|
||
async def getSprintStats(soloData: dict) -> Mapping[str, bool | int | float]:
|
||
# 获取40L统计数据
|
||
sprintStats = {}
|
||
if soloData['data']['records']['40l']['record'] is None:
|
||
sprintStats['Played'] = False
|
||
else:
|
||
sprintStats['Played'] = True
|
||
if soloData['data']['records']['40l']['rank'] is None:
|
||
sprintStats['Rank'] = False
|
||
else:
|
||
sprintStats['Rank'] = soloData['data']['records']['40l']['rank']
|
||
sprintStats['Time'] = round(
|
||
soloData['data']['records']['40l']['record']['endcontext']['finalTime'] / 1000, 2)
|
||
return sprintStats
|
||
|
||
|
||
async def getBlitzStats(soloData: dict) -> dict[str, Any]:
|
||
# 获取Blitz统计数据
|
||
blitzStats = {}
|
||
if soloData['data']['records']['blitz']['record'] is None:
|
||
blitzStats['Played'] = False
|
||
else:
|
||
blitzStats['Played'] = True
|
||
if soloData['data']['records']['blitz']['rank'] is None:
|
||
blitzStats['Rank'] = False
|
||
else:
|
||
blitzStats['Rank'] = soloData['data']['records']['blitz']['rank']
|
||
blitzStats['Score'] = soloData['data']['records']['blitz']['record']['endcontext']['score']
|
||
return blitzStats
|
||
|
||
|
||
async def generateMessage(userName: str = None, userID: str = None) -> str:
|
||
# 生成消息
|
||
userData, soloData = await gather(getUserData(userName=userName, userID=userID), getSoloData(userName=userName, userID=userID))
|
||
if userData[0] is False:
|
||
return '用户信息请求失败'
|
||
elif userData[1] is False:
|
||
return f'用户信息请求错误:\n{userData[2]["error"]}'
|
||
userName = userData[2]['data']['user']['username'].upper()
|
||
leagueStats = await getLeagueStats(userData[2])
|
||
message = ''
|
||
if leagueStats['Played'] is False:
|
||
message += f'用户 {userName} 没有排位统计数据'
|
||
else:
|
||
if leagueStats['Rank'] is False and leagueStats['Ranked'] is False:
|
||
message += f'用户 {userName} 暂未完成定级赛'
|
||
elif leagueStats['Rank'] is False and leagueStats['Ranked'] is True:
|
||
message += f'用户 {userName} 暂无段位, {leagueStats["Rating"]} TR'
|
||
else:
|
||
message += f'{leagueStats["Rank"]} 段用户 {userName} {leagueStats["Rating"]} TR (#{leagueStats["Standing"]})'
|
||
message += f', 段位分 {leagueStats["Glicko"]}±{leagueStats["RD"]}, 最近十场的数据:' if leagueStats['Ranked'] is True else ', 最近十场的数据:'
|
||
message += f'\nL\'PM: {leagueStats["LPM"]} ( {leagueStats["PPS"]} pps )'
|
||
message += f'\nAPM: {leagueStats["APM"]} ( x{leagueStats["APL"]} )'
|
||
if leagueStats["VS"] != 0:
|
||
message += f'\nADPM: {leagueStats["ADPM"]} ( x{leagueStats["ADPL"]} ) ( {leagueStats["VS"]}vs )'
|
||
if soloData[0] is False:
|
||
return f'{message}\nSolo统计数据请求失败'
|
||
elif soloData[1] is False:
|
||
return f'{message}\nSolo统计数据请求错误:\n{soloData[2]["error"]}'
|
||
sprintStats, blitzStats = await gather(getSprintStats(soloData[2]), getBlitzStats(soloData[2]))
|
||
if sprintStats['Played'] is True:
|
||
message += f'\n40L: {sprintStats["Time"]}s'
|
||
if sprintStats['Rank'] is not False:
|
||
message += f' ( #{sprintStats["Rank"]} )'
|
||
if blitzStats['Played'] is True:
|
||
message += f'\nBlitz: {blitzStats["Score"]}'
|
||
if blitzStats['Rank'] is not False:
|
||
message += f' ( #{blitzStats["Rank"]} )'
|
||
return message
|