lixiangwuxian/qq_bot
0
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 在 scoresaber 模块中添加 ss+n 命令，允许用户查询提升排名所需的分数，并优化 FetchPlayerData 函数的返回类型以提高错误处理能力

Browse Source
This commit is contained in:
lixiangwuxian 2025-03-31 02:28:06 +08:00
parent cfe6c177e1
commit 6454b55a90
3 changed files with 219 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
handler/scoresaber/score.go
View File

@ -1,6 +1,7 @@
package scoresaber
import (
"fmt"
"log"
"os"
"strconv"
@ -25,8 +26,75 @@ func init() {
handler.RegisterHandler("最新ss", getMyRecentScore, constants.LEVEL_USER)
handler.RegisterHelpInform("最新ss", "scoresaber", "查看您的最新游戏记录")
handler.RegisterHandler("截ss", screenshotSS, constants.LEVEL_USER)
handler.RegisterHelpInform("截ss", "scoresaber", "scoresaber主页截图")
handler.RegisterHandler("jss", screenshotSS, constants.LEVEL_USER)
handler.RegisterHelpInform("ss+n", "scoresaber", "区排名升高n位还需要打出多少pp")
handler.RegisterHandler("ss+n", ssPlusN, constants.LEVEL_USER)
// handler.RegisterHelpInform("截ss", "scoresaber", "scoresaber主页截图")
// handler.RegisterHandler("jss", screenshotSS, constants.LEVEL_USER)
}
func ssPlusN(msg model.Message) (reply model.Reply) {
var (
resultStr strings.Builder
err error
maxRetries = 5 // 最大重试次数
attempts = 0
)
var N int
if len(msg.RawMsg) > len("ss+") {
N, err = strconv.Atoi(msg.RawMsg[len("ss+"):])
if err != nil {
return model.Reply{
ReplyMsg: "请输入一个整数",
ReferOriginMsg: true,
FromMsg: msg,
}
}
}
// 获取当前用户在区中的排名
userIdStr := strconv.Itoa(int(msg.UserId))
var userInfo scoresaber.PlayerData
for attempts < maxRetries {
err = nil
userInfo, err = scoresaber.FetchPlayerData(userIdStr)
if err != nil {
break
}
attempts++
}
if err != nil {
return model.Reply{
ReplyMsg: "获取您的分数时出现问题，请稍后重试。" + err.Error(),
ReferOriginMsg: true,
FromMsg: msg,
}
}
resultStr.WriteString(fmt.Sprintf("您当前的区排名为：%d\n", userInfo.CountryRank))
// 获取当前用户所在区对应+N位的玩家列表
leaderboard, err := scoresaber.FetchCountryLeaderboard(userInfo.Country, userInfo.CountryRank-N, userInfo.ID)
if err != nil {
return model.Reply{
ReplyMsg: "获取您的分数时出现问题，请稍后重试。" + err.Error(),
ReferOriginMsg: true,
FromMsg: msg,
}
}
if userInfo.CountryRank-N < 0 {
resultStr.WriteString(fmt.Sprintf("注意：你最多只需要提升%d名就是%s区Top1了\n", N, userInfo.Country))
}
//寻找leaderboard中排名为userInfo.CountryRank-N的玩家
var targetPlayer scoresaber.PlayerData
for _, player := range leaderboard.Players {
if player.CountryRank == userInfo.CountryRank-N {
targetPlayer = player
break
}
}
resultStr.WriteString(fmt.Sprintf("您只需要再打出%.2fpp就能达到%s区第%d名了", targetPlayer.PP-userInfo.PP, userInfo.Country, targetPlayer.CountryRank))
return model.Reply{
ReplyMsg: resultStr.String(),
ReferOriginMsg: true,
FromMsg: msg,
}
}
func getSSProfile(msg model.Message) (reply model.Reply) {

15
service/scoresaber/bind_ss.go
View File

@ -90,7 +90,7 @@ func (ss *ssQuery) BindSS(qqId string, ssId string) (reply string) {
return "ssId格式错误,应当为一串数字(是您的scoresaber主页链接中的末尾数字部分,一般和您的steamID相同)"
}
data, err := FetchPlayerData(ssId)
if data == nil {
if data.ID == "" {
if err != nil {
return "请求出错，报错如下，如果确定命令没问题可以重新试试:" + err.Error()
}
@ -106,7 +106,7 @@ func (ss *ssQuery) BindSS(qqId string, ssId string) (reply string) {
}
rows.Close()
// 获取当前绑定账号的信息
if currentData, err := FetchPlayerData(currentSsId); err == nil && currentData != nil {
if currentData, err := FetchPlayerData(currentSsId); err == nil && currentData.ID != "" {
return fmt.Sprintf("您已绑定至ss账号：%s,请先输入\"解绑ss\"解绑", currentData.Name)
}
return "您已绑定过ss账号,请先输入\"解绑ss\"解绑"
@ -167,10 +167,12 @@ func (ss *ssQuery) GetScore(ssId string) (reply string, err error) {
// 查询玩家数据
data, err := FetchPlayerData(ssId)
if data == nil {
if err != nil {
return "查询出错，报错如下" + err.Error(), errors.New("查询出错，报错如下" + err.Error())
}
if data.ID == "" {
return "未找到玩家,请检查ID后重试", errors.New("未找到玩家,请检查ID后重试")
}
// 构建 PlayerDataLite 结构体
dataLite := PlayerDataLite{
ID: data.ID,
@ -232,9 +234,12 @@ func (ss *ssQuery) GetScore(ssId string) (reply string, err error) {
func (ss *ssQuery) GetScoreWithoutUpdate(ssId string) (reply string, err error) {
// 查询玩家数据
data, err := FetchPlayerData(ssId)
if data == nil {
if err != nil {
return "查询出错，报错如下" + err.Error(), errors.New("查询出错，报错如下" + err.Error())
}
if data.ID == "" {
return "未找到玩家,请检查ID后重试", errors.New("未找到玩家,请检查ID后重试")
}
// 返回当前数据的字符串表示
return data.ToString(), nil
}

145
service/scoresaber/user_info.go
View File

@ -9,13 +9,13 @@ import (
)
// fetchPlayerData 函数请求 Scoresaber API，并解析完整的玩家信息
func FetchPlayerData(ssID string) (*PlayerData, error) {
func FetchPlayerData(ssID string) (PlayerData, error) {
url := fmt.Sprintf("https://scoresaber.com/api/player/%s/full", ssID)
// 创建请求
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return nil, err
return PlayerData{}, err
}
// 设置请求头
@ -41,7 +41,7 @@ func FetchPlayerData(ssID string) (*PlayerData, error) {
resp, err = client.Do(req)
}
if err != nil {
return nil, err
return PlayerData{}, err
}
defer resp.Body.Close()
@ -50,7 +50,7 @@ func FetchPlayerData(ssID string) (*PlayerData, error) {
if resp.Header.Get("Content-Encoding") == "gzip" {
reader, err = gzip.NewReader(resp.Body)
if err != nil {
return nil, err
return PlayerData{}, err
}
defer reader.(*gzip.Reader).Close()
} else {
@ -62,8 +62,141 @@ func FetchPlayerData(ssID string) (*PlayerData, error) {
err = json.NewDecoder(reader).Decode(&playerData)
if err != nil {
// log.Printf("got body %v", reader.)
return nil, err
return PlayerData{}, err
}
return &playerData, nil
return playerData, nil
}
// {
// "players": [
// {
// "id": "76561199001767132",
// "name": "ViSi",
// "profilePicture": "https://cdn.scoresaber.com/avatars/76561199001767132.jpg",
// "bio": "<p>a girl using index cons</p>\n<p>7/1/2022 i beat ov sacrament</p>\n<p>7/28/2022 i hit 1k hours, i suppose im the worst 1k hours player atm</p>\n<p><a href=\"https://space.bilibili.com/36819692?spm_id_from=333.1007.0.0\">BiliBili</a></p>\n<p><a href=\"https://www.youtube.com/channel/UCS8Ljbehybj7G2RHV7S-kKA\">YouTube</a></p>\n<p><a href=\"https://www.twitch.tv/visi_bs\">Twitch</a></p>\n<p><a href=\"https://twitter.com/VioletSilence1\">Twitter</a></p>",
// "country": "CN",
// "pp": 15113.41,
// "rank": 120,
// "countryRank": 1,
// "role": null,
// "badges": null,
// "histories": "103,103,104,104,104,104,104,105,104,104,104,104,105,105,105,106,107,107,107,108,111,112,113,113,113,113,113,113,113,115,114,115,115,114,114,114,118,118,118,118,119,119,119,119,119,119,121,121,121",
// "permissions": 0,
// "banned": false,
// "inactive": false,
// "scoreStats": {
// "totalScore": 4525384588,
// "totalRankedScore": 1648718461,
// "averageRankedAccuracy": 96.73215,
// "totalPlayCount": 4001,
// "rankedPlayCount": 1351,
// "replaysWatched": 482
// },
// "firstSeen": "2021-04-15T06:56:45.000Z"
// },
// ...
// {
// "id": "76561198796873048",
// "name": "Jia_Yue 甩尾の嘉玥",
// "profilePicture": "https://cdn.scoresaber.com/avatars/76561198796873048.jpg",
// "bio": null,
// "country": "CN",
// "pp": 9984.614,
// "rank": 2127,
// "countryRank": 50,
// "role": null,
// "badges": null,
// "histories": "2202,2170,2176,2182,2184,2183,2183,2224,2196,2197,2192,2210,2206,2210,2197,2195,2190,2186,2187,2187,2190,2191,2187,2190,2186,2186,2184,2183,2178,2202,2167,2163,2163,2151,2150,2145,2164,2132,2135,2130,2126,2128,2124,2120,2122,2125,2125,2129,2125",
// "permissions": 0,
// "banned": false,
// "inactive": false,
// "scoreStats": {
// "totalScore": 2377399561,
// "totalRankedScore": 1304174761,
// "averageRankedAccuracy": 82.02609,
// "totalPlayCount": 3001,
// "rankedPlayCount": 1410,
// "replaysWatched": 19
// },
// "firstSeen": "2022-03-24T06:11:27.000Z"
// }
// ],
// "metadata": {
// "total": 1927,
// "page": 1,
// "itemsPerPage": 50
// }
// }
type LeaderboardData struct {
Players []PlayerData `json:"players"`
Metadata struct {
Total int `json:"total"`
Page int `json:"page"`
ItemsPerPage int `json:"itemsPerPage"`
} `json:"metadata"`
}
func FetchCountryLeaderboard(country string, offset int, ssID string) (LeaderboardData, error) {
// 根据偏移量计算页数
if offset < 0 {
offset = 0
}
page := offset/50 + 1
url := fmt.Sprintf("https://scoresaber.com/api/players?countries=%s&page=%d", country, page)
// 创建请求
req, err := http.NewRequest("GET", url, nil)
if err != nil {
return LeaderboardData{}, err
}
// 设置请求头
req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:130.0) Gecko/20100101 Firefox/130.0")
req.Header.Set("Accept", "application/json, text/plain, */*")
req.Header.Set("Accept-Language", "zh-CN,zh;q=0.8,zh-TW;q=0.7,zh-HK;q=0.5,en-US;q=0.3,en;q=0.2")
req.Header.Set("Accept-Encoding", "gzip")
req.Header.Set("DNT", "1")
req.Header.Set("Connection", "keep-alive")
req.Header.Set("Referer", fmt.Sprintf("https://scoresaber.com/u/%s", ssID))
req.Header.Set("Sec-Fetch-Dest", "empty")
req.Header.Set("Sec-Fetch-Mode", "cors")
req.Header.Set("Sec-Fetch-Site", "same-origin")
req.Header.Set("Sec-GPC", "1")
req.Header.Set("Pragma", "no-cache")
req.Header.Set("Cache-Control", "no-cache")
req.Header.Set("TE", "trailers")
// 发送请求，失败则重试至多3次
client := &http.Client{}
resp, err := client.Do(req)
for i := 0; i < 3 && err != nil; i++ {
resp, err = client.Do(req)
}
if err != nil {
return LeaderboardData{}, err
}
defer resp.Body.Close()
// 处理压缩响应
var reader io.Reader
if resp.Header.Get("Content-Encoding") == "gzip" {
reader, err = gzip.NewReader(resp.Body)
if err != nil {
return LeaderboardData{}, err
}
defer reader.(*gzip.Reader).Close()
} else {
reader = resp.Body
}
// 解析响应体
var leaderboardData LeaderboardData
err = json.NewDecoder(reader).Decode(&leaderboardData)
if err != nil {
return LeaderboardData{}, err
}
return leaderboardData, nil
}