feat: send recode adjust

2025-12-04 21:16:32 +08:00
parent 0639f5ec6d
commit ff194daef2
22 changed files with 743 additions and 424 deletions
@@ -0,0 +1,83 @@
+package channels
+
+import (
+	"encoding/json"
+	"fmt"
+	"message-nest/models"
+	"message-nest/pkg/message"
+	"message-nest/service/send_way_service"
+	"regexp"
+	"strings"
+)
+
+type CustomChannel struct{ *BaseChannel }
+
+func NewCustomChannel() *CustomChannel {
+	return &CustomChannel{BaseChannel: NewBaseChannel(MessageTypeCustom, []string{FormatTypeText})}
+}
+
+func (c *CustomChannel) FormatContent(content *UnifiedMessageContent) (string, string, error) {
+	if content.HasText() {
+		return FormatTypeText, content.Text, nil
+	}
+	if content.HasMarkdown() {
+		return FormatTypeText, markdownToText(content.Markdown), nil
+	}
+	if content.HasHTML() {
+		return FormatTypeText, htmlToText(content.HTML), nil
+	}
+	return FormatTypeText, "", nil
+}
+
+func (c *CustomChannel) SendUnified(msgObj interface{}, ins models.SendTasksIns, content *UnifiedMessageContent) (string, string) {
+	auth, ok := msgObj.(send_way_service.WayDetailCustom)
+	if !ok {
+		return "", "类型转换失败"
+	}
+	_, formattedContent, err := c.FormatContent(content)
+	if err != nil {
+		return "", err.Error()
+	}
+	cli := message.CustomWebhook{}
+	data, _ := json.Marshal(formattedContent)
+	dataStr := strings.Trim(string(data), "\"")
+	bodyStr := strings.Replace(auth.Body, "TEXT", dataStr, -1)
+	res, err := cli.Request(auth.Webhook, bodyStr)
+	var errMsg string
+	if err != nil {
+		errMsg = fmt.Sprintf("发送失败：%s", err.Error())
+	}
+	return string(res), errMsg
+}
+
+func markdownToText(md string) string {
+	t := md
+	t = regexp.MustCompile(`#{1,6}\s+`).ReplaceAllString(t, "")
+	t = regexp.MustCompile(`\*\*([^*]+)\*\*`).ReplaceAllString(t, "$1")
+	t = regexp.MustCompile(`\*([^*]+)\*`).ReplaceAllString(t, "$1")
+	t = regexp.MustCompile(`__([^_]+)__`).ReplaceAllString(t, "$1")
+	t = regexp.MustCompile(`_([^_]+)_`).ReplaceAllString(t, "$1")
+	t = regexp.MustCompile(`\[([^\]]+)\]\([^)]+\)`).ReplaceAllString(t, "$1")
+	t = regexp.MustCompile("```[^`]*```").ReplaceAllString(t, "")
+	t = regexp.MustCompile("`([^`]+)`").ReplaceAllString(t, "$1")
+	t = regexp.MustCompile(`(?m)^>\s+`).ReplaceAllString(t, "")
+	t = regexp.MustCompile(`(?m)^[\*\-\+]\s+`).ReplaceAllString(t, "")
+	t = regexp.MustCompile(`(?m)^\d+\.\s+`).ReplaceAllString(t, "")
+	return strings.TrimSpace(t)
+}
+
+func htmlToText(html string) string {
+	t := html
+	t = regexp.MustCompile(`(?i)<script[^>]*>.*?</script>`).ReplaceAllString(t, "")
+	t = regexp.MustCompile(`(?i)<style[^>]*>.*?</style>`).ReplaceAllString(t, "")
+	t = regexp.MustCompile(`(?i)<br\s*/?>`).ReplaceAllString(t, "\n")
+	t = regexp.MustCompile(`(?i)</p>`).ReplaceAllString(t, "\n")
+	t = regexp.MustCompile(`<[^>]+>`).ReplaceAllString(t, "")
+	t = strings.ReplaceAll(t, "&nbsp;", " ")
+	t = strings.ReplaceAll(t, "&lt;", "<")
+	t = strings.ReplaceAll(t, "&gt;", ">")
+	t = strings.ReplaceAll(t, "&amp;", "&")
+	t = strings.ReplaceAll(t, "&quot;", "\"")
+	t = regexp.MustCompile(`\n{3,}`).ReplaceAllString(t, "\n\n")
+	return strings.TrimSpace(t)
+}