Merge pull request '🆕 新增水群年度排行榜' (#6) from dev into main

Reviewed-on: https://gitee.ltd/lxh/go-wxhelper/pulls/6

config.yaml

@@ -21,14 +21,15 @@ mysql:
 task:
   enable: false
   syncFriends:
-      enable: true
+      enable: false
       cron: '*/5 * * * *' # 五分钟一次
   waterGroup:
-      enable: false
+      enable: true
       cron:
         yesterday: '30 9 * * *' # 每天9:30
         week: '30 9 * * 1' # 每周一9:30
         month: '30 9 1 * *' # 每月1号9:30
+        year: '55 0 1 1 *' # 每年1月1号9:30
 
 # AI回复
 ai:

config/task.go

@@ -28,4 +28,5 @@ type waterGroupCron struct {
 	Yesterday string `json:"yesterday" yaml:"yesterday"` // 昨日排行榜
 	Week      string `json:"week" yaml:"week"`           // 周排行榜
 	Month     string `json:"month" yaml:"month"`         // 月排行榜
+	Year      string `json:"year" yaml:"year"`           // 年排行榜
 }

initialization/plugin.go

@@ -5,6 +5,7 @@ import (
 	"go-wechat/model"
 	plugin "go-wechat/plugin"
 	"go-wechat/plugin/plugins"
+	"go-wechat/service"
 )
 
 // Plugin
@@ -24,8 +25,8 @@ func Plugin() {
 
 	// 私聊指令消息
 	dispatcher.RegisterHandler(func(m *model.Message) bool {
-		// 私聊消息直接进去
-		return m.IsPrivateText()
+		// 私聊消息 或 群聊艾特机器人并且以/开头的消息
+		return (m.IsPrivateText() || (m.IsAt() && m.CleanContentStartWith("/"))) && service.CheckIsEnableCommand(m.FromUser)
 	}, plugins.Command)
 
 	// AI消息插件

model/message.go

@@ -3,6 +3,7 @@ package model
 import (
 	"encoding/xml"
 	"go-wechat/types"
+	"regexp"
 	"strings"
 )
 
@@ -99,3 +100,28 @@ func (m Message) IsPrivateText() bool {
 	// 发信人不以@chatroom结尾且消息类型为文本
 	return !strings.HasSuffix(m.FromUser, "chatroom") && m.Type == types.MsgTypeText
 }
+
+// CleanContentStartWith
+// @description: 判断是否包含指定消息前缀
+// @receiver m
+// @param prefix
+// @return bool
+func (m Message) CleanContentStartWith(prefix string) bool {
+	content := m.Content
+
+	// 如果是@消息，过滤掉@的内容
+	if m.IsAt() {
+		re := regexp.MustCompile(`@([^ | ]+)`)
+		matches := re.FindStringSubmatch(content)
+		if len(matches) > 0 {
+			// 过滤掉第一个匹配到的
+			content = strings.Replace(content, matches[0], "", 1)
+		}
+	}
+
+	// 去掉最前面的空格
+	content = strings.TrimLeft(content, " ")
+	content = strings.TrimLeft(content, " ")
+
+	return strings.HasPrefix(content, prefix)
+}

plugin/plugins/command.go

@@ -2,7 +2,9 @@ package plugins
 
 import (
 	"go-wechat/plugin"
+	"go-wechat/plugin/plugins/command"
 	"go-wechat/utils"
+	"regexp"
 	"strings"
 )
 
@@ -10,20 +12,36 @@ import (
 // @description: 自定义指令
 // @param m
 func Command(m *plugin.MessageContext) {
+	// 如果是群聊，提取出消息
+	content := m.Content
+
+	if m.IsGroup() {
+		re := regexp.MustCompile(`@([^ | ]+)`)
+		matches := re.FindStringSubmatch(content)
+		if len(matches) > 0 {
+			// 过滤掉第一个匹配到的
+			content = strings.Replace(content, matches[0], "", 1)
+		}
+		// 去掉最前面的空格
+		content = strings.TrimLeft(content, " ")
+		content = strings.TrimLeft(content, " ")
+	}
 	// 判断是不是指令
-	if !strings.HasPrefix(m.Content, "/") {
+	if !strings.HasPrefix(content, "/") {
 		return
 	}
 
 	// 用空格分割消息，下标0表示指令
-	msgArray := strings.Split(m.Content, " ")
+	msgArray := strings.Split(content, " ")
 	cmd := msgArray[0]
 
 	switch cmd {
 	case "/帮助", "/h", "/help", "/?", "/？":
-		helpCmd(m)
-	case "/ls", "/雷神":
-		leiGodCmd(m.FromUser, msgArray[1], msgArray[2:]...)
+		command.HelpCmd(m)
+	case "/雷神", "/ls":
+		command.LeiGodCmd(m.FromUser, msgArray[1], msgArray[2:]...)
+	case "/肯德基", "/kfc":
+		command.KfcCrazyThursdayCmd(m.FromUser)
 	default:
 		utils.SendMessage(m.FromUser, m.GroupUser, "指令错误", 0)
 	}
@@ -31,54 +49,3 @@ func Command(m *plugin.MessageContext) {
 	// 中止后续消息处理
 	m.Abort()
 }
-
-// helpCmd
-// @description: 帮助指令
-func helpCmd(m *plugin.MessageContext) {
-	str := `帮助菜单:
-指令消息必须以'/'开头，比如: '/帮助'。
-支持的指令:
-
-#1. 雷神加速器
-/ls option args
-option: 指令选项，可选值: 
-  绑定账户：'绑定'、'b'，参数: 账户名 密码 [-f]，-f表示强制绑定，非必传项
-  详情: '详情'、'i'
-  暂停: '暂停'、'p'
-示例: 绑定:
-/ls 绑定 123456 123456 或者 /ls b 123456 123456
-`
-	utils.SendMessage(m.FromUser, m.GroupUser, str, 0)
-
-}
-
-// leiGodCmd
-// @description: 雷神加速器指令
-// @param userId
-// @param cmd
-// @param args
-// @return string
-func leiGodCmd(userId, cmd string, args ...string) {
-	lg := newLeiGod(userId)
-
-	var replyMsg string
-	switch cmd {
-	case "绑定", "b":
-		var force bool
-		if len(args) == 3 && args[2] == "-f" {
-			force = true
-		}
-		replyMsg = lg.binding(args[0], args[1], force)
-	case "详情", "i":
-		replyMsg = lg.info()
-	case "暂停", "p":
-		replyMsg = lg.pause()
-	default:
-		replyMsg = "指令错误"
-	}
-
-	// 返回消息
-	if strings.TrimSpace(replyMsg) != "" {
-		utils.SendMessage(userId, "", replyMsg, 0)
-	}
-}

plugin/plugins/command/help.go

@@ -0,0 +1,29 @@
+package command
+
+import (
+	"go-wechat/plugin"
+	"go-wechat/utils"
+)
+
+// HelpCmd
+// @description: 帮助指令
+func HelpCmd(m *plugin.MessageContext) {
+	str := `帮助菜单:
+指令消息必须以'/'开头，比如: '/帮助'。
+支持的指令:
+
+#1. 雷神加速器
+/ls option args
+option: 指令选项，可选值: 
+  绑定账户：'绑定'、'b'，参数: 账户名 密码 [-f]，-f表示强制绑定，非必传项
+  详情: '详情'、'i'
+  暂停: '暂停'、'p'
+示例: 绑定:
+/ls 绑定 123456 123456 或者 /ls b 123456 123456
+
+#2. 肯德基疯狂星期四文案
+/kfc、/肯德基
+`
+	utils.SendMessage(m.FromUser, m.GroupUser, str, 0)
+
+}

plugin/plugins/command/kfc.go

@@ -0,0 +1,95 @@
+package command
+
+import (
+	"github.com/go-resty/resty/v2"
+	"go-wechat/utils"
+	"log"
+)
+
+// KfcCrazyThursdayCmd
+// @description: 肯德基疯狂星期四文案
+// @param userId string 发信人
+func KfcCrazyThursdayCmd(userId string) {
+	// 随机选一个接口调用
+	str := kfcApi1()
+	if str == "" {
+		str = kfcApi2()
+	}
+	if str == "" {
+		str = kfcApi3()
+	}
+	if str == "" {
+		str = "文案获取失败"
+	}
+
+	// 发送消息
+	utils.SendMessage(userId, "", str, 0)
+}
+
+// kfcApi1
+// @description: 肯德基疯狂星期四文案接口1
+// @return string
+func kfcApi1() string {
+	res := resty.New()
+	resp, err := res.R().
+		Post("https://api.jixs.cc/api/wenan-fkxqs/index.php")
+	if err != nil {
+		log.Panicf("KFC接口1文案获取失败: %s", err.Error())
+	}
+	log.Printf("KFC接口1文案获取结果: %s", resp.String())
+	return resp.String()
+}
+
+// kfcApi2
+// @description: 肯德基疯狂星期四文案接口2
+// @return string
+func kfcApi2() string {
+	type result struct {
+		Code int    `json:"code"`
+		Text string `json:"text"`
+		Data struct {
+			Msg string `json:"msg"`
+		} `json:"data"`
+	}
+
+	var resData result
+
+	res := resty.New()
+	resp, err := res.R().
+		SetResult(&resData).
+		Post("https://api.jixs.cc/api/wenan-fkxqs/index.php")
+	if err != nil {
+		log.Panicf("KFC接口2文案获取失败: %s", err.Error())
+	}
+	log.Printf("KFC接口2文案获取结果: %s", resp.String())
+	if resData.Data.Msg != "" {
+		return resData.Data.Msg
+	}
+	return resp.String()
+}
+
+// kfcApi3
+// @description: 肯德基疯狂星期四文案接口3
+// @return string
+func kfcApi3() string {
+	type result struct {
+		Code int    `json:"code"`
+		Msg  string `json:"msg"`
+		Text string `json:"text"`
+	}
+
+	var resData result
+
+	res := resty.New()
+	resp, err := res.R().
+		SetResult(&resData).
+		Post("https://api.pearktrue.cn/api/kfc")
+	if err != nil {
+		log.Panicf("KFC接口3文案获取失败: %s", err.Error())
+	}
+	log.Printf("KFC接口3文案获取结果: %s", resp.String())
+	if resData.Text != "" {
+		return resData.Text
+	}
+	return resp.String()
+}

plugin/plugins/command/leigod.go

@@ -1,4 +1,4 @@
-package plugins
+package command
 
 import (
 	"encoding/json"
@@ -11,6 +11,7 @@ import (
 	"go-wechat/vo"
 	"gorm.io/gorm"
 	"log"
+	"strings"
 )
 
 // leiGod
@@ -33,6 +34,37 @@ func newLeiGod(userId string) leiGodI {
 	return &leiGod{userId: userId}
 }
 
+// LeiGodCmd
+// @description: 雷神加速器指令
+// @param userId
+// @param cmd
+// @param args
+// @return string
+func LeiGodCmd(userId, cmd string, args ...string) {
+	lg := newLeiGod(userId)
+
+	var replyMsg string
+	switch cmd {
+	case "绑定", "b":
+		var force bool
+		if len(args) == 3 && args[2] == "-f" {
+			force = true
+		}
+		replyMsg = lg.binding(args[0], args[1], force)
+	case "详情", "i":
+		replyMsg = lg.info()
+	case "暂停", "p":
+		replyMsg = lg.pause()
+	default:
+		replyMsg = "指令错误"
+	}
+
+	// 返回消息
+	if strings.TrimSpace(replyMsg) != "" {
+		utils.SendMessage(userId, "", replyMsg, 0)
+	}
+}
+
 // binding
 // @description: 绑定雷神加速器账号
 // @receiver l

service/friend.go

@@ -50,3 +50,13 @@ func GetAllEnableChatRank() (records []entity.Friend, err error) {
 	err = client.MySQL.Where("enable_chat_rank = ?", 1).Where("wxid LIKE '%@chatroom'").Find(&records).Error
 	return
 }
+
+// CheckIsEnableCommand
+// @description: 检查用户是否启用了指令
+// @param userId
+// @return flag
+func CheckIsEnableCommand(userId string) (flag bool) {
+	var coo int64
+	client.MySQL.Model(&entity.Friend{}).Where("enable_command = 1").Where("wxid = ?", userId).Count(&coo)
+	return coo > 0
+}

tasks/tasks.go

@@ -32,6 +32,9 @@ func InitTasks() {
 		if config.Conf.Task.WaterGroup.Cron.Month != "" {
 			_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Month).Do(watergroup.Month)
 		}
+		if config.Conf.Task.WaterGroup.Cron.Year != "" {
+			_, _ = s.Cron(config.Conf.Task.WaterGroup.Cron.Year).Do(watergroup.Year)
+		}
 	}
 
 	// 更新好友列表

tasks/watergroup/utils.go

@@ -33,6 +33,8 @@ func getRankData(groupId, date string) (rank []rankUser, err error) {
 		tx.Where("YEARWEEK(date_format(tm.create_at, '%Y-%m-%d')) = YEARWEEK(now()) - 1")
 	case "month":
 		tx.Where("PERIOD_DIFF(date_format(now(), '%Y%m'), date_format(create_at, '%Y%m')) = 1")
+	case "year":
+		tx.Where("YEAR(tm.create_at) = YEAR(NOW()) - 1")
 	}
 
 	// 查询指定时间段全部数据

tasks/watergroup/year.go

@@ -0,0 +1,97 @@
+package watergroup
+
+import (
+	"fmt"
+	"go-wechat/config"
+	"go-wechat/service"
+	"go-wechat/utils"
+	"log"
+	"strings"
+	"time"
+)
+
+// Year
+// @description: 年排行榜
+func Year() {
+	groups, err := service.GetAllEnableChatRank()
+	if err != nil {
+		log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
+		return
+	}
+
+	for _, group := range groups {
+		// 消息统计
+		dealYear(group.Wxid)
+
+		res, ok := config.Conf.Resource["wordcloud"]
+		if !ok {
+			continue
+		}
+
+		// 获取上周周数
+		year := time.Now().Local().AddDate(0, 0, -1).Year()
+		// 发送词云
+		fileName := fmt.Sprintf("%d_%s.png", year, group.Wxid)
+		utils.SendImage(group.Wxid, fmt.Sprintf(res.Path, fileName), 0)
+	}
+}
+
+// dealYear
+// @description: 处理年度排行榜
+// @param gid
+func dealYear(gid string) {
+	notifyMsgs := []string{"#年度水群排行榜"}
+
+	// 获取上周消息总数
+	records, err := getRankData(gid, "year")
+	if err != nil {
+		log.Printf("获取去年消息排行失败, 错误信息: %v", err)
+		return
+	}
+	log.Printf("去年消息总数: %+v", records)
+	// 莫得消息，直接返回
+	if len(records) == 0 {
+		log.Printf("去年本群[%s]无对话记录", gid)
+		return
+	}
+	// 计算消息总数
+	var msgCount int64
+	for _, v := range records {
+		msgCount += v.Count
+	}
+	// 组装消息总数推送信息
+	notifyMsgs = append(notifyMsgs, " ")
+	notifyMsgs = append(notifyMsgs, "亲爱的群友们，新年已经悄悄来临，让我们一起迎接这充满希望和美好的时刻。在这个特殊的日子里，我要向你们致以最真挚的祝福。")
+	notifyMsgs = append(notifyMsgs, "首先，我想对去年在群中表现出色、积极参与的成员们表示衷心的祝贺和感谢！你们的活跃与奉献让群聊更加充满了生机和活力。你们的贡献不仅仅是为了自己，更是为了我们整个群体的进步与成长。")
+	notifyMsgs = append(notifyMsgs, "特此给去年年度活跃成员排行榜上的朋友们送上真诚的祝福。你们的热情、智慧和参与度，令我们很是钦佩。愿新的一年中，你们继续保持着你们的活力和激情，为群中带来更多的惊喜和启迪。")
+	notifyMsgs = append(notifyMsgs, "对于那些未上榜的朋友们，我要说，你们也是我们群聊中非常重要的一部分。你们或许没有在排行榜上留下痕迹，但你们的存在和参与同样不可或缺。你们为群聊注入了新的思维和观点，为我们提供了不同的视角和见解。")
+	notifyMsgs = append(notifyMsgs, "因此，我想特别鼓励未上榜的朋友们，继续发扬你们的热情和积极性。无论是在分享知识、讨论问题、还是互相支持鼓励，你们的贡献都是宝贵的。让我们共同创造一个更加活跃和有意义的群聊环境。")
+	notifyMsgs = append(notifyMsgs, "最后，让我们一起迈向新的一年，相信自己的潜力和可能性，用我们的友谊和互助支持彼此。愿新的一年给我们带来更多的快乐、成功和成长。")
+	notifyMsgs = append(notifyMsgs, "祝福你们新年快乐！让我们一起迎接2022年的到来！")
+	notifyMsgs = append(notifyMsgs, "     -- OpnWechat")
+	notifyMsgs = append(notifyMsgs, " ")
+	notifyMsgs = append(notifyMsgs, fmt.Sprintf("🗣️ 去年本群 %d 位朋友共产生 %d 条发言", len(records), msgCount))
+	notifyMsgs = append(notifyMsgs, "\n🏵 活跃用户排行榜 🏵")
+
+	notifyMsgs = append(notifyMsgs, " ")
+	for i, r := range records {
+		// 只取前十条
+		if i >= 10 {
+			break
+		}
+		log.Printf("账号: %s[%s] -> %d", r.Nickname, r.GroupUser, r.Count)
+		badge := "🏆"
+		switch i {
+		case 0:
+			badge = "🥇"
+		case 1:
+			badge = "🥈"
+		case 2:
+			badge = "🥉"
+		}
+		notifyMsgs = append(notifyMsgs, fmt.Sprintf("%s %s -> %d条", badge, r.Nickname, r.Count))
+	}
+
+	log.Printf("排行榜: \n%s", strings.Join(notifyMsgs, "\n"))
+	go utils.SendMessage(gid, "", strings.Join(notifyMsgs, "\n"), 0)
+}