119 lines
2.9 KiB
Go
119 lines
2.9 KiB
Go
package scoresaber
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"git.lxtend.com/qqbot/handler"
|
|
"git.lxtend.com/qqbot/model"
|
|
"git.lxtend.com/qqbot/service/scoresaber"
|
|
)
|
|
|
|
func init() {
|
|
handler.RegisterHandler("查ss", getMySS)
|
|
handler.RegisterHandler("绑定ss", bindSS)
|
|
handler.RegisterHandler("解绑ss", unbindSS)
|
|
handler.RegisterHandler("最新ss", getMyRecentScore)
|
|
handler.RegisterHandler("最热ss", getRecentScore)
|
|
}
|
|
|
|
func getMySS(msg model.Message) (reply model.Reply) {
|
|
resultStr := ""
|
|
var err error
|
|
for ; err != nil; resultStr, err = scoresaber.SSQuery.GetScore(strconv.Itoa(int(msg.UserId))) {
|
|
|
|
}
|
|
return model.Reply{
|
|
ReplyMsg: resultStr,
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
|
|
func bindSS(msg model.Message) (reply model.Reply) {
|
|
return model.Reply{
|
|
ReplyMsg: scoresaber.SSQuery.BindSS(strconv.Itoa(int(msg.UserId)), msg.Msg[len("绑定ss "):]),
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
|
|
func unbindSS(msg model.Message) (reply model.Reply) {
|
|
return model.Reply{
|
|
ReplyMsg: scoresaber.SSQuery.UnbindSS(strconv.Itoa(int(msg.UserId))),
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
|
|
func getRecentScore(msg model.Message) (reply model.Reply) {
|
|
count := 1
|
|
if len(msg.Msg) > len("最热ss ") {
|
|
var err error
|
|
count, err = strconv.Atoi(msg.Msg[len("最热ss "):])
|
|
if err != nil || count <= 0 {
|
|
return model.Reply{
|
|
ReplyMsg: "",
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
}
|
|
scoreMsg := ""
|
|
for _, v := range scoresaber.ScoresManager.GetRecentScores(count, " WHERE country = 'CN' ") {
|
|
scoreMsg += v.ToString() + "\n\n"
|
|
}
|
|
if len(scoreMsg) > 0 {
|
|
scoreMsg = scoreMsg[:len(scoreMsg)-len("\n\n")]
|
|
}
|
|
return model.Reply{
|
|
ReplyMsg: scoreMsg,
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
|
|
func getMyRecentScore(msg model.Message) (reply model.Reply) {
|
|
count := 1
|
|
scoreMsg := ""
|
|
if len(msg.Msg) > len("最新ss ") {
|
|
var err error
|
|
count, err = strconv.Atoi(msg.Msg[len("最新ss "):])
|
|
if err != nil || count <= 0 {
|
|
return model.Reply{
|
|
ReplyMsg: "",
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
}
|
|
var userName string
|
|
recordCount := 0
|
|
records, err := scoresaber.SSQuery.GetRecentScores(count, strconv.Itoa(int(msg.UserId)))
|
|
if err != nil {
|
|
return model.Reply{
|
|
ReplyMsg: err.Error(),
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
for _, v := range records {
|
|
scoreMsg += v.ToString() + "\n\n"
|
|
userName = v.Name
|
|
recordCount++
|
|
}
|
|
if len(scoreMsg) > 0 {
|
|
scoreMsg = scoreMsg[:len(scoreMsg)-len("\n\n")] //去掉最后一个换行符
|
|
} else {
|
|
return model.Reply{
|
|
ReplyMsg: "无最近游戏记录",
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|
|
return model.Reply{
|
|
ReplyMsg: "玩家 " + userName + " 的" + strconv.Itoa(recordCount) + "条最近记录为：\n" + scoreMsg,
|
|
ReferOriginMsg: true,
|
|
FromMsg: msg,
|
|
}
|
|
}
|