From 1374e674b73eb1919aefbaf32b2a84364babe6a1 Mon Sep 17 00:00:00 2001 From: cosmo Date: Tue, 5 Dec 2023 01:26:29 +0800 Subject: [PATCH 1/6] beta77 --- handlers/send_group_msg.go | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/handlers/send_group_msg.go b/handlers/send_group_msg.go index f2e5258e..bda0e112 100644 --- a/handlers/send_group_msg.go +++ b/handlers/send_group_msg.go @@ -197,7 +197,7 @@ func handleSendGroupMsg(client callapi.Client, api openapi.OpenAPI, apiv2 openap mylog.Printf("发送组合消息失败: %v", err) return // 或其他错误处理 } - if ret.Message.Ret == 22009 { + if ret != nil && ret.Message.Ret == 22009 { mylog.Printf("信息发送失败,加入到队列中,下次被动信息进行发送") var pair echo.MessageGroupPair pair.Group = message.Params.GroupID.(string) @@ -231,7 +231,7 @@ func handleSendGroupMsg(client callapi.Client, api openapi.OpenAPI, apiv2 openap if err != nil { mylog.Printf("发送文本群组信息失败: %v", err) } - if ret.Message.Ret == 22009 { + if ret != nil && ret.Message.Ret == 22009 { mylog.Printf("信息发送失败,加入到队列中,下次被动信息进行发送") var pair echo.MessageGroupPair pair.Group = message.Params.GroupID.(string) @@ -276,7 +276,7 @@ func handleSendGroupMsg(client callapi.Client, api openapi.OpenAPI, apiv2 openap if err != nil { mylog.Printf("发送文本报错信息失败: %v", err) } - if ret.Message.Ret == 22009 { + if ret != nil && ret.Message.Ret == 22009 { mylog.Printf("信息发送失败,加入到队列中,下次被动信息进行发送") var pair echo.MessageGroupPair pair.Group = message.Params.GroupID.(string) @@ -304,7 +304,7 @@ func handleSendGroupMsg(client callapi.Client, api openapi.OpenAPI, apiv2 openap if err != nil { mylog.Printf("发送图片失败: %v", err) } - if ret.Message.Ret == 22009 { + if ret != nil && ret.Message.Ret == 22009 { mylog.Printf("信息发送失败,加入到队列中,下次被动信息进行发送") var pair echo.MessageGroupPair pair.Group = message.Params.GroupID.(string) @@ -759,7 +759,7 @@ func SendStackMessages(apiv2 openapi.OpenAPI, messageid string, originalGroupID } // 检查错误码 - if ret.Message.Ret == 22009 { + if ret != nil && ret.Message.Ret == 22009 { mylog.Printf("信息再次发送失败,加入到队列中,下次被动信息进行发送") echo.PushGlobalStack(pair) } From ddd33c38f04a053680ff0e886f49920420c2b4f4 Mon Sep 17 00:00:00 2001 From: cosmo Date: Tue, 5 Dec 2023 01:37:33 +0800 Subject: [PATCH 2/6] beta77 --- handlers/send_group_msg.go | 1 + 1 file changed, 1 insertion(+) diff --git a/handlers/send_group_msg.go b/handlers/send_group_msg.go index bda0e112..1cab51fb 100644 --- a/handlers/send_group_msg.go +++ b/handlers/send_group_msg.go @@ -116,6 +116,7 @@ func handleSendGroupMsg(client callapi.Client, api openapi.OpenAPI, apiv2 openap message.Params.GroupID = originalGroupID if SSM { mylog.Printf("正在使用Msgid:%v 补发之前失败的主动信息,请注意AtoP不要设置超过3,否则可能会影响正常信息发送", messageID) + mylog.Printf("originalGroupID:%v ", originalGroupID) SendStackMessages(apiv2, messageID, originalGroupID) } mylog.Println("群组发信息messageText:", messageText) From 43c88a3de56ece59a62c823f011f94e919600c4d Mon Sep 17 00:00:00 2001 From: cosmo Date: Tue, 5 Dec 2023 01:48:31 +0800 Subject: [PATCH 3/6] beta77 --- handlers/send_group_msg.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/handlers/send_group_msg.go b/handlers/send_group_msg.go index 1cab51fb..b1d2e562 100644 --- a/handlers/send_group_msg.go +++ b/handlers/send_group_msg.go @@ -115,8 +115,8 @@ func handleSendGroupMsg(client callapi.Client, api openapi.OpenAPI, apiv2 openap } message.Params.GroupID = originalGroupID if SSM { - mylog.Printf("正在使用Msgid:%v 补发之前失败的主动信息,请注意AtoP不要设置超过3,否则可能会影响正常信息发送", messageID) - mylog.Printf("originalGroupID:%v ", originalGroupID) + //mylog.Printf("正在使用Msgid:%v 补发之前失败的主动信息,请注意AtoP不要设置超过3,否则可能会影响正常信息发送", messageID) + //mylog.Printf("originalGroupID:%v ", originalGroupID) SendStackMessages(apiv2, messageID, originalGroupID) } mylog.Println("群组发信息messageText:", messageText) @@ -744,8 +744,10 @@ func uploadMedia(ctx context.Context, groupID string, richMediaMessage *dto.Rich // 发送栈中的消息 func SendStackMessages(apiv2 openapi.OpenAPI, messageid string, originalGroupID string) { count := config.GetAtoPCount() + mylog.Printf("取出数量: %v", count) pairs := echo.PopGlobalStackMulti(count) for _, pair := range pairs { + mylog.Printf("%v: %v", pair.Group, originalGroupID) if pair.Group == originalGroupID { // 发送消息 messageID := pair.GroupMessage.MsgID From 0e644e6462976db617a876beffdfb7994988443d Mon Sep 17 00:00:00 2001 From: cosmo Date: Tue, 5 Dec 2023 02:08:35 +0800 Subject: [PATCH 4/6] beta77 --- echo/echo.go | 25 ++++++++++++++++--------- handlers/send_group_msg.go | 6 ++++-- 2 files changed, 20 insertions(+), 11 deletions(-) diff --git a/echo/echo.go b/echo/echo.go index e15af59b..91f22e9a 100644 --- a/echo/echo.go +++ b/echo/echo.go @@ -195,23 +195,30 @@ func PushGlobalStack(pair MessageGroupPair) { globalMessageGroupStack.stack = append(globalMessageGroupStack.stack, pair) } -// PopGlobalStackMulti 从全局栈中取出指定数量的 MessageGroupPair +// PopGlobalStackMulti 从全局栈中取出指定数量的 MessageGroupPair，但不删除它们 func PopGlobalStackMulti(count int) []MessageGroupPair { globalMessageGroupStack.mu.Lock() defer globalMessageGroupStack.mu.Unlock() - // 如果 count 为 0 或栈为空，则不取出元素 if count == 0 || len(globalMessageGroupStack.stack) == 0 { return nil } - if count >= len(globalMessageGroupStack.stack) { - result := globalMessageGroupStack.stack - globalMessageGroupStack.stack = nil - return result + if count > len(globalMessageGroupStack.stack) { + count = len(globalMessageGroupStack.stack) } - result := globalMessageGroupStack.stack[len(globalMessageGroupStack.stack)-count:] - globalMessageGroupStack.stack = globalMessageGroupStack.stack[:len(globalMessageGroupStack.stack)-count] - return result + return globalMessageGroupStack.stack[:count] +} + +// RemoveFromGlobalStack 从全局栈中删除指定下标的元素 +func RemoveFromGlobalStack(index int) { + globalMessageGroupStack.mu.Lock() + defer globalMessageGroupStack.mu.Unlock() + + if index < 0 || index >= len(globalMessageGroupStack.stack) { + return // 下标越界检查 + } + + globalMessageGroupStack.stack = append(globalMessageGroupStack.stack[:index], globalMessageGroupStack.stack[index+1:]...) } diff --git a/handlers/send_group_msg.go b/handlers/send_group_msg.go index b1d2e562..ce2d5b2c 100644 --- a/handlers/send_group_msg.go +++ b/handlers/send_group_msg.go @@ -746,7 +746,7 @@ func SendStackMessages(apiv2 openapi.OpenAPI, messageid string, originalGroupID count := config.GetAtoPCount() mylog.Printf("取出数量: %v", count) pairs := echo.PopGlobalStackMulti(count) - for _, pair := range pairs { + for i, pair := range pairs { mylog.Printf("%v: %v", pair.Group, originalGroupID) if pair.Group == originalGroupID { // 发送消息 @@ -758,7 +758,9 @@ func SendStackMessages(apiv2 openapi.OpenAPI, messageid string, originalGroupID ret, err := apiv2.PostGroupMessage(context.TODO(), pair.Group, pair.GroupMessage) if err != nil { mylog.Printf("发送组合消息失败: %v", err) - continue // 其他错误处理 + continue + } else { + echo.RemoveFromGlobalStack(i) } // 检查错误码 From b5aed78e5accbfdd4a0657eea4a48141d9231378 Mon Sep 17 00:00:00 2001 From: cosmo Date: Tue, 5 Dec 2023 17:42:06 +0800 Subject: [PATCH 5/6] beta79 --- Processor/Processor.go | 26 ++++++++ config/config.go | 13 ++++ handlers/message_parser.go | 12 ++-- handlers/send_group_msg.go | 128 +++++++++++++++++++++++++++++++++++- template/config_template.go | 1 + 5 files changed, 175 insertions(+), 5 deletions(-) diff --git a/Processor/Processor.go b/Processor/Processor.go index 493f0759..2619bba1 100644 --- a/Processor/Processor.go +++ b/Processor/Processor.go @@ -274,13 +274,18 @@ func (p *Processors) HandleFrameworkCommand(messageText string, data interface{} var err error var now, new, newpro1, newpro2 string var realid, realid2 string + var guildid, guilduserid string switch v := data.(type) { case *dto.WSGroupATMessageData: realid = v.Author.ID case *dto.WSATMessageData: realid = v.Author.ID + guildid = v.GuildID + guilduserid = v.Author.ID case *dto.WSMessageData: realid = v.Author.ID + guildid = v.GuildID + guilduserid = v.Author.ID case *dto.WSDirectMessageData: realid = v.Author.ID case *dto.WSC2CMessageData: @@ -320,6 +325,27 @@ func (p *Processors) HandleFrameworkCommand(messageText string, data interface{} virtualValueIncluded = contains(masterIDs, new) } + //unlock指令 + if Type == "guild" && strings.HasPrefix(cleanedMessage, config.GetUnlockPrefix()) { + dm := &dto.DirectMessageToCreate{ + SourceGuildID: guildid, + RecipientID: guilduserid, + } + cdm, err := p.Api.CreateDirectMessage(context.TODO(), dm) + if err != nil { + mylog.Printf("unlock指令创建dm失败:%v", err) + } + msg := &dto.MessageToCreate{ + Content: "欢迎使用Gensokyo框架部署QQ机器人", + MsgType: 0, + MsgID: "", + } + _, err = p.Api.PostDirectMessage(context.TODO(), cdm, msg) + if err != nil { + mylog.Printf("unlock指令发送失败:%v", err) + } + } + // me指令处理逻辑 if strings.HasPrefix(cleanedMessage, config.GetMePrefix()) { if err != nil { diff --git a/config/config.go b/config/config.go index 7af3419a..c7e7b69a 100644 --- a/config/config.go +++ b/config/config.go @@ -91,6 +91,7 @@ type Settings struct { ReconnecTimes int `yaml:"reconnect_times"` HeartBeatInterval int `yaml:"heart_beat_interval"` LaunchReconectTimes int `yaml:"launch_reconnect_times"` + UnlockPrefix string `yaml:"unlock_prefix"` } // LoadConfig 从文件中加载配置并初始化单例配置 @@ -1093,3 +1094,15 @@ func GetLaunchReconectTimes() int { } return instance.Settings.LaunchReconectTimes } + +// 获取GetUnlockPrefix +func GetUnlockPrefix() string { + mu.Lock() + defer mu.Unlock() + + if instance == nil { + mylog.Println("Warning: instance is nil when trying to UnlockPrefix value.") + return "/unlock" + } + return instance.Settings.UnlockPrefix +} diff --git a/handlers/message_parser.go b/handlers/message_parser.go index 67ff5db8..b03ba32d 100644 --- a/handlers/message_parser.go +++ b/handlers/message_parser.go @@ -147,21 +147,25 @@ func parseMessageContent(paramsMessage callapi.ParamsContent) (string, map[strin } else { localRecordPattern = regexp.MustCompile(`\[CQ:record,file=file://([^\]]+?)\]`) } - urlImagePattern := regexp.MustCompile(`\[CQ:image,file=https?://(.+)\]`) + httpUrlImagePattern := regexp.MustCompile(`\[CQ:image,file=http://(.+)\]`) + httpsUrlImagePattern := regexp.MustCompile(`\[CQ:image,file=https://(.+)\]`) base64ImagePattern := regexp.MustCompile(`\[CQ:image,file=base64://(.+)\]`) base64RecordPattern := regexp.MustCompile(`\[CQ:record,file=base64://(.+)\]`) - urlRecordPattern := regexp.MustCompile(`\[CQ:record,file=https?://(.+)\]`) + httpUrlRecordPattern := regexp.MustCompile(`\[CQ:record,file=http://(.+)\]`) + httpsUrlRecordPattern := regexp.MustCompile(`\[CQ:record,file=https://(.+)\]`) patterns := []struct { key string pattern *regexp.Regexp }{ {"local_image", localImagePattern}, - {"url_image", urlImagePattern}, + {"url_image", httpUrlImagePattern}, + {"url_images", httpsUrlImagePattern}, {"base64_image", base64ImagePattern}, {"base64_record", base64RecordPattern}, {"local_record", localRecordPattern}, - {"url_record", urlRecordPattern}, + {"url_record", httpUrlRecordPattern}, + {"url_records", httpsUrlRecordPattern}, } foundItems := make(map[string][]string) diff --git a/handlers/send_group_msg.go b/handlers/send_group_msg.go index ce2d5b2c..a15ec981 100644 --- a/handlers/send_group_msg.go +++ b/handlers/send_group_msg.go @@ -533,6 +533,71 @@ func generateGroupMessage(id string, foundItems map[string][]string, messageText newpiclink = "http://" + imageURLs[0] } + // 发链接图片 + return &dto.RichMediaMessage{ + EventID: id, + FileType: 1, // 1代表图片 + URL: newpiclink, // 新图片链接 + Content: "", // 这个字段文档没有了 + SrvSendMsg: false, + } + } else if imageURLs, ok := foundItems["url_images"]; ok && len(imageURLs) > 0 { + var newpiclink string + if config.GetUrlPicTransfer() { + // 从URL下载图片 + resp, err := http.Get("https://" + imageURLs[0]) + if err != nil { + mylog.Printf("Error downloading the image: %v", err) + return &dto.MessageToCreate{ + Content: "错误: 下载图片失败", + MsgID: id, + MsgSeq: msgseq, + MsgType: 0, // 默认文本类型 + } + } + defer resp.Body.Close() + + // 读取图片数据 + imageData, err := io.ReadAll(resp.Body) + if err != nil { + mylog.Printf("Error reading the image data: %v", err) + return &dto.MessageToCreate{ + Content: "错误: 读取图片数据失败", + MsgID: id, + MsgSeq: msgseq, + MsgType: 0, + } + } + + // 转换为base64 + base64Encoded := base64.StdEncoding.EncodeToString(imageData) + + // 上传图片并获取新的URL + newURL, err := images.UploadBase64ImageToServer(base64Encoded) + if err != nil { + mylog.Printf("Error uploading base64 encoded image: %v", err) + return &dto.MessageToCreate{ + Content: "错误: 上传图片失败", + MsgID: id, + MsgSeq: msgseq, + MsgType: 0, + } + } + // 将图片链接缩短 避免 url not allow + // if config.GetLotusValue() { + // // 连接到另一个gensokyo + // newURL = url.GenerateShortURL(newURL) + // } else { + // // 自己是主节点 + // newURL = url.GenerateShortURL(newURL) + // // 使用getBaseURL函数来获取baseUrl并与newURL组合 + // newURL = url.GetBaseURL() + "/url/" + newURL + // } + newpiclink = newURL + } else { + newpiclink = "https://" + imageURLs[0] + } + // 发链接图片 return &dto.RichMediaMessage{ EventID: id, @@ -624,7 +689,68 @@ func generateGroupMessage(id string, foundItems map[string][]string, messageText if err != nil { mylog.Printf("Error uploading base64 encoded image: %v", err) return &dto.MessageToCreate{ - Content: "错误: 上传图片失败", + Content: "错误: 上传语音失败", + MsgID: id, + MsgSeq: msgseq, + MsgType: 0, + } + } + + // 发链接语音 + return &dto.RichMediaMessage{ + EventID: id, + FileType: 3, // 3代表语音 + URL: newURL, // 新语音链接 + Content: "", // 这个字段文档没有了 + SrvSendMsg: false, + } + } else if imageURLs, ok := foundItems["url_records"]; ok && len(imageURLs) > 0 { + // 从URL下载语音 + resp, err := http.Get("https://" + imageURLs[0]) + if err != nil { + mylog.Printf("Error downloading the record: %v", err) + return &dto.MessageToCreate{ + Content: "错误: 下载语音失败", + MsgID: id, + MsgSeq: msgseq, + MsgType: 0, // 默认文本类型 + } + } + defer resp.Body.Close() + + // 读取语音数据 + recordData, err := io.ReadAll(resp.Body) + if err != nil { + mylog.Printf("Error reading the record data: %v", err) + return &dto.MessageToCreate{ + Content: "错误: 读取语音数据失败", + MsgID: id, + MsgSeq: msgseq, + MsgType: 0, + } + } + //判断并转码 + if !silk.IsAMRorSILK(recordData) { + mt, ok := silk.CheckAudio(bytes.NewReader(recordData)) + if !ok { + mylog.Errorf("voice type error: " + mt) + return nil + } + recordData = silk.EncoderSilk(recordData) + mylog.Errorf("音频转码ing") + if err != nil { + return nil + } + } + // 转换为base64 + base64Encoded := base64.StdEncoding.EncodeToString(recordData) + + // 上传语音并获取新的URL + newURL, err := images.UploadBase64RecordToServer(base64Encoded) + if err != nil { + mylog.Printf("Error uploading base64 encoded image: %v", err) + return &dto.MessageToCreate{ + Content: "错误: 上传语音失败", MsgID: id, MsgSeq: msgseq, MsgType: 0, diff --git a/template/config_template.go b/template/config_template.go index 0df0aa95..2d7a3899 100644 --- a/template/config_template.go +++ b/template/config_template.go @@ -115,6 +115,7 @@ settings: bind_prefix : "/bind" #需设置 #增强配置项 master_id 可触发 me_prefix : "/me" #需设置 #增强配置项 master_id 可触发 + unlock_prefix : "/unlock" #频道私信卡住了? gsk可以帮到你 在任意子频道发送unlock 你会收到来自机器人的频道私信 #穿透\cos\oss类配置(可选!) frp_port : "0" #不使用请保持为0,frp的端口,frp有内外端口,请在frp软件设置gensokyo的port,并将frp显示的对外端口填入这里 From dfe4d1bd363c8e8126a0c52f55895b2570d2fbf5 Mon Sep 17 00:00:00 2001 From: cosmo Date: Tue, 5 Dec 2023 18:26:39 +0800 Subject: [PATCH 6/6] beta79 --- Processor/ProcessC2CMessage.go | 14 +++--- Processor/ProcessChannelDirectMessage.go | 6 +-- Processor/ProcessGroupMessage.go | 21 ++++---- Processor/ProcessGuildATMessage.go | 4 +- Processor/ProcessGuildNormalMessage.go | 4 +- config/config.go | 11 ++++ handlers/message_parser.go | 64 +++++++++++++++--------- idmap/service.go | 4 +- template/config_template.go | 1 + template/config_template.yml | 3 +- 10 files changed, 80 insertions(+), 52 deletions(-) diff --git a/Processor/ProcessC2CMessage.go b/Processor/ProcessC2CMessage.go index e3e31e55..3bd90c8b 100644 --- a/Processor/ProcessC2CMessage.go +++ b/Processor/ProcessC2CMessage.go @@ -70,7 +70,7 @@ func (p *Processors) ProcessC2CMessage(data *dto.WSC2CMessageData) error { } } //转换at - messageText := handlers.RevertTransformedText(data, "group_private", p.Api, p.Apiv2) + messageText := handlers.RevertTransformedText(data, "group_private", p.Api, p.Apiv2, userid64) if messageText == "" { mylog.Printf("信息被自定义黑白名单拦截") return nil @@ -125,12 +125,6 @@ func (p *Processors) ProcessC2CMessage(data *dto.WSC2CMessageData) error { } else { //将私聊信息转化为群信息(特殊需求情况下) - //转换at - messageText := handlers.RevertTransformedText(data, "group_private", p.Api, p.Apiv2) - if messageText == "" { - mylog.Printf("信息被自定义黑白名单拦截") - return nil - } //转换appid AppIDString := strconv.FormatUint(p.Settings.AppID, 10) //构造echo @@ -156,6 +150,12 @@ func (p *Processors) ProcessC2CMessage(data *dto.WSC2CMessageData) error { mylog.Fatalf("Error storing ID: %v", err) } } + //转换at + messageText := handlers.RevertTransformedText(data, "group_private", p.Api, p.Apiv2, userid64) + if messageText == "" { + mylog.Printf("信息被自定义黑白名单拦截") + return nil + } //框架内指令 p.HandleFrameworkCommand(messageText, data, "group_private") //映射str的messageID到int diff --git a/Processor/ProcessChannelDirectMessage.go b/Processor/ProcessChannelDirectMessage.go index 93970c64..93de2c6e 100644 --- a/Processor/ProcessChannelDirectMessage.go +++ b/Processor/ProcessChannelDirectMessage.go @@ -90,7 +90,7 @@ func (p *Processors) ProcessChannelDirectMessage(data *dto.WSDirectMessageData) } messageID := int(messageID64) //转换at - messageText := handlers.RevertTransformedText(data, "guild_private", p.Api, p.Apiv2) + messageText := handlers.RevertTransformedText(data, "guild_private", p.Api, p.Apiv2, userid64) if messageText == "" { mylog.Printf("信息被自定义黑白名单拦截") return nil @@ -150,7 +150,7 @@ func (p *Processors) ProcessChannelDirectMessage(data *dto.WSDirectMessageData) //获取s s := client.GetGlobalS() //转换at - messageText := handlers.RevertTransformedText(data, "guild_private", p.Api, p.Apiv2) + messageText := handlers.RevertTransformedText(data, "guild_private", p.Api, p.Apiv2, 10000) //todo 这里未转换 if messageText == "" { mylog.Printf("信息被自定义黑白名单拦截") return nil @@ -278,7 +278,7 @@ func (p *Processors) ProcessChannelDirectMessage(data *dto.WSDirectMessageData) //直接储存 适用于私信场景私聊 idmap.WriteConfigv2(data.ChannelID, "guild_id", data.GuildID) //转换at - messageText := handlers.RevertTransformedText(data, "guild_private", p.Api, p.Apiv2) + messageText := handlers.RevertTransformedText(data, "guild_private", p.Api, p.Apiv2, userid64) if messageText == "" { mylog.Printf("信息被自定义黑白名单拦截") return nil diff --git a/Processor/ProcessGroupMessage.go b/Processor/ProcessGroupMessage.go index 735d3959..75014b00 100644 --- a/Processor/ProcessGroupMessage.go +++ b/Processor/ProcessGroupMessage.go @@ -21,17 +21,6 @@ func (p *Processors) ProcessGroupMessage(data *dto.WSGroupATMessageData) error { // 获取s s := client.GetGlobalS() - // 转换at - messageText := handlers.RevertTransformedText(data, "group", p.Api, p.Apiv2) - if messageText == "" { - mylog.Printf("信息被自定义黑白名单拦截") - return nil - } - //群没有at,但用户可以选择加一个 - if config.GetAddAtGroup() { - messageText = "[CQ:at,qq=" + config.GetAppIDStr() + "] " + messageText - } - // 转换appid AppIDString := strconv.FormatUint(p.Settings.AppID, 10) @@ -66,6 +55,16 @@ func (p *Processors) ProcessGroupMessage(data *dto.WSGroupATMessageData) error { return nil } } + // 转换at + messageText := handlers.RevertTransformedText(data, "group", p.Api, p.Apiv2, GroupID64) + if messageText == "" { + mylog.Printf("信息被自定义黑白名单拦截") + return nil + } + //群没有at,但用户可以选择加一个 + if config.GetAddAtGroup() { + messageText = "[CQ:at,qq=" + config.GetAppIDStr() + "] " + messageText + } //框架内指令 p.HandleFrameworkCommand(messageText, data, "group") //映射str的messageID到int diff --git a/Processor/ProcessGuildATMessage.go b/Processor/ProcessGuildATMessage.go index 4ce622a6..faf4ab0a 100644 --- a/Processor/ProcessGuildATMessage.go +++ b/Processor/ProcessGuildATMessage.go @@ -27,7 +27,7 @@ func (p *Processors) ProcessGuildATMessage(data *dto.WSATMessageData) error { //获取s s := client.GetGlobalS() //转换at - messageText := handlers.RevertTransformedText(data, "guild", p.Api, p.Apiv2) + messageText := handlers.RevertTransformedText(data, "guild", p.Api, p.Apiv2, 10000) //todo 这里未转换 if messageText == "" { mylog.Printf("信息被自定义黑白名单拦截") return nil @@ -151,7 +151,7 @@ func (p *Processors) ProcessGuildATMessage(data *dto.WSATMessageData) error { //转成int再互转 idmap.WriteConfigv2(fmt.Sprint(ChannelID64), "guild_id", data.GuildID) //转换at和图片 - messageText := handlers.RevertTransformedText(data, "guild", p.Api, p.Apiv2) + messageText := handlers.RevertTransformedText(data, "guild", p.Api, p.Apiv2, ChannelID64) if messageText == "" { mylog.Printf("信息被自定义黑白名单拦截") return nil diff --git a/Processor/ProcessGuildNormalMessage.go b/Processor/ProcessGuildNormalMessage.go index 18a6ea15..3add806a 100644 --- a/Processor/ProcessGuildNormalMessage.go +++ b/Processor/ProcessGuildNormalMessage.go @@ -26,7 +26,7 @@ func (p *Processors) ProcessGuildNormalMessage(data *dto.WSMessageData) error { //获取s s := client.GetGlobalS() //转换at - messageText := handlers.RevertTransformedText(data, "guild", p.Api, p.Apiv2) + messageText := handlers.RevertTransformedText(data, "guild", p.Api, p.Apiv2, 10000) //这里未转换 if messageText == "" { mylog.Printf("信息被自定义黑白名单拦截") return nil @@ -155,7 +155,7 @@ func (p *Processors) ProcessGuildNormalMessage(data *dto.WSMessageData) error { //转成int再互转 idmap.WriteConfigv2(fmt.Sprint(ChannelID64), "guild_id", data.GuildID) //转换at - messageText := handlers.RevertTransformedText(data, "guild", p.Api, p.Apiv2) + messageText := handlers.RevertTransformedText(data, "guild", p.Api, p.Apiv2, ChannelID64) if messageText == "" { mylog.Printf("信息被自定义黑白名单拦截") return nil diff --git a/config/config.go b/config/config.go index c7e7b69a..0d15edee 100644 --- a/config/config.go +++ b/config/config.go @@ -92,6 +92,7 @@ type Settings struct { HeartBeatInterval int `yaml:"heart_beat_interval"` LaunchReconectTimes int `yaml:"launch_reconnect_times"` UnlockPrefix string `yaml:"unlock_prefix"` + WhiteBypass []int64 `yaml:"white_bypass"` } // LoadConfig 从文件中加载配置并初始化单例配置 @@ -1106,3 +1107,13 @@ func GetUnlockPrefix() string { } return instance.Settings.UnlockPrefix } + +// 获取白名单例外群数组 +func GetWhiteBypass() []int64 { + mu.Lock() + defer mu.Unlock() + if instance != nil { + return instance.Settings.WhiteBypass + } + return nil // 返回nil，如果instance为nil +} diff --git a/handlers/message_parser.go b/handlers/message_parser.go index b03ba32d..41467482 100644 --- a/handlers/message_parser.go +++ b/handlers/message_parser.go @@ -250,7 +250,7 @@ func transformMessageText(messageText string) string { } // 处理at和其他定形文到onebotv11格式(cq码) -func RevertTransformedText(data interface{}, msgtype string, api openapi.OpenAPI, apiv2 openapi.OpenAPI) string { +func RevertTransformedText(data interface{}, msgtype string, api openapi.OpenAPI, apiv2 openapi.OpenAPI, vgid int64) string { var msg *dto.Message var menumsg bool var messageText string @@ -330,36 +330,52 @@ func RevertTransformedText(data interface{}, msgtype string, api openapi.OpenAPI } } - //检查是否启用白名单模式 + // 检查是否启用白名单模式 if config.GetWhitePrefixMode() { - // 获取白名单数组 - whitePrefixes := config.GetWhitePrefixs() - // 加锁以安全地读取 TemporaryCommands - idmap.MutexT.Lock() - temporaryCommands := make([]string, len(idmap.TemporaryCommands)) - copy(temporaryCommands, idmap.TemporaryCommands) - idmap.MutexT.Unlock() - - // 合并白名单和临时指令 - allPrefixes := append(whitePrefixes, temporaryCommands...) - // 默认设置为不匹配 - matched := false - - // 遍历白名单数组，检查是否有匹配项 - for _, prefix := range allPrefixes { - if strings.HasPrefix(messageText, prefix) { - // 找到匹配项，保留 messageText 并跳出循环 - matched = true + // 获取白名单例外群数组（现在返回 int64 数组） + whiteBypass := config.GetWhiteBypass() + bypass := false + + // 检查vgid是否在白名单例外数组中 + for _, id := range whiteBypass { + if id == vgid { + bypass = true break } } - // 如果没有匹配项，则将 messageText 置为兜底回复 兜底回复可空 - if !matched { - messageText = "" - SendMessage(config.GetNoWhiteResponse(), data, msgtype, api, apiv2) + // 如果vgid不在白名单例外数组中，则应用白名单过滤 + if !bypass { + // 获取白名单数组 + whitePrefixes := config.GetWhitePrefixs() + // 加锁以安全地读取 TemporaryCommands + idmap.MutexT.Lock() + temporaryCommands := make([]string, len(idmap.TemporaryCommands)) + copy(temporaryCommands, idmap.TemporaryCommands) + idmap.MutexT.Unlock() + + // 合并白名单和临时指令 + allPrefixes := append(whitePrefixes, temporaryCommands...) + // 默认设置为不匹配 + matched := false + + // 遍历白名单数组，检查是否有匹配项 + for _, prefix := range allPrefixes { + if strings.HasPrefix(messageText, prefix) { + // 找到匹配项，保留 messageText 并跳出循环 + matched = true + break + } + } + + // 如果没有匹配项，则将 messageText 置为兜底回复 兜底回复可空 + if !matched { + messageText = "" + SendMessage(config.GetNoWhiteResponse(), data, msgtype, api, apiv2) + } } } + //检查是否启用黑名单模式 if config.GetBlackPrefixMode() { // 获取黑名单数组 diff --git a/idmap/service.go b/idmap/service.go index 1ae53b96..6fa525a6 100644 --- a/idmap/service.go +++ b/idmap/service.go @@ -783,12 +783,12 @@ func RetrieveVirtualValuev2Pro(realValue string, realValueSub string) (string, s return "", "", fmt.Errorf("error response from server: %s", response["error"]) } - firstValue, ok := response["firstValue"].(string) + firstValue, ok := response["id"].(string) if !ok { return "", "", fmt.Errorf("invalid response format for first value") } - secondValue, ok := response["secondValue"].(string) + secondValue, ok := response["subid"].(string) if !ok { return "", "", fmt.Errorf("invalid response format for second value") } diff --git a/template/config_template.go b/template/config_template.go index 2d7a3899..47cf5b9f 100644 --- a/template/config_template.go +++ b/template/config_template.go @@ -84,6 +84,7 @@ settings: white_prefix_mode : false #公域 过审用 指令白名单模式开关 如果审核严格 请开启并设置白名单指令 以白名单开头的指令会被通过,反之被拦截 white_prefixs : [""] #可设置多个 比如设置 机器人 测试 则只有信息以机器人 测试开头会相应 remove_prefix remove_at 需为true时生效 + white_bypass : [] #格式[1,2,3],白名单不生效的群或用户(私聊时),用于设置自己的灰度沙箱群/灰度沙箱私聊,避免开发测试时反复开关白名单的不便,请勿用于生产环境. No_White_Response : "" #默认不兜底,强烈建议设置一个友善的兜底回复,告知审核机器人已无隐藏指令,如:你输入的指令不对哦,@机器人来获取可用指令 black_prefix_mode : false #公私域 过审用 指令黑名单模式开关 过滤被审核打回的指令不响应 无需改机器人后端 diff --git a/template/config_template.yml b/template/config_template.yml index c9211ce8..6565cfa9 100644 --- a/template/config_template.yml +++ b/template/config_template.yml @@ -81,4 +81,5 @@ settings: AMsgRetryAsPMsg_Count : 1 #当主动信息发送失败时,自动转为后续的被动信息发送,需要开启Lazy message id,该配置项为每次跟随被动信息发送的信息数量,最大5,建议1-3 reconnect_times : 100 #反向ws连接失败后的重试次数,希望一直重试,可设置9999 heart_beat_interval : 10 #反向ws心跳间隔 单位秒 推荐5-10 - launch_reconnect_times : 1 #启动时尝试反向ws连接次数,建议先打开应用端再开启gensokyo,因为启动时连接会阻塞webui启动,默认只连接一次,可自行增大 \ No newline at end of file + launch_reconnect_times : 1 #启动时尝试反向ws连接次数,建议先打开应用端再开启gensokyo,因为启动时连接会阻塞webui启动,默认只连接一次,可自行增大 + white_bypass : [] #格式[1,2,3],白名单不生效的群,用于设置自己的灰度沙箱,避免测试时候反复开关白名单的不便. \ No newline at end of file