feat: 添加steam上线速报

2024-10-28 02:58:26 +08:00
parent 7e28953a56
commit dce09c6e1f
3 changed files with 626 additions and 48 deletions
--- a/handler/steamplaying/service.go
+++ b/handler/steamplaying/service.go
@@ -0,0 +1,376 @@
+package steamplaying
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io"
+	"net"
+	"net/http"
+	"sync"
+	"time"
+
+	"git.lxtend.com/qqbot/config"
+	"git.lxtend.com/qqbot/sqlite3"
+	"golang.org/x/net/proxy"
+)
+
+var SteamAPIKey = ""
+var ProxyAddr = ""
+
+func init() {
+	SteamAPIKey = config.ConfigManager.GetProperty("steam_api_key")
+	ProxyAddr = config.ConfigManager.GetProperty("proxy_addr")
+}
+
+func init() {
+	createSteamUserTable := `CREATE TABLE IF NOT EXISTS steam_user (
+		id INTEGER PRIMARY KEY AUTOINCREMENT,
+		qqid TEXT,
+		steamid TEXT UNIQUE
+		);`
+	createSteamUserForGroupTable := `CREATE TABLE IF NOT EXISTS steam_user_for_group (
+		id INTEGER PRIMARY KEY AUTOINCREMENT,
+		group_id TEXT,
+		steamid TEXT
+		);`
+	sqlite3.TryCreateTable(createSteamUserTable)
+	sqlite3.TryCreateTable(createSteamUserForGroupTable)
+}
+
+func bindSteamUser(qqid int64, steamid string) error {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+	var steamUser []SteamUser
+	err = tx.Select(&steamUser, "SELECT * FROM steam_user WHERE qqid = ?", qqid)
+	if err != nil {
+		return err
+	}
+	if len(steamUser) > 0 {
+		return errors.New("已绑定")
+	}
+	_, err = tx.Exec("INSERT INTO steam_user (qqid, steamid) VALUES (:qqid, :steamid)", qqid, steamid)
+	if err != nil {
+		return err
+	}
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func unbindSteamUser(qqid int64) error {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+	_, err = tx.Exec("DELETE FROM steam_user WHERE qqid = ?", qqid)
+	if err != nil {
+		return err
+	}
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func bindUserInGroup(groupID int64, steamid string) error {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+	var steamUser []SteamUserForGroup
+	err = tx.Select(&steamUser, "SELECT * FROM steam_user_for_group WHERE group_id = ? AND steamid = ?", groupID, steamid)
+	if err != nil {
+		return err
+	}
+	if len(steamUser) > 0 {
+		return errors.New("已绑定")
+	}
+	_, err = tx.Exec("INSERT INTO steam_user_for_group (group_id, steamid) VALUES (:group_id, :steamid)", groupID, steamid)
+	if err != nil {
+		return err
+	}
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func unbindUserInGroup(groupID int64, steamid string) error {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+	_, err = tx.Exec("DELETE FROM steam_user_for_group WHERE group_id = ? AND steamid = ?", groupID, steamid)
+	if err != nil {
+		return err
+	}
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func unbindUserInAllGroup(steamid string) error {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+	_, err = tx.Exec("DELETE FROM steam_user_for_group WHERE steamid = ?", steamid)
+	if err != nil {
+		return err
+	}
+	err = tx.Commit()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func getSteamUser(qqid int64) (SteamUser, error) {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return SteamUser{}, err
+	}
+	defer tx.Rollback()
+	var steamUser []SteamUser
+	err = tx.Select(&steamUser, "SELECT * FROM steam_user WHERE qqid = ?", qqid)
+	if err != nil || len(steamUser) == 0 {
+		return SteamUser{}, err
+	}
+	return steamUser[0], nil
+}
+
+func getSteamUsersInGroup(groupID int64) ([]SteamUserForGroup, error) {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return nil, err
+	}
+	defer tx.Rollback()
+	var steamUsers []SteamUserForGroup
+	err = tx.Select(&steamUsers, "SELECT * FROM steam_user_for_group WHERE group_id = ?", groupID)
+	if err != nil {
+		return nil, err
+	}
+	return steamUsers, nil
+}
+
+func checkSteamGameStatus(steamID []string) (string, error) {
+	if len(steamID) == 0 {
+		return "疑似没人在玩游戏", nil
+	}
+	var glbErr error
+	var writeMutex sync.Mutex
+	var Players []PlayerSummary
+	var wg sync.WaitGroup
+	for step := 0; step < len(steamID)/100+1; step++ {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			url := "https://api.steampowered.com/ISteamUser/GetPlayerSummaries/v0002/?key=%s&steamids=%s"
+			fullSteamID := ""
+			for i := step * 100; i < (step+1)*100 && i < len(steamID); i++ {
+				fullSteamID += steamID[i] + ","
+			}
+			fullSteamID = fullSteamID[:len(fullSteamID)-1]
+			url = fmt.Sprintf(url, SteamAPIKey, fullSteamID)
+
+			dialer, _ := proxy.SOCKS5("tcp", ProxyAddr, nil, proxy.Direct)
+			httpTransport := &http.Transport{
+				Dial: func(network, addr string) (net.Conn, error) {
+					return dialer.Dial(network, addr)
+				},
+			}
+			client := &http.Client{
+				Transport: httpTransport,
+				Timeout:   10 * time.Second, // 设置请求超时时间
+			}
+			var resp *http.Response
+			maxRetry := 5
+			for i := 0; i < maxRetry; i++ {
+				glbErr = nil
+				var err error
+				resp, err = client.Get(url)
+				if err != nil {
+					if i == maxRetry-1 {
+						glbErr = errors.New("Get方法请求Steam API失败")
+						return
+					}
+					continue
+				}
+				break
+			}
+
+			defer resp.Body.Close()
+			body, err := io.ReadAll(resp.Body)
+			if err != nil {
+				glbErr = err
+				return
+			}
+			var userResponse GetPlayerSummariesResponse
+			if err := json.Unmarshal(body, &userResponse); err != nil {
+				glbErr = err
+				return
+			}
+			writeMutex.Lock()
+			Players = append(Players, userResponse.Response.Players...)
+			writeMutex.Unlock()
+		}()
+	}
+	wg.Wait()
+	if glbErr != nil {
+		return "", glbErr
+	}
+	gameStatusList := ""
+	for _, userState := range Players {
+		if userState.ToGameStatus() != "" {
+			gameStatusList += userState.ToGameStatus() + "\n"
+		}
+	}
+	if gameStatusList != "" {
+		gameStatusList = gameStatusList[:len(gameStatusList)-1]
+	} else {
+		gameStatusList = "疑似没有人在玩游戏"
+	}
+	return gameStatusList, nil
+}
+
+func checkDiffSteamGameStatus(steamID []string, lastTimeStat map[string]string) (string, error) {
+	if len(steamID) == 0 {
+		return "疑似没人在玩游戏", nil
+	}
+	var glbErr error
+	var writeMutex sync.Mutex
+	var Players []PlayerSummary
+	var wg sync.WaitGroup
+	for step := 0; step < len(steamID)/100+1; step++ {
+		wg.Add(1)
+		go func() {
+			defer wg.Done()
+			url := "https://api.steampowered.com/ISteamUser/GetPlayerSummaries/v0002/?key=%s&steamids=%s"
+			fullSteamID := ""
+			for i := step * 100; i < (step+1)*100 && i < len(steamID); i++ {
+				fullSteamID += steamID[i] + ","
+			}
+			fullSteamID = fullSteamID[:len(fullSteamID)-1]
+			url = fmt.Sprintf(url, SteamAPIKey, fullSteamID)
+
+			dialer, _ := proxy.SOCKS5("tcp", ProxyAddr, nil, proxy.Direct)
+			httpTransport := &http.Transport{
+				Dial: func(network, addr string) (net.Conn, error) {
+					return dialer.Dial(network, addr)
+				},
+			}
+			client := &http.Client{
+				Transport: httpTransport,
+				Timeout:   10 * time.Second, // 设置请求超时时间
+			}
+			var resp *http.Response
+			maxRetry := 5
+			for i := 0; i < maxRetry; i++ {
+				glbErr = nil
+				var err error
+				resp, err = client.Get(url)
+				if err != nil {
+					if i == maxRetry-1 {
+						glbErr = errors.New("Get方法请求Steam API失败")
+						return
+					}
+					continue
+				}
+				break
+			}
+
+			defer resp.Body.Close()
+			body, err := io.ReadAll(resp.Body)
+			if err != nil {
+				glbErr = err
+				return
+			}
+			var userResponse GetPlayerSummariesResponse
+			if err := json.Unmarshal(body, &userResponse); err != nil {
+				glbErr = err
+				return
+			}
+			writeMutex.Lock()
+			Players = append(Players, userResponse.Response.Players...)
+			writeMutex.Unlock()
+		}()
+	}
+	wg.Wait()
+	if glbErr != nil {
+		return "", glbErr
+	}
+	gameStatusListStr := ""
+	for _, userState := range Players {
+		if lastTimeStat[userState.SteamID] == userState.GameID {
+			continue
+		}
+		if userState.ToGameStatus() != "" {
+			gameStatusListStr += userState.ToGameStatus() + "\n"
+		}
+		lastTimeStat[userState.SteamID] = userState.GameID
+	}
+	if gameStatusListStr != "" {
+		gameStatusListStr = gameStatusListStr[:len(gameStatusListStr)-1]
+	}
+	return gameStatusListStr, nil
+}
+
+func checkSteamIDValid(steamID string) (bool, error) {
+	url := "https://api.steampowered.com/ISteamUser/GetPlayerSummaries/v0002/?key=%s&steamids=%s"
+	url = fmt.Sprintf(url, SteamAPIKey, steamID)
+	dialer, _ := proxy.SOCKS5("tcp", ProxyAddr, nil, proxy.Direct)
+	httpTransport := &http.Transport{
+		Dial: func(network, addr string) (net.Conn, error) {
+			return dialer.Dial(network, addr)
+		},
+	}
+	client := &http.Client{
+		Transport: httpTransport,
+		Timeout:   10 * time.Second, // 设置请求超时时间
+	}
+	resp, err := client.Get(url)
+	if err != nil {
+		return false, err
+	}
+	defer resp.Body.Close()
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return false, err
+	}
+	var userResponse GetPlayerSummariesResponse
+	if err := json.Unmarshal(body, &userResponse); err != nil {
+		return false, err
+	}
+	if len(userResponse.Response.Players) == 0 {
+		return false, nil
+	}
+	return true, nil
+}
+
+func getAllGroupID() ([]int64, error) {
+	tx, err := sqlite3.GetTran()
+	if err != nil {
+		return nil, err
+	}
+	defer tx.Rollback()
+	var groupIDs []int64
+	err = tx.Select(&groupIDs, "SELECT DISTINCT group_id FROM steam_user_for_group")
+	if err != nil {
+		return nil, err
+	}
+	return groupIDs, nil
+}