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

feat: 添加steam上线速报

Browse Source
This commit is contained in:
lixiangwuxian
2024-10-28 02:58:26 +08:00
parent 7e28953a56
commit dce09c6e1f
3 changed files with 626 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

231
handler/steamplaying/steam_playing.go
View File

@@ -2,80 +2,215 @@ package steamplaying
import (
"fmt"
"io"
"net/http"
"regexp"
"time"
"git.lxtend.com/qqbot/action"
"git.lxtend.com/qqbot/constants"
"git.lxtend.com/qqbot/handler"
"git.lxtend.com/qqbot/model"
"git.lxtend.com/qqbot/util"
"golang.org/x/net/proxy"
)
func init() {
// Register the handler with the server
handler.RegisterHandler("绑steam", bindSteam, constants.LEVEL_USER)
handler.RegisterHelpInform("绑steam", "绑定您的steam账号。可以通过右上角-账户明细页面 https://store.steampowered.com/account/ 查看，位于页面左上角")
handler.RegisterHandler("群绑定steam", bindSteamInGroup, constants.LEVEL_USER)
handler.RegisterHelpInform("群绑定steam", "在群内启用你的steam游戏状态查询")
// handler.RegisterHandler("群通报steam", bindSteamInGroupBroadCast, constants.LEVEL_USER)
// handler.RegisterHelpInform("群通报steam", "在群内启用你的steam游戏上线通报")
handler.RegisterHandler("解绑steam", unbindSteam, constants.LEVEL_USER)
handler.RegisterHelpInform("解绑steam", "解绑您的steam账号，并解绑所有群监听")
handler.RegisterHandler("群解绑steam", unbindSteamInGroup, constants.LEVEL_USER)
handler.RegisterHelpInform("群解绑steam", "解绑本群群监听steam游戏状态")
handler.RegisterHandler("查房", checkSteamPlaying, constants.LEVEL_USER)
handler.RegisterHelpInform("查房", "查看群内成员的steam游戏状态")
go RoundCheckSteamPlaying()
}
func checkSteamPlaying(msg model.Message) model.Reply {
tokens := util.SplitN(msg.RawMsg, 2)
if len(tokens) < 2 {
func bindSteam(msg model.Message) model.Reply {
token := util.SplitN(msg.RawMsg, 2)
if len(token) < 2 {
return model.Reply{
ReplyMsg: "请输入正确的steamID",
ReplyMsg: "请输入steamID",
ReferOriginMsg: true,
FromMsg: msg,
}
}
status, err := checkSteamGameStatus(tokens[1])
if err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("检查失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
} else {
re := regexp.MustCompile(`https://steamcommunity\.com/profiles/([0-9]+)`)
steamIdInUrl := re.FindStringSubmatch(token[1])
if steamIdInUrl != nil {
token[1] = steamIdInUrl[1]
}
if valid, err := checkSteamIDValid(token[1]); !valid {
return model.Reply{
ReplyMsg: fmt.Sprintf("steamID无效: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
} else if err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("steamID验证失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
}
if err := bindSteamUser(msg.UserId, token[1]); err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("绑定失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
}
}
return model.Reply{
ReplyMsg: status,
ReplyMsg: fmt.Sprintf("绑定steam用户%s成功", token[1]),
ReferOriginMsg: true,
FromMsg: msg,
}
}
func checkSteamGameStatus(steamID string) (string, error) {
dialer, err := proxy.SOCKS5("tcp", "100.94.183.43:2080", nil, proxy.Direct)
if err != nil {
return "", fmt.Errorf("设置SOCKS5代理失败: %v", err)
func bindSteamInGroup(msg model.Message) model.Reply {
if user, err := getSteamUser(msg.UserId); err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("查询steam绑定失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
} else if err := bindUserInGroup(msg.GroupInfo.GroupId, user.SteamID); err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("绑定至群监听列表失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
}
url := fmt.Sprintf("https://steamcommunity.com/id/%s", steamID)
httpTransport := &http.Transport{}
httpTransport.Dial = dialer.Dial
client := &http.Client{
Transport: httpTransport,
Timeout: 10 * time.Second, // 设置请求超时时间
return model.Reply{
ReplyMsg: "绑定至群监听列表成功",
ReferOriginMsg: true,
FromMsg: msg,
}
}
func unbindSteam(msg model.Message) model.Reply {
if user, err := getSteamUser(msg.UserId); err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("解绑失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
} else {
if err := unbindUserInAllGroup(user.SteamID); err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("解绑所有群监听列表失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
}
if err := unbindSteamUser(msg.UserId); err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("解绑失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
}
}
return model.Reply{
ReplyMsg: "解绑成功",
ReferOriginMsg: true,
FromMsg: msg,
}
}
func unbindSteamInGroup(msg model.Message) model.Reply {
if user, err := getSteamUser(msg.UserId); err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("群监听解绑失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
} else {
if err := unbindUserInGroup(msg.GroupInfo.GroupId, user.SteamID); err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("群监听解绑失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
}
}
return model.Reply{
ReplyMsg: "群监听解绑成功，本群内将不再查询你的steam游戏状态",
ReferOriginMsg: true,
FromMsg: msg,
}
}
func checkSteamPlaying(msg model.Message) model.Reply {
users, err := getSteamUsersInGroup(msg.GroupInfo.GroupId)
if err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("获取群成员steam列表失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
}
var steamIds []string
for _, user := range users {
steamIds = append(steamIds, user.SteamID)
}
gameList, err := checkSteamGameStatus(steamIds)
if err != nil {
return model.Reply{
ReplyMsg: fmt.Sprintf("获取游戏列表失败: %v", err),
ReferOriginMsg: true,
FromMsg: msg,
}
}
return model.Reply{
ReplyMsg: gameList,
ReferOriginMsg: true,
FromMsg: msg,
}
}
func RoundCheckSteamPlaying() {
once := true
playingMap := map[int64]map[string]string{}
for {
time.Sleep(5 * time.Second)
groups, err := getAllGroupID()
if err != nil {
fmt.Println("获取群列表失败: ", err)
continue
}
for _, group := range groups {
if _, ok := playingMap[group]; !ok {
playingMap[group] = map[string]string{}
}
users, err := getSteamUsersInGroup(group)
if err != nil {
fmt.Println("获取群成员steam列表失败: ", err)
continue
}
var steamIds []string
for _, user := range users {
steamIds = append(steamIds, user.SteamID)
}
gameList, err := checkDiffSteamGameStatus(steamIds, playingMap[group])
if err != nil {
fmt.Println("获取游戏列表失败: ", err)
continue
}
if gameList != "" && !once {
msg := model.Reply{
ReplyMsg: "速报：\n" + gameList,
ReferOriginMsg: false,
FromMsg: model.Message{GroupInfo: model.GroupInfo{GroupId: group}},
}
action.ActionManager.SendMsg(msg)
}
}
once = false
}
resp, err := client.Get(url)
if err != nil {
return "", fmt.Errorf("获取页面失败: %v", err)
}
defer resp.Body.Close()
body, err := io.ReadAll(resp.Body)
if err != nil {
return "", fmt.Errorf("读取页面内容失败: %v", err)
}
fmt.Print(string(body))
// 提取游戏名称
gameNameRegex := regexp.MustCompile(`(?s)<div\s+class="profile_in_game_name">\s*(.*?)\s*</div>`)
matches := gameNameRegex.FindSubmatch(body)
if len(matches) < 2 {
return "", fmt.Errorf("%s好像没有在玩游戏", steamID)
}
gameName := string(matches[1])
return fmt.Sprintf("该用户正在玩: %s", gameName), nil
}