wip:移除

src/jfw/modules/weixin/src/wx/reply.go

@@ -1,33 +1,33 @@
 package wx
 
 import (
-    "app.yhyue.com/moapp/jybase/common"
-    "app.yhyue.com/moapp/jybase/redis"
-    "crypto/md5"
-    "encoding/hex"
-    "fmt"
-    "html"
-    "jy/src/jfw/modules/weixin/src/github.com/wizjin/weixin"
-    "jy/src/jfw/modules/weixin/src/tools"
-    "log"
-    "regexp"
-    "strings"
+	"app.yhyue.com/moapp/jybase/common"
+	"app.yhyue.com/moapp/jybase/redis"
+	"crypto/md5"
+	"encoding/hex"
+	"fmt"
+	"html"
+	"jy/src/jfw/modules/weixin/src/github.com/wizjin/weixin"
+	"jy/src/jfw/modules/weixin/src/tools"
+	"log"
+	"regexp"
+	"strings"
 )
 
 const (
-    TableWxReplyRule    = "jianyu.wx_reply_rule"    // 规则表
-    TableWxReplyKeys    = "jianyu.wx_reply_keys"    // 关键词表
-    TableWxReplyContent = "jianyu.wx_reply_content" // 回复内容表
-    ReplyContentTEXT    = 1                         // 回复类型： 文本
-    ReplyContentImage   = 2                         // 图片
-    ReplyContentArticle = 3                         // 图文
-    ReplayImagePath     = "./res/reply/upload"      // 回复图片上传路径
+	TableWxReplyRule    = "jianyu.wx_reply_rule"    // 规则表
+	TableWxReplyKeys    = "jianyu.wx_reply_keys"    // 关键词表
+	TableWxReplyContent = "jianyu.wx_reply_content" // 回复内容表
+	ReplyContentTEXT    = 1                         // 回复类型： 文本
+	ReplyContentImage   = 2                         // 图片
+	ReplyContentArticle = 3                         // 图文
+	ReplayImagePath     = "./res/reply/upload"      // 回复图片上传路径
 )
 
 // KeysReply 关键词回复
 func KeysReply(keys string, w weixin.ResponseWriter) bool {
-    // 获取回复内容
-    query := fmt.Sprintf(`SELECT
+	// 获取回复内容
+	query := fmt.Sprintf(`SELECT
     wrc.*
 FROM
     %s wrk
@@ -39,83 +39,86 @@ WHERE
     wrk.key = ? AND wrr.state = 1
         AND wrr.start_time <= NOW()
         AND wrr.end_time >= NOW();`, TableWxReplyKeys, TableWxReplyRule, TableWxReplyContent)
-    rs := tools.Mysql.SelectBySql(query, keys)
-    // 没匹配上规则
-    if rs == nil || len(*rs) == 0 {
-        return false
-    }
-    // 处理规则
-    for i := 0; i < len(*rs); i++ {
-        reply_type := common.IntAll((*rs)[i]["reply_type"])
-        title := common.ObjToString((*rs)[i]["title"])
-        url := common.ObjToString((*rs)[i]["url"])
-        pic_url := common.ObjToString((*rs)[i]["pic_url"])
-        desc := common.ObjToString((*rs)[i]["desc"])
-        switch reply_type {
-        case ReplyContentTEXT:
-            //编码及换行处理
-            title = html.UnescapeString(title)
-            title = strings.ReplaceAll(strings.ReplaceAll(title, "<br />", "\n"), "\\n", "\n")
-            // 使用正则表达式匹配HTML标签
-            re, err := regexp.Compile("\\<[^a]+?\\>")
-            if err != nil {
-                fmt.Println("Error compiling regex:", err)
-            } else {
-                log.Println("---1---:", title)
-                // 使用正则表达式的ReplaceAllString函数移除HTML标签
-                title = re.ReplaceAllString(title, "")
-                log.Println("---2---:", title)
-                err := w.PostText(title)
-                if err != nil {
-                    log.Println("文字 发送失败", err)
-                }
-            }
-        case ReplyContentImage:
-            // 先上传 获得id
-            var i int
-            for {
-                i++
-                // 计算MD5哈希值
-                hash := md5.Sum([]byte(url))
-                CBKey := fmt.Sprintf("reply_%s_%s", keys, hex.EncodeToString(hash[:]))
-                mediaId := redis.GetStr("other", CBKey)
-                if mediaId == "" {
-                    _mediaId, err := w.UploadMediaFromFile("image", pic_url)
-                    log.Println(_mediaId, "图片上传微信", err)
-                    if _mediaId != "" && err == nil {
-                        mediaId = _mediaId
-                    }
-                }
-                log.Println("发送图片", mediaId)
-                if mediaId != "" {
-                    go func() {
-                        redis.Put("other", CBKey, mediaId, 24*60*60*3)
-                    }()
-                    err := w.PostImage(mediaId)
-                    if err == nil {
-                        break
-                    }
-                    redis.Del("other", CBKey)
-                    log.Println("发送失败", err)
-                }
-                if i > 3 {
-                    break
-                }
-            }
-        case ReplyContentArticle:
-            article := weixin.Article{
-                Title:       title,
-                Url:         url,
-                PicUrl:      pic_url,
-                Description: desc,
-            }
-            err := w.PostNews([]weixin.Article{
-                article,
-            })
-            if err != nil {
-                log.Println("reply 图文消息 err:", err, article)
-            }
-        }
-    }
-    return true
+	rs := tools.Mysql.SelectBySql(query, keys)
+	// 没匹配上规则
+	if rs == nil || len(*rs) == 0 {
+		return false
+	}
+	// 处理规则
+	for i := 0; i < len(*rs); i++ {
+		reply_type := common.IntAll((*rs)[i]["reply_type"])
+		title := common.ObjToString((*rs)[i]["title"])
+		url := common.ObjToString((*rs)[i]["url"])
+		pic_url := common.ObjToString((*rs)[i]["pic_url"])
+		desc := common.ObjToString((*rs)[i]["desc"])
+		switch reply_type {
+		case ReplyContentTEXT:
+			//编码及换行处理
+			title = html.UnescapeString(title)
+			title = strings.ReplaceAll(strings.ReplaceAll(title, "<br />", "\n"), "\\n", "\n")
+			// 移除a标签中 title
+			reTitle, _ := regexp.Compile("title=\"[^\"]*\"")
+			title = reTitle.ReplaceAllString(title, "")
+			// 使用正则表达式匹配HTML标签
+			re, err := regexp.Compile("\\<[^a]+?\\>")
+			if err != nil {
+				fmt.Println("Error compiling regex:", err)
+			} else {
+				log.Println("---1---:", title)
+				// 使用正则表达式的ReplaceAllString函数移除HTML标签
+				title = re.ReplaceAllString(title, "")
+				log.Println("---2---:", title)
+				err := w.PostText(title)
+				if err != nil {
+					log.Println("文字 发送失败", err)
+				}
+			}
+		case ReplyContentImage:
+			// 先上传 获得id
+			var i int
+			for {
+				i++
+				// 计算MD5哈希值
+				hash := md5.Sum([]byte(url))
+				CBKey := fmt.Sprintf("reply_%s_%s", keys, hex.EncodeToString(hash[:]))
+				mediaId := redis.GetStr("other", CBKey)
+				if mediaId == "" {
+					_mediaId, err := w.UploadMediaFromFile("image", pic_url)
+					log.Println(_mediaId, "图片上传微信", err)
+					if _mediaId != "" && err == nil {
+						mediaId = _mediaId
+					}
+				}
+				log.Println("发送图片", mediaId)
+				if mediaId != "" {
+					go func() {
+						redis.Put("other", CBKey, mediaId, 24*60*60*3)
+					}()
+					err := w.PostImage(mediaId)
+					if err == nil {
+						break
+					}
+					redis.Del("other", CBKey)
+					log.Println("发送失败", err)
+				}
+				if i > 2 {
+					break
+				}
+			}
+		case ReplyContentArticle:
+			article := weixin.Article{
+				Title:       title,
+				Url:         url,
+				PicUrl:      pic_url,
+				Description: desc,
+			}
+			err := w.PostNews([]weixin.Article{
+				article,
+			})
+			if err != nil {
+				log.Println("reply 图文消息 err:", err, article)
+			}
+		}
+	}
+	return true
 }