update

config/config.go

@@ -80,6 +80,7 @@ type Settings struct {
 	SendError              bool     `yaml:"send_error"`
 	AddAtGroup             bool     `yaml:"add_at_group"`
 	UrlPicTransfer         bool     `yaml:"url_pic_transfer"`
+	LotusPassword          string   `yaml:"lotus_password"`
 }
 
 // LoadConfig 从文件中加载配置并初始化单例配置
@@ -950,3 +951,15 @@ func GetUrlPicTransfer() bool {
 	}
 	return instance.Settings.UrlPicTransfer
 }
+
+// 获取GetLotusPassword的值
+func GetLotusPassword() string {
+	mu.Lock()
+	defer mu.Unlock()
+
+	if instance == nil {
+		mylog.Println("Warning: instance is nil when trying to GetLotusPassword value.")
+		return ""
+	}
+	return instance.Settings.LotusPassword
+}

handlers/send_group_msg.go

@@ -133,6 +133,7 @@ func handleSendGroupMsg(client callapi.Client, api openapi.OpenAPI, apiv2 openap
 			richMediaMessage, ok := groupReply.(*dto.RichMediaMessage)
 			if !ok {
 				mylog.Printf("Error: Expected RichMediaMessage type for key ")
+				return
 			}
 			// 上传图片并获取FileInfo
 			fileInfo, err := uploadMedia(context.TODO(), message.Params.GroupID.(string), richMediaMessage, apiv2)

server/wsserver.go

@@ -59,17 +59,18 @@ func wsHandler(api openapi.OpenAPI, apiV2 openapi.OpenAPI, p *Processor.Processo
 		token = c.Query("access_token")
 	}
 
-	if token == "" {
-		mylog.Printf("Connection failed due to missing token. Headers: %v", c.Request.Header)
-		c.JSON(http.StatusUnauthorized, gin.H{"error": "Missing token"})
-		return
-	}
-
-	// 使用GetWsServerToken()来获取有效的token
+	// 获取配置中的有效 token
 	validToken := config.GetWsServerToken()
-	if token != validToken {
-		mylog.Printf("Connection failed due to incorrect token. Headers: %v, Provided token: %s", c.Request.Header, tokenFromHeader)
-		c.JSON(http.StatusForbidden, gin.H{"error": "Incorrect token"})
+
+	// 如果配置的 token 不为空，但提供的 token 为空或不匹配
+	if validToken != "" && (token == "" || token != validToken) {
+		if token == "" {
+			mylog.Printf("Connection failed due to missing token. Headers: %v", c.Request.Header)
+			c.JSON(http.StatusUnauthorized, gin.H{"error": "Missing token"})
+		} else {
+			mylog.Printf("Connection failed due to incorrect token. Headers: %v, Provided token: %s", c.Request.Header, token)
+			c.JSON(http.StatusForbidden, gin.H{"error": "Incorrect token"})
+		}
 		return
 	}
 

template/config_template.go

@@ -37,6 +37,7 @@ settings:
   lotus: false                                       # lotus特性默认为false,当为true时,将会连接到另一个lotus为false的gensokyo。
                                                      # 使用它提供的图床和idmaps服务(场景:同一个机器人在不同服务器运行,或内网需要发送base64图)。
                                                      # 如果需要发送base64图片,需要设置正确的公网server_dir和开放对应的port
+  lotus_password : ""                                # lotus鉴权 设置后,从gsk需要保持相同密码来访问主gsk
 
   #增强配置项                                           
 

template/config_template.yml

@@ -71,4 +71,5 @@ settings:
   No_White_Response : ""            #默认不兜底,强烈建议设置一个友善的兜底回复,告知审核机器人已无隐藏指令,如:你输入的指令不对哦,@机器人来获取可用指令
   send_error : true                 #将报错用文本发出,避免机器人被审核报无响应
   add_at_group : false              #自动在群聊指令前加上at,某些机器人写法特别,必须有at才反应时,请打开,默认请关闭(如果需要at,不需要at指令混杂,请优化代码适配群场景,群场景目前没有at概念
-  url_pic_transfer : false          #把图片url(任意来源图链)变成你备案的白名单url 需要较高上下行+ssl+自备案域名+设置白名单域名(暂时不需要)
+  url_pic_transfer : false          #把图片url(任意来源图链)变成你备案的白名单url 需要较高上下行+ssl+自备案域名+设置白名单域名(暂时不需要)
+  lotus_password : ""               # lotus鉴权 设置后,从gsk需要保持相同密码来访问主gsk

url/shorturl.go

@@ -1,6 +1,7 @@
 package url
 
 import (
+	"crypto/md5"
 	"crypto/sha256"
 	"encoding/base64"
 	"encoding/hex"
@@ -72,6 +73,7 @@ func isValidURL(toTest string) bool {
 		return false
 	}
 	if parsedURL.Scheme != "http" && parsedURL.Scheme != "https" {
+		mylog.Printf("链接%v缺少协议头,请添加https://或http://", toTest)
 		return false
 	}
 
@@ -131,10 +133,19 @@ func GenerateShortURL(longURL string) string {
 	if config.GetLotusValue() {
 		serverDir := config.GetServer_dir()
 		requestURL := fmt.Sprintf("%s://%s:%s/url", protocol, serverDir, portValue) // 改变变量名以避免冲突
-
 		// 使用 url.Values 构造请求数据
 		formData := url.Values{}
-		formData.Set("url", (longURL))
+		formData.Set("url", longURL)
+		// 获取密码
+		password := config.GetLotusPassword()
+		// 如果密码不为空，则计算 MD5 值并添加 token
+		if password != "" {
+			hasher := md5.New()
+			hasher.Write([]byte(password))
+			token := hex.EncodeToString(hasher.Sum(nil))
+			// 添加 token
+			formData.Set("token", token)
+		}
 
 		// 创建请求
 		req, err := http.NewRequest("POST", requestURL, strings.NewReader(formData.Encode()))
@@ -305,10 +316,13 @@ func isMalicious(decoded string) bool {
 // 短链接服务handler
 func CreateShortURLHandler(c *gin.Context) {
 	rawURL := c.PostForm("url")
+	token := c.PostForm("token") // 接收 token 参数
+
 	longURL := decodeBase64IfNeeded(rawURL)
 
-	if longURL == "" || isMalicious(longURL) || !isValidURL(longURL) {
-		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid URL"})
+	// 检查 URL 是否有效，以及在密码不为空时检查 token
+	if longURL == "" || isMalicious(longURL) || !isValidURL(longURL) || (config.GetLotusPassword() != "" && !isValidToken(token)) {
+		c.JSON(http.StatusBadRequest, gin.H{"error": "Invalid URL or token"})
 		return
 	}
 
@@ -322,6 +336,24 @@ func CreateShortURLHandler(c *gin.Context) {
 	c.JSON(http.StatusOK, gin.H{"shortURL": baseUrl + "/url/" + shortURL})
 }
 
+// isValidToken 检查 token 是否有效
+func isValidToken(token string) bool {
+	// 从配置中获取密码
+	password := config.GetLotusPassword()
+
+	// 如果密码为空，直接返回 true
+	if password == "" {
+		return true
+	}
+
+	// 计算 MD5
+	hasher := md5.New()
+	hasher.Write([]byte(password))
+	md5Password := hex.EncodeToString(hasher.Sum(nil))
+
+	return md5Password == token
+}
+
 // 短链接baseurl
 func GetBaseURL() string {
 	serverDir := config.GetServer_dir()