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

feat: webhook回调支持口令鉴权

Browse Source
This commit is contained in:
lixiangwuxian 2025-03-24 11:29:05 +08:00
parent 9d512fd4e3
commit 15a3677408
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
config/config.go
View File

@ -18,6 +18,9 @@ type Config struct {
OpenaiApiBaseUrl string `yaml:"openai_api_base_url"` OpenaiApiBaseUrl string `yaml:"openai_api_base_url"`
OpenaiModelName string `yaml:"openai_model_name"` OpenaiModelName string `yaml:"openai_model_name"`
OpenaiPrompt string `yaml:"openai_prompt"` OpenaiPrompt string `yaml:"openai_prompt"`
WebServer struct {
Token string `yaml:"token"`
} `yaml:"web_server"`
} }
func init() { func init() {

6
handler/restart/restart.go
View File

@ -8,6 +8,7 @@ import (
"os/exec" "os/exec"
"time" "time"
"git.lxtend.com/qqbot/config"
"git.lxtend.com/qqbot/constants" "git.lxtend.com/qqbot/constants"
"git.lxtend.com/qqbot/handler" "git.lxtend.com/qqbot/handler"
"git.lxtend.com/qqbot/model" "git.lxtend.com/qqbot/model"
@ -186,6 +187,11 @@ func BuildBotHandler(c *gin.Context) {
} }
func RestartBotHandler(c *gin.Context) { func RestartBotHandler(c *gin.Context) {
token := c.Request.Header.Get("Authorization")
if token != config.ConfigManager.GetConfig().WebServer.Token {
c.JSON(http.StatusUnauthorized, gin.H{"error": "Unauthorized"})
return
}
err := restartProgram() err := restartProgram()
log.Println("重启程序...") log.Println("重启程序...")
if err != nil { if err != nil {