Skip to content

Commit

Permalink
Beta42 (#106)
Browse files Browse the repository at this point in the history 
* beta9

* beta11

* beta11

* beta13

* beta12

* beta13

* beta15

* beta16

* beta17

* beta18

* beta19

* beta20

* beta21

* add_playermanager_api

* beta22

* beta23

* beta24

* beta25

* beta26

* beta27

* beta28

* beta28

* beta29

* beta30

* beta31

* beta32

* beta33

* beta34

* beta35

* beta35

* beta36

* beta37

* beta38

* beta39

* beta40

* beta41

* beta41
  • Loading branch information
@Hoshinonyaruko
Hoshinonyaruko authored Feb 25, 2024
1 parent ca9409a commit 60d0d7c
Show file tree
Hide file tree
Showing 9 changed files with 350 additions and 48 deletions.
265 changes: 233 additions & 32 deletions bot/bot.go
View file
Original file line number Diff line number Diff line change
@@ -1,11 +1,13 @@
package bot

import (
"bufio"
"bytes"
"encoding/json"
"fmt"
"io"
"net/http"
"os"
"strconv"
"strings"
"time"
Expand Down Expand Up @@ -102,6 +104,8 @@ func GensokyoHandlerClosure(c *gin.Context, config config.Config) {
// 去除字符串前后的空格
msg = strings.TrimSpace(msg)

//英文

// 处理以 "getbot" 开头的消息
if strings.HasPrefix(msg, "getbot") {
getBotHandler(msg, message, config)
Expand All @@ -120,6 +124,44 @@ func GensokyoHandlerClosure(c *gin.Context, config config.Config) {
return
}

// 处理以 "kick" 开头的消息
if strings.HasPrefix(msg, "kick") {
kickorbanHandler(msg, message, config, "kick")
return
}

// 处理以 "ban" 开头的消息
if strings.HasPrefix(msg, "ban") {
kickorbanHandler(msg, message, config, "ban")
return
}

// 处理以 "Broadcast" 开头的消息
if strings.HasPrefix(msg, "Broadcast") {
broadcastMessageHandler(msg, message, config)
return
}

// 处理以 "restart" 开头的消息
if strings.HasPrefix(msg, "restart") {
restartHandler(msg, message, config)
return
}

// 处理以 "commonlist" 开头的消息
if strings.HasPrefix(msg, "commonlist") {
listCommandsHandler(message, config)
return
}

// 处理以 "playernum" 开头的消息
if strings.HasPrefix(msg, "playernum") {
getplayernumHandler(message, config)
return
}

//简体中文

// 处理以 "玩家列表" 开头的消息
if strings.HasPrefix(msg, "玩家列表") {
getplayerHandler(msg, message, config, false)
Expand All @@ -132,36 +174,18 @@ func GensokyoHandlerClosure(c *gin.Context, config config.Config) {
return
}

// 处理以 "kick" 开头的消息
if strings.HasPrefix(msg, "kick") {
kickorbanHandler(msg, message, config, "kick")
return
}

// 处理以 "踢人" 开头的消息
if strings.HasPrefix(msg, "踢人") {
kickorbanHandler(msg, message, config, "kick")
return
}

// 处理以 "ban" 开头的消息
if strings.HasPrefix(msg, "ban") {
kickorbanHandler(msg, message, config, "ban")
return
}

// 处理以 "封禁" 开头的消息
if strings.HasPrefix(msg, "封禁") {
kickorbanHandler(msg, message, config, "ban")
return
}

// 处理以 "Broadcast" 开头的消息
if strings.HasPrefix(msg, "Broadcast") {
broadcastMessageHandler(msg, message, config)
return
}

// 处理以 "广播" 开头的消息
if strings.HasPrefix(msg, "广播") {
broadcastMessageHandler(msg, message, config)
Expand All @@ -174,21 +198,41 @@ func GensokyoHandlerClosure(c *gin.Context, config config.Config) {
return
}

// 处理以 "restart" 开头的消息
if strings.HasPrefix(msg, "restart") {
// 处理以 "指令列表" 开头的消息
if strings.HasPrefix(msg, "指令列表") {
listCommandsHandler(message, config)
return
}

// 处理以 "玩家数量" 开头的消息
if strings.HasPrefix(msg, "玩家数量") {
getplayernumHandler(message, config)
return
}

//繁体中文

// 處理以 "廣播" 開頭的消息
if strings.HasPrefix(msg, "廣播") {
broadcastMessageHandler(msg, message, config)
return
}

// 處理以 "重啟伺服器" 開頭的消息
if strings.HasPrefix(msg, "重啟伺服器") {
restartHandler(msg, message, config)
return
}

// 处理以 "指令列表" 开头的消息
if strings.HasPrefix(msg, "指令列表") {
listCommandsHandler(message, config)
// 处理以 "命令列表" 开头的消息
if strings.HasPrefix(msg, "命令列表") {
listCommandsHandlertc(message, config)
return
}

// 处理以 "commonlist" 开头的消息
if strings.HasPrefix(msg, "commonlist") {
listCommandsHandler(message, config)
// 处理以 "玩家數量" 开头的消息
if strings.HasPrefix(msg, "玩家數量") {
getplayernumHandler(message, config)
return
}

Expand Down Expand Up @@ -636,18 +680,20 @@ func listCommandsHandler(message OnebotGroupMessage, config config.Config) {
// 构建指令列表
commands := []string{
"getbot - 获取机器人信息",
"player - 获取玩家信息",
"update player - 更新玩家信息",
"玩家列表 - 显示玩家列表",
"刷新玩家列表 - 刷新并显示玩家列表",
"kick - 踢出玩家",
"踢人 - 踢出玩家",
"ban - 封禁玩家",
"封禁 - 封禁玩家",
"Broadcast - 发送广播消息",
"广播 - 发送广播消息",
"重启服务器 - 重启游戏服务器",
"restart - 重启游戏服务器",
"玩家数量- 查询玩家数量",
"player - Retrieve player information",
"update player - Update player information",
"kick - Kick a player out",
"ban - Ban a player",
"Broadcast - Send a broadcast message",
"restart - Restart the game server",
"playernum - Query the number of players",
}

// 将指令列表转换为字符串，每个指令后换行
Expand All @@ -656,3 +702,158 @@ func listCommandsHandler(message OnebotGroupMessage, config config.Config) {
// 发送指令列表
sendGroupMessage(message.GroupID, message.UserID, "可用指令列表:\n"+commandsStr, config)
}

func listCommandsHandlertc(message OnebotGroupMessage, config config.Config) {
// 构建指令列表
commands := []string{
"getbot - 獲取機器人資訊",
"玩家列表 - 顯示玩家列表",
"刷新玩家列表 - 刷新並顯示玩家列表",
"踢人 - 踢出玩家",
"封禁 - 封禁玩家",
"廣播 - 發送廣播消息",
"重啟伺服器 - 重啟遊戲伺服器",
"玩家數量- 查询玩家數量",
"player - Retrieve player information",
"update player - Update player information",
"kick - Kick a player out",
"ban - Ban a player",
"Broadcast - Send a broadcast message",
"restart - Restart the game server",
"playernum - Query the number of players",
}

// 將指令列表轉換為字符串，每個指令後換行
commandsStr := strings.Join(commands, "\n")

// 發送指令列表
sendGroupMessage(message.GroupID, message.UserID, "可用指令列表:\n"+commandsStr, config)
}

// 发送基于命令的消息
func SendCommandMessages(command string, config config.Config) error {
// 尝试打开guilds.txt文件
file, err := os.Open("guilds.txt")
if err != nil {
if os.IsNotExist(err) {
// 如果文件不存在，则创建文件
file, err = os.Create("guilds.txt")
if err != nil {
return fmt.Errorf("failed to create guilds.txt: %w", err)
}
// 文件已创建但为空，直接返回
file.Close()
return nil
} else {
// 其他错误，返回错误信息
return fmt.Errorf("failed to open guilds.txt: %w", err)
}
}
defer file.Close()

scanner := bufio.NewScanner(file)
for scanner.Scan() {
groupID := scanner.Text()
// 转换groupID为int64
var gID int64
_, err := fmt.Sscanf(groupID, "%d", &gID)
if err != nil {
return fmt.Errorf("failed to parse groupID: %w", err)
}

// 根据命令读取相应的文件
fileName := command + ".txt"
message, err := os.ReadFile(fileName)
if err != nil {
if os.IsNotExist(err) {
// 文件不存在，创建文件
file, err := os.Create(fileName)
if err != nil {
return fmt.Errorf("failed to create %s: %w", fileName, err)
}
file.Close() // 关闭新创建的文件
continue // 文件已创建但为空，继续下一个groupID
} else {
return fmt.Errorf("failed to read %s: %w", fileName, err)
}
}
if len(message) == 0 {
// 文件为空，继续下一个groupID
continue
}

// 发送消息
err = sendGroupMessage(gID, 0, string(message), config)
if err != nil {
return fmt.Errorf("failed to send group message: %w", err)
}
}

if err := scanner.Err(); err != nil {
return fmt.Errorf("error reading guilds.txt: %w", err)
}

return nil
}

func getplayernumHandler(message OnebotGroupMessage, config config.Config) {

// 尝试获取用户的IP和UUID
userIPData, err := RetrieveIPByUserID(message.UserID)
if err != nil {
// 发送错误消息
sendGroupMessage(message.GroupID, message.UserID, "没有正确设置,请使用palworld-go面板,在机器人管理或服务器主人处获取指令,然后发给我", config)
return
}

// 检查IP是否为空
if userIPData.IP != "" {
// 根据https值确定使用HTTP还是HTTPS
baseURL := "http://" + userIPData.IP
if userIPData.Https {
baseURL = "https://" + userIPData.IP
}

// 创建HTTP客户端
client := &http.Client{}
apiURL := baseURL + "/api/getplayernum"
req, err := http.NewRequest("GET", apiURL, nil) // GET请求不需要请求体
if err != nil {
sendGroupMessage(message.GroupID, message.UserID, "创建请求失败: "+err.Error(), config)
return
}
req.Header.Set("Content-Type", "application/json")
req.AddCookie(&http.Cookie{Name: "login_cookie", Value: userIPData.UUID})

resp, err := client.Do(req)
if err != nil {
sendGroupMessage(message.GroupID, message.UserID, "发送请求失败: "+err.Error(), config)
return
}
defer resp.Body.Close()

// 检查响应状态
if resp.StatusCode != http.StatusOK {
sendGroupMessage(message.GroupID, message.UserID, "获取玩家数量失败", config)
return
}

// 解析响应体中的JSON数据
var result struct {
TotalPlayers int `json:"total_players"`
OnlinePlayers int `json:"online_players"`
}
if err := json.NewDecoder(resp.Body).Decode(&result); err != nil {
sendGroupMessage(message.GroupID, message.UserID, "解析响应失败: "+err.Error(), config)
return
}

// 发送成功消息，包括总玩家数量和在线玩家数量
successMessage := fmt.Sprintf("总玩家数量: %d, 在线玩家数量: %d", result.TotalPlayers, result.OnlinePlayers)
sendGroupMessage(message.GroupID, message.UserID, successMessage, config)
} else {
// 发送错误消息
sendGroupMessage(message.GroupID, message.UserID, "没有获取到面板信息,请使用palworld-go面板,在机器人管理或服务器主人处获取指令,然后发给我", config)
return
}
}
2 changes: 2 additions & 0 deletions config/config.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,8 @@ type Config struct {
SteamCmdPath string `json:"steamCmdPath"` // 自定义steamcmd路径
EnableUe4Debug bool `json:"enableUe4Debug"` // 是否开启UE4 Debug窗口
EnableEngineSetting bool `json:"enableEngineSetting"` // 是否开启引擎设置
EnableBotNotification bool `json:"enableBotNotification"` // 是否开启机器人广播
EnableRebootLater bool `json:"enableRebootLater"` // 是否开启延时关闭&重启
}

// 默认配置
Expand Down
10 changes: 10 additions & 0 deletions front/palworld-front/src/pages/IndexView.vue
View file
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,16 @@
label="是否启用引擎配置,部分电脑修改引擎配置会卡顿,可关闭"
class="q-my-md"
/>
<q-toggle
v-model="config.enableBotNotification"
label="是否启用机器人服务器情况播报"
class="q-my-md"
/>
<q-toggle
v-model="config.enableRebootLater"
label="主机管理页面点关闭按钮前是否广播通知玩家(延迟60s关闭)"
class="q-my-md"
/>
<q-input
filled
v-model="config.maintenanceWarningMessage"
Expand Down
2 changes: 1 addition & 1 deletion rconclient.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ func HandleMemoryUsage(threshold float64, RconClient *RconClient, config config.
// 判断是否使用新的DLL方式发送广播
if config.UseDll {
// 构建请求的URL
base := "http://127.0.0.1:" + fmt.Sprintf("%d", config.DllPort) + "/rcon?text="
base := "http://127.0.0.1:" + config.DllPort + "/rcon?text="
message := url.QueryEscape("broadcast " + config.MaintenanceWarningMessage)
fullURL := base + message

Expand Down
2 changes: 1 addition & 1 deletion sys/restart_unix.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ func setConsoleTitleWindows(title string) error {
return nil
}

func KillProcess() error {
func KillProcess(config config.Config) error {
var cmd *exec.Cmd

if runtime.GOOS == "windows" {
Expand Down
Loading

0 comments on commit 60d0d7c

Please sign in to comment.