Skip to content

Commit

Permalink
Beta485 (#490)
Browse files Browse the repository at this point in the history 
* beta447

* beta448

* beta449

* beta450

* beta451

* beta452

* beta453

* beta454

* beta455

* btea455

* beta456

* beta457

* beta458

* beta460

* beta460

* beta461

* beta462

* beta463

* beta464

* beta465

* beta467

* beta468

* beta469

* beta470

* beta471

* beta472

* beta473

* beta473

* beta475

* beta476

* beta478

* beta479

* beta479

* beta480

* beta481

* beta482

* beta483

* beta484

* beta485
  • Loading branch information
@Hoshinonyaruko
Hoshinonyaruko authored Aug 20, 2024
1 parent 6f4deeb commit f3cd1b8
Showing 1 changed file with 105 additions and 97 deletions.
202 changes: 105 additions & 97 deletions handlers/message_parser.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -441,6 +441,8 @@ func SendGuildPrivateResponse(client callapi.Client, err error, message *callapi
func parseMessageContent(paramsMessage callapi.ParamsContent, message callapi.ActionMessage, client callapi.Client, api openapi.OpenAPI, apiv2 openapi.OpenAPI) (string, map[string][]string) {
messageText := ""

foundItems := make(map[string][]string)

switch message := paramsMessage.Message.(type) {
case string:
mylog.Printf("params.message is a string\n")
Expand All @@ -457,7 +459,6 @@ func parseMessageContent(paramsMessage callapi.ParamsContent, message callapi.Ac
messageText = ProcessCQAvatar(paramsMessage.GroupID.(string), messageText)
}
case []interface{}:
//多个映射组成的切片
mylog.Printf("params.message is a slice (segment_type_koishi)\n")
for _, segment := range message {
segmentMap, ok := segment.(map[string]interface{})
Expand All @@ -474,128 +475,130 @@ func parseMessageContent(paramsMessage callapi.ParamsContent, message callapi.Ac
switch segmentType {
case "text":
segmentContent, _ = segmentMap["data"].(map[string]interface{})["text"].(string)
// 应用替换规则
if config.GetEnableChangeWord() {
segmentContent = acnode.CheckWordOUT(segmentContent)
}
case "image":
fileContent, _ := segmentMap["data"].(map[string]interface{})["file"].(string)
segmentContent = "[CQ:image,file=" + fileContent + "]"
case "voice":
fileContent, _ := segmentMap["data"].(map[string]interface{})["file"].(string)
segmentContent = "[CQ:record,file=" + fileContent + "]"
case "record":
foundItems["image"] = append(foundItems["image"], fileContent)

case "voice", "record":
fileContent, _ := segmentMap["data"].(map[string]interface{})["file"].(string)
segmentContent = "[CQ:record,file=" + fileContent + "]"
foundItems["record"] = append(foundItems["record"], fileContent)

case "at":
qqNumber, _ := segmentMap["data"].(map[string]interface{})["qq"].(string)
segmentContent = "[CQ:at,qq=" + qqNumber + "]"
foundItems["at"] = append(foundItems["at"], qqNumber)

case "avatar":
qqNumber, _ := segmentMap["data"].(map[string]interface{})["qq"].(string)
var avatarCQCode string
if paramsMessage.GroupID == nil {
segmentContent, _ = GetAvatarCQCodeNoGroupID(qqNumber)
avatarCQCode, _ = GetAvatarCQCodeNoGroupID(qqNumber)
} else {
segmentContent, _ = GetAvatarCQCode(paramsMessage.GroupID.(string), qqNumber)
avatarCQCode, _ = GetAvatarCQCode(paramsMessage.GroupID.(string), qqNumber)
}
foundItems["avatar"] = append(foundItems["avatar"], avatarCQCode)

case "markdown":
mdContent, ok := segmentMap["data"].(map[string]interface{})["data"]
if ok {
var mdContentEncoded string
if mdContentMap, isMap := mdContent.(map[string]interface{}); isMap {
// mdContent是map[string]interface{}，按map处理
mdContentBytes, err := json.Marshal(mdContentMap)
if err != nil {
mylog.Printf("Error marshaling mdContentMap to JSON:%v", err)
continue
}
encoded := base64.StdEncoding.EncodeToString(mdContentBytes)
segmentContent = "[CQ:markdown,data=base64://" + encoded + "]"
mdContentEncoded = base64.StdEncoding.EncodeToString(mdContentBytes)
} else if mdContentStr, isString := mdContent.(string); isString {
// mdContent是string
if strings.HasPrefix(mdContentStr, "base64://") {
// 如果以base64://开头，直接使用
segmentContent = "[CQ:markdown,data=" + mdContentStr + "]"
mdContentEncoded = mdContentStr
} else {
// 处理实体化后的JSON文本
mdContentStr = strings.ReplaceAll(mdContentStr, "&", "&")
mdContentStr = strings.ReplaceAll(mdContentStr, "[", "[")
mdContentStr = strings.ReplaceAll(mdContentStr, "]", "]")
mdContentStr = strings.ReplaceAll(mdContentStr, ",", ",")

// 将处理过的字符串视为JSON对象，进行序列化和编码
var jsonMap map[string]interface{}
if err := json.Unmarshal([]byte(mdContentStr), &jsonMap); err != nil {
mylog.Printf("Error unmarshaling string to JSON:%v", err)
continue
}
mdContentBytes, err := json.Marshal(jsonMap)
if err != nil {
mylog.Printf("Error marshaling jsonMap to JSON:%v", err)
continue
}
encoded := base64.StdEncoding.EncodeToString(mdContentBytes)
segmentContent = "[CQ:markdown,data=base64://" + encoded + "]"
mdContentEncoded = base64.StdEncoding.EncodeToString(mdContentBytes)
}
} else {
mylog.Printf("Error marshaling markdown segment wrong type.")
continue
}
foundItems["markdown"] = append(foundItems["markdown"], mdContentEncoded)
} else {
mylog.Printf("Error marshaling markdown segment to interface,contain type but data is nil.")
mylog.Printf("Error: markdown segment data is nil.")
}

default:
mylog.Printf("Unhandled segment type: %s", segmentType)
}

messageText += segmentContent

}
case map[string]interface{}:
//单个映射
mylog.Printf("params.message is a map (segment_type_trss)\n")
messageType, _ := message["type"].(string)

switch messageType {
case "text":
messageText, _ = message["data"].(map[string]interface{})["text"].(string)
// 应用替换规则
if config.GetEnableChangeWord() {
messageText = acnode.CheckWordOUT(messageText)
}

case "image":
fileContent, _ := message["data"].(map[string]interface{})["file"].(string)
messageText = "[CQ:image,file=" + fileContent + "]"
case "voice":
fileContent, _ := message["data"].(map[string]interface{})["file"].(string)
messageText = "[CQ:record,file=" + fileContent + "]"
case "record":
foundItems["image"] = append(foundItems["image"], fileContent)

case "voice", "record":
fileContent, _ := message["data"].(map[string]interface{})["file"].(string)
messageText = "[CQ:record,file=" + fileContent + "]"
foundItems["record"] = append(foundItems["record"], fileContent)

case "at":
qqNumber, _ := message["data"].(map[string]interface{})["qq"].(string)
messageText = "[CQ:at,qq=" + qqNumber + "]"
foundItems["at"] = append(foundItems["at"], qqNumber)

case "avatar":
qqNumber, _ := message["data"].(map[string]interface{})["qq"].(string)
var avatarCQCode string
if paramsMessage.GroupID == nil {
messageText, _ = GetAvatarCQCodeNoGroupID(qqNumber)
avatarCQCode, _ = GetAvatarCQCodeNoGroupID(qqNumber)
} else {
messageText, _ = GetAvatarCQCode(paramsMessage.GroupID.(string), qqNumber)
avatarCQCode, _ = GetAvatarCQCode(paramsMessage.GroupID.(string), qqNumber)
}
foundItems["avatar"] = append(foundItems["avatar"], avatarCQCode)

case "markdown":
mdContent, ok := message["data"].(map[string]interface{})["data"]
if ok {
var mdContentEncoded string
if mdContentMap, isMap := mdContent.(map[string]interface{}); isMap {
// mdContent是map[string]interface{}，按map处理
mdContentBytes, err := json.Marshal(mdContentMap)
if err != nil {
mylog.Printf("Error marshaling mdContentMap to JSON:%v", err)
}
encoded := base64.StdEncoding.EncodeToString(mdContentBytes)
messageText = "[CQ:markdown,data=base64://" + encoded + "]"
mdContentEncoded = base64.StdEncoding.EncodeToString(mdContentBytes)
} else if mdContentStr, isString := mdContent.(string); isString {
// mdContent是string
if strings.HasPrefix(mdContentStr, "base64://") {
// 如果以base64://开头，直接使用
messageText = "[CQ:markdown,data=" + mdContentStr + "]"
mdContentEncoded = mdContentStr
} else {
// 处理实体化后的JSON文本
mdContentStr = strings.ReplaceAll(mdContentStr, "&", "&")
mdContentStr = strings.ReplaceAll(mdContentStr, "[", "[")
mdContentStr = strings.ReplaceAll(mdContentStr, "]", "]")
mdContentStr = strings.ReplaceAll(mdContentStr, ",", ",")

// 将处理过的字符串视为JSON对象，进行序列化和编码
var jsonMap map[string]interface{}
if err := json.Unmarshal([]byte(mdContentStr), &jsonMap); err != nil {
mylog.Printf("Error unmarshaling string to JSON:%v", err)
Expand All @@ -604,16 +607,20 @@ func parseMessageContent(paramsMessage callapi.ParamsContent, message callapi.Ac
if err != nil {
mylog.Printf("Error marshaling jsonMap to JSON:%v", err)
}
encoded := base64.StdEncoding.EncodeToString(mdContentBytes)
messageText = "[CQ:markdown,data=base64://" + encoded + "]"
mdContentEncoded = base64.StdEncoding.EncodeToString(mdContentBytes)
}
} else {
mylog.Printf("Error marshaling mdContent wrong type.")
mylog.Printf("Error: markdown content has an unexpected type.")
}
foundItems["markdown"] = append(foundItems["markdown"], mdContentEncoded)
} else {
mylog.Printf("Error marshaling markdown segment to interface,contain type but data is nil.")
mylog.Printf("Error: markdown segment data is nil.")
}

default:
mylog.Printf("Unhandled message type: %s", messageType)
}

default:
mylog.Println("Unsupported message format: params.message field is not a string, map or slice")
}
Expand All @@ -626,61 +633,62 @@ func parseMessageContent(paramsMessage callapi.ParamsContent, message callapi.Ac
messageText = transformMessageTextAt(messageText, paramsMessage.GroupID.(string))
}

//mylog.Printf(messageText)

// 正则表达式部分
var localImagePattern *regexp.Regexp
var localRecordPattern *regexp.Regexp
if runtime.GOOS == "windows" {
localImagePattern = regexp.MustCompile(`\[CQ:image,file=file:///([^\]]+?)\]`)
} else {
localImagePattern = regexp.MustCompile(`\[CQ:image,file=file://([^\]]+?)\]`)
}
if runtime.GOOS == "windows" {
localRecordPattern = regexp.MustCompile(`\[CQ:record,file=file:///([^\]]+?)\]`)
} else {
localRecordPattern = regexp.MustCompile(`\[CQ:record,file=file://([^\]]+?)\]`)
}
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://(.+?)\]`)
httpUrlRecordPattern := regexp.MustCompile(`\[CQ:record,file=http://(.+?)\]`)
httpsUrlRecordPattern := regexp.MustCompile(`\[CQ:record,file=https://(.+?)\]`)
httpUrlVideoPattern := regexp.MustCompile(`\[CQ:video,file=http://(.+?)\]`)
httpsUrlVideoPattern := regexp.MustCompile(`\[CQ:video,file=https://(.+?)\]`)
mdPattern := regexp.MustCompile(`\[CQ:markdown,data=base64://(.+?)\]`)
qqMusicPattern := regexp.MustCompile(`\[CQ:music,type=qq,id=(\d+)\]`)

patterns := []struct {
key string
pattern *regexp.Regexp
}{
{"local_image", localImagePattern},
{"url_image", httpUrlImagePattern},
{"url_images", httpsUrlImagePattern},
{"base64_image", base64ImagePattern},
{"base64_record", base64RecordPattern},
{"local_record", localRecordPattern},
{"url_record", httpUrlRecordPattern},
{"url_records", httpsUrlRecordPattern},
{"markdown", mdPattern},
{"qqmusic", qqMusicPattern},
{"url_video", httpUrlVideoPattern},
{"url_videos", httpsUrlVideoPattern},
}
// 当匹配到复古cq码上报类型,使用低效率正则.
if _, ok := paramsMessage.Message.(string); ok {
// 正则表达式部分
var localImagePattern *regexp.Regexp
var localRecordPattern *regexp.Regexp
if runtime.GOOS == "windows" {
localImagePattern = regexp.MustCompile(`\[CQ:image,file=file:///([^\]]+?)\]`)
} else {
localImagePattern = regexp.MustCompile(`\[CQ:image,file=file://([^\]]+?)\]`)
}
if runtime.GOOS == "windows" {
localRecordPattern = regexp.MustCompile(`\[CQ:record,file=file:///([^\]]+?)\]`)
} else {
localRecordPattern = regexp.MustCompile(`\[CQ:record,file=file://([^\]]+?)\]`)
}
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://(.+?)\]`)
httpUrlRecordPattern := regexp.MustCompile(`\[CQ:record,file=http://(.+?)\]`)
httpsUrlRecordPattern := regexp.MustCompile(`\[CQ:record,file=https://(.+?)\]`)
httpUrlVideoPattern := regexp.MustCompile(`\[CQ:video,file=http://(.+?)\]`)
httpsUrlVideoPattern := regexp.MustCompile(`\[CQ:video,file=https://(.+?)\]`)
mdPattern := regexp.MustCompile(`\[CQ:markdown,data=base64://(.+?)\]`)
qqMusicPattern := regexp.MustCompile(`\[CQ:music,type=qq,id=(\d+)\]`)

patterns := []struct {
key string
pattern *regexp.Regexp
}{
{"local_image", localImagePattern},
{"url_image", httpUrlImagePattern},
{"url_images", httpsUrlImagePattern},
{"base64_image", base64ImagePattern},
{"base64_record", base64RecordPattern},
{"local_record", localRecordPattern},
{"url_record", httpUrlRecordPattern},
{"url_records", httpsUrlRecordPattern},
{"markdown", mdPattern},
{"qqmusic", qqMusicPattern},
{"url_video", httpUrlVideoPattern},
{"url_videos", httpsUrlVideoPattern},
}

foundItems := make(map[string][]string)
for _, pattern := range patterns {
matches := pattern.pattern.FindAllStringSubmatch(messageText, -1)
for _, match := range matches {
if len(match) > 1 {
foundItems[pattern.key] = append(foundItems[pattern.key], match[1])
for _, pattern := range patterns {
matches := pattern.pattern.FindAllStringSubmatch(messageText, -1)
for _, match := range matches {
if len(match) > 1 {
foundItems[pattern.key] = append(foundItems[pattern.key], match[1])
}
}
// 移动替换操作到这里，确保所有匹配都被处理后再进行替换
messageText = pattern.pattern.ReplaceAllString(messageText, "")
}
// 移动替换操作到这里，确保所有匹配都被处理后再进行替换
messageText = pattern.pattern.ReplaceAllString(messageText, "")
}

//最后再处理Url
messageText = transformMessageTextUrl(messageText, message, client, api, apiv2)

Expand Down

0 comments on commit f3cd1b8

Please sign in to comment.