Merge pull request 'dev' (#12) from dev into main

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

app/friend.go

@@ -85,6 +85,29 @@ func ChangeEnableWelcomeStatus(ctx *gin.Context) {
 	ctx.String(http.StatusOK, "操作成功")
 }
 
+// ChangeEnableCommandStatus
+// @description: 修改是否开启指令
+// @param ctx
+func ChangeEnableCommandStatus(ctx *gin.Context) {
+	var p changeStatusParam
+	if err := ctx.ShouldBindJSON(&p); err != nil {
+		ctx.String(http.StatusBadRequest, "参数错误")
+		return
+	}
+	log.Printf("待修改的群Id：%s", p.WxId)
+
+	err := client.MySQL.Model(&entity.Friend{}).
+		Where("wxid = ?", p.WxId).
+		Update("`enable_command`", gorm.Expr(" !`enable_command`")).Error
+	if err != nil {
+		log.Printf("修改指令启用状态失败：%s", err)
+		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
+		return
+	}
+
+	ctx.String(http.StatusOK, "操作成功")
+}
+
 // ChangeSkipGroupRankStatus
 // @description: 修改是否跳过水群排行榜
 // @param ctx

app/index.go

@@ -3,6 +3,7 @@ package app
 import (
 	"fmt"
 	"github.com/gin-gonic/gin"
+	"go-wechat/config"
 	"go-wechat/service"
 	"net/http"
 )
@@ -21,6 +22,7 @@ func Index(ctx *gin.Context) {
 	}
 	result["friends"] = friends
 	result["groups"] = groups
+	result["vnc"] = config.Conf.Wechat.VncUrl
 	// 渲染页面
 	ctx.HTML(http.StatusOK, "index.html", result)
 }

config.yaml

@@ -2,6 +2,8 @@
 wechat:
   # 微信HOOK接口地址
   host: 10.0.0.73:19088
+  # 微信容器映射出来的vnc页面地址，没有就不填
+  vncUrl: http://192.168.1.175:19087/vnc_lite.html
   # 是否在启动的时候自动设置hook服务的回调
   autoSetCallback: false
   # 回调IP，如果是Docker运行，本参数必填(填auto表示自动，不适用于 docker 环境)，如果Docker修改了映射，格式为 ip:port
@@ -21,14 +23,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: '0 9 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"`           // 年排行榜
 }

config/wechat.go

@@ -6,6 +6,7 @@ import "strings"
 // @description: 微信助手
 type wechat struct {
 	Host            string   `json:"host" yaml:"host"`                       // 接口地址
+	VncUrl          string   `json:"vncUrl" yaml:"vncUrl"`                   // vnc页面地址
 	AutoSetCallback bool     `json:"autoSetCallback" yaml:"autoSetCallback"` // 是否自动设置回调地址
 	Callback        string   `json:"callback" yaml:"callback"`               // 回调地址
 	Forward         []string `json:"forward" yaml:"forward"`                 // 转发地址

docker-compose.yaml

@@ -2,7 +2,7 @@ version: '3.9'
 
 services:
   wechat:
-    image: lxh01/wxhelper-docker:3.9.5.81
+    image: lxh01/wxhelper-docker:3.9.5.81-v11
     container_name: gw-wechat
     restart: unless-stopped
     environment:
@@ -10,7 +10,7 @@ services:
     volumes:
       - ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
     ports:
-      - "8080:8080"
+      - "19087:8080"
       - "19088:19088"
     healthcheck:
       test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]

entity/friend.go

@@ -31,6 +31,7 @@ type GroupUser struct {
 	HeadImage    string     `json:"headImage"`                                              // 头像
 	Nickname     string     `json:"nickname"`                                               // 昵称
 	IsMember     bool       `json:"isMember" gorm:"type:tinyint(1) default 0 not null"`     // 是否群成员
+	IsAdmin      bool       `json:"isAdmin" gorm:"type:tinyint(1) default 0 not null"`      // 是否群主
 	JoinTime     time.Time  `json:"joinTime"`                                               // 加入时间
 	LeaveTime    *time.Time `json:"leaveTime"`                                              // 离开时间
 	SkipChatRank bool       `json:"skipChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否跳过聊天排行

initialization/plugin.go

@@ -26,13 +26,14 @@ func Plugin() {
 	// 私聊指令消息
 	dispatcher.RegisterHandler(func(m *model.Message) bool {
 		// 私聊消息 或 群聊艾特机器人并且以/开头的消息
-		return (m.IsPrivateText() || (m.IsAt() && m.CleanContentStartWith("/"))) && service.CheckIsEnableCommand(m.FromUser)
+		isGroupAt := m.IsAt() && !m.IsAtAll()
+		return (m.IsPrivateText() || isGroupAt) && m.CleanContentStartWith("/") && service.CheckIsEnableCommand(m.FromUser)
 	}, plugins.Command)
 
 	// AI消息插件
 	dispatcher.RegisterHandler(func(m *model.Message) bool {
 		// 群内@或者私聊文字消息
-		return m.IsAt() || m.IsPrivateText()
+		return (m.IsAt() && !m.IsAtAll()) || m.IsPrivateText()
 	}, plugins.AI)
 
 	// 欢迎新成员

model/message.go

@@ -2,6 +2,7 @@ package model
 
 import (
 	"encoding/xml"
+	"github.com/duke-git/lancet/v2/slice"
 	"go-wechat/types"
 	"regexp"
 	"strings"
@@ -31,6 +32,21 @@ type systemMsgDataXml struct {
 	Type   string `xml:"type,attr"`
 }
 
+// atMsgDataXml
+// @description: 微信@消息的xml结构
+type atMsgDataXml struct {
+	XMLName     xml.Name `xml:"msgsource"`
+	Text        string   `xml:",chardata"`
+	AtUserList  string   `xml:"atuserlist"`
+	Silence     string   `xml:"silence"`
+	MemberCount string   `xml:"membercount"`
+	Signature   string   `xml:"signature"`
+	TmpNode     struct {
+		Text        string `xml:",chardata"`
+		PublisherID string `xml:"publisher-id"`
+	} `xml:"tmp_node"`
+}
+
 // sysMsg
 // @description: 消息主体
 type sysMsg struct{}
@@ -92,6 +108,22 @@ func (m Message) IsAt() bool {
 	return strings.HasSuffix(m.DisplayFullContent, "在群聊中@了你")
 }
 
+// IsAtAll
+// @description: 是否是At所有人的消息
+// @receiver m
+// @return bool
+func (m Message) IsAtAll() bool {
+	// 解析raw里面的xml
+	var d atMsgDataXml
+	if err := xml.Unmarshal([]byte(m.Raw), &d); err != nil {
+		return false
+	}
+	// 转换@用户列表为数组
+	atUserList := strings.Split(d.AtUserList, ",")
+	// 判断是否包含@所有人
+	return slice.Contain(atUserList, "notify@all")
+}
+
 // IsPrivateText
 // @description: 是否是私聊消息
 // @receiver m

readme.md

@@ -14,6 +14,8 @@ vim config.yaml # 编辑配置文件，内容如下，最新配置请参考项
 wechat:
   # 微信HOOK接口地址
   host: wechat:19088
+  # 微信容器映射出来的vnc页面地址，没有就不填
+  vncUrl: http://192.168.1.175:19087/vnc_lite.html
   # 是否在启动的时候自动设置hook服务的回调
   autoSetCallback: true
   # 回调IP，如果是Docker运行，本参数必填，如果Docker修改了映射，格式为 ip:port，如果使用项目提供的docker-compsoe.yaml文件启动，可以填`auto`
@@ -46,7 +48,7 @@ version: '3.9'
 
 services:
   wechat:
-    image: lxh01/wxhelper-docker:3.9.5.81
+    image: lxh01/wxhelper-docker:3.9.5.81-v11
     container_name: gw-wechat
     restart: unless-stopped
     environment:
@@ -54,7 +56,7 @@ services:
     volumes:
       - ./data/wechat:/home/app/.wine/drive_c/users/app/Documents/WeChat\ Files
     ports:
-      - "8080:8080"
+      - "19087:8080"
       - "19088:19088"
     healthcheck:
       test: ["CMD", "curl", "-f", "http://localhost:19088/api/checkLogin"]

router/router.go

@@ -23,6 +23,7 @@ func Init(g *gin.Engine) {
 	api := g.Group("/api")
 	api.PUT("/ai/status", app.ChangeEnableAiStatus)               // 修改是否开启AI状态
 	api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus)     // 修改是否开启迎新状态
+	api.PUT("/command/status", app.ChangeEnableCommandStatus)     // 修改是否开启指令状态
 	api.PUT("/grouprank/status", app.ChangeEnableGroupRankStatus) // 修改是否开启水群排行榜状态
 	api.PUT("/grouprank/skip", app.ChangeSkipGroupRankStatus)     // 修改是否跳过水群排行榜状态
 	api.GET("/group/users", app.GetGroupUsers)                    // 获取群成员列表

tasks/friends/friends.go

@@ -155,6 +155,7 @@ func syncGroupUsers(tx *gorm.DB, gid string) {
 					Nickname:  cp.Nickname,
 					Wxid:      cp.Wxid,
 					IsMember:  true,
+					IsAdmin:   wxid == baseResp.Data.Admin,
 					JoinTime:  time.Now().Local(),
 				}).Error
 				if err != nil {
@@ -168,6 +169,7 @@ func syncGroupUsers(tx *gorm.DB, gid string) {
 					"head_image": cp.HeadImage,
 					"nickname":   cp.Nickname,
 					"is_member":  true,
+					"is_admin":   wxid == baseResp.Data.Admin,
 					"leave_time": nil,
 				}
 				err = tx.Model(&entity.GroupUser{}).Where("group_id = ?", gid).Where("wxid = ?", wxid).Updates(pm).Error

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,96 @@
+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, fmt.Sprintf("祝福你们新年快乐！让我们一起迎接%d年的到来！", time.Now().Local().Year()))
+	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)
+}

views/index.html

@@ -37,6 +37,7 @@
                         <th>最后活跃时间</th>
                         <th>是否在通讯录</th>
                         <th>是否启用AI</th>
+                        <th>是否启用指令</th>
                     </tr>
                     </thead>
                     <tbody>
@@ -67,6 +68,14 @@
                             <label class="swap swap-flip {{ checkSwap .EnableAi }}">
                                 <input type="checkbox" onclick="changeAiEnableStatus({{.Wxid}})"/>
 
+                                <div class="swap-on">✔️已启用</div>
+                                <div class="swap-off">❌已禁用</div>
+                            </label>
+                        </td>
+                        <td>
+                            <label class="swap swap-flip {{ checkSwap .EnableCommand }}">
+                                <input type="checkbox" onclick="changeCommandEnableStatus({{.Wxid}})"/>
+
                                 <div class="swap-on">✔️已启用</div>
                                 <div class="swap-off">❌已禁用</div>
                             </label>
@@ -91,6 +100,7 @@
                         <th>是否启用AI</th>
                         <th>是否启用水群排行榜</th>
                         <th>是否启用迎新</th>
+                        <th>是否启用指令</th>
                         <th>操作</th>
                     </tr>
                     </thead>
@@ -143,6 +153,14 @@
                                 <div class="swap-off">❌已禁用</div>
                             </label>
                         </td>
+                        <td>
+                            <label class="swap swap-flip {{ checkSwap .EnableCommand }}">
+                                <input type="checkbox" onclick="changeCommandEnableStatus({{.Wxid}})"/>
+
+                                <div class="swap-on">✔️已启用</div>
+                                <div class="swap-off">❌已禁用</div>
+                            </label>
+                        </td>
                         <td>
                             <button class="btn btn-link" onclick="getGroupUsers({{.Wxid}}, {{.Nickname}})">查看群成员</button>
                         </td>
@@ -151,6 +169,15 @@
                     </tbody>
                 </table>
             </div>
+
+            {{ if ne .vnc "" }}
+            <input type="radio" name="friend_tab" role="tab" class="tab" aria-label="运行状态"/>
+            <div role="tabpanel" class="tab-content p-6">
+                <div style="height: 747px;width: 1280px;overflow: hidden !important;">
+                    <iframe src="{{ .vnc }}" frameborder="0" style="width: 100%;height: 100%;pointer-events: none;"></iframe>
+                </div>
+            </div>
+            {{ end }}
         </div>
 
 
@@ -174,6 +201,7 @@
                     <th>微信Id</th>
                     <th>昵称</th>
                     <th>是否群成员</th>
+                    <th>是否群主</th>
                     <th>加群时间</th>
                     <th>最后活跃时间</th>
                     <th>退群时间</th>

views/static/js/index.js

@@ -51,6 +51,22 @@ function changeWelcomeEnableStatus(wxId) {
     })
 }
 
+// 修改指令权限启用状态
+function changeCommandEnableStatus(wxId) {
+    axios({
+        method: 'put',
+        url: '/api/command/status',
+        data: {
+            wxId: wxId
+        }
+    }).then(function (response) {
+        console.log(`返回结果: ${JSON.stringify(response)}`);
+    }).catch(function (error) {
+        console.log(`错误信息: ${error}`);
+        alert("修改失败")
+    })
+}
+
 // 修改群成员是否参与排行榜状态
 function changeUserGroupRankSkipStatus(groupId, userId) {
     console.log("修改水群排行榜开启状态: ", groupId, userId)
@@ -96,43 +112,20 @@ function getGroupUsers(groupId, groupName) {
         // 渲染群成员列表
         const groupUsers = response.data
         // 循环渲染数据
-        for (let i = 0; i < groupUsers.length; i++) {
-            const groupUser = groupUsers[i]
+        groupUsers.forEach((groupUser, i) => {
+            const { wxid, nickname, isMember, isAdmin, joinTime, lastActiveTime, leaveTime, skipChatRank } = groupUser;
 
-            let row = tbody.insertRow(i); // 插入新行
-
-            // 微信Id
-            let wxId = row.insertCell(0);
-            wxId.innerHTML = groupUser.wxid;
-
-            // 昵称
-            let nickname = row.insertCell(1);
-            nickname.innerHTML = groupUser.nickname;
-
-            // 是否群成员
-            let isMember = row.insertCell(2);
-            if (groupUser.isMember) {
-                isMember.innerHTML = '<div class="badge badge-info gap-2">是</div>';
-            } else {
-                isMember.innerHTML = '<div class="badge badge-error gap-2">否</div>';
-            }
-
-            // 加群时间
-            let joinTime = row.insertCell(3);
-            joinTime.innerHTML = groupUser.joinTime;
-
-            // 最后活跃时间
-            let lastActiveTime = row.insertCell(4);
-            lastActiveTime.innerHTML = groupUser.lastActiveTime;
-
-            // 退群时间
-            let leaveTime = row.insertCell(5);
-            leaveTime.innerHTML = groupUser.leaveTime;
-
-            // 是否跳过水群排行榜
-            let skipChatRank = row.insertCell(6);
-            skipChatRank.innerHTML = `<input type="checkbox" class="toggle toggle-error" ${groupUser.skipChatRank ? 'checked' : ''} onclick="changeUserGroupRankSkipStatus(\'${groupId}\', \'${groupUser.wxid}\')" />`;
-        }
+            let row = tbody.insertRow(i);
+            // Insert data into cells
+            row.insertCell(0).innerHTML = wxid;
+            row.insertCell(1).innerHTML = nickname;
+            row.insertCell(2).innerHTML = `<div class="badge badge-${isMember ? 'info' : 'error'} gap-2">${isMember ? '是' : '否'}</div>`;
+            row.insertCell(3).innerHTML = `<div class="badge badge-${isAdmin ? 'info' : 'error'} gap-2">${isAdmin ? '是' : '否'}</div>`;
+            row.insertCell(4).innerHTML = joinTime;
+            row.insertCell(5).innerHTML = lastActiveTime;
+            row.insertCell(6).innerHTML = leaveTime;
+            row.insertCell(7).innerHTML = `<input type="checkbox" class="toggle toggle-error" ${skipChatRank ? 'checked' : ''} onclick="changeUserGroupRankSkipStatus('${groupId}', '${wxid}')" />`;
+        });
     }).catch(function (error) {
         console.log(`错误信息: ${error}`);
     }).finally(function () {

vo/friend.go

@@ -15,6 +15,7 @@ type FriendItem struct {
 	EnableAi       bool           // 是否使用AI
 	EnableChatRank bool           // 是否使用聊天排行
 	EnableWelcome  bool           // 是否使用迎新
+	EnableCommand  bool           // 是否启用指令
 	IsOk           bool           // 是否还在通讯库(群聊是要还在群里也算)
 	LastActiveTime types.DateTime // 最后活跃时间
 }
@@ -27,6 +28,7 @@ type GroupUserItem struct {
 	HeadImage      string         `json:"headImage"`      // 头像
 	Nickname       string         `json:"nickname"`       // 昵称
 	IsMember       bool           `json:"isMember" `      // 是否群成员
+	IsAdmin        bool           `json:"isAdmin"`        // 是否群主
 	JoinTime       types.DateTime `json:"joinTime"`       // 加入时间
 	LastActiveTime types.DateTime `json:"lastActiveTime"` // 最后活跃时间
 	LeaveTime      types.DateTime `json:"leaveTime"`      // 离开时间