Merge pull request 'hotfix' (#79) from hotfix into main

Reviewed-on: lxh/go-wxhelper#79

app/friend.go

@@ -235,6 +235,29 @@ func ChangeEnableNewsStatus(ctx *gin.Context) {
 	ctx.String(http.StatusOK, "操作成功")
 }
 
+// ChangeEnableHotTopStatus
+// @description: 修改是否开启热搜
+// @param ctx
+func ChangeEnableHotTopStatus(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_hot_top`", gorm.Expr(" !`enable_hot_top`")).Error
+	if err != nil {
+		log.Printf("修改热榜启用状态失败：%s", err)
+		ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
+		return
+	}
+
+	ctx.String(http.StatusOK, "操作成功")
+}
+
 // AutoClearMembers
 // @description: 自动清理群成员
 // @param ctx

config.yaml

@@ -19,11 +19,13 @@ system:
     welcome: true
     # 每日早报
     news: true
+    # 热榜
+    hotTop: true
 
 # 微信HOOK配置
 wechat:
   # 微信HOOK接口地址
-  host: 10.0.0.73:19088
+  host: 10.0.0.79:19088
   # 微信容器映射出来的vnc页面地址，没有就不填
   #  vncUrl: http://192.168.1.175:19087/vnc_lite.html
   # 是否在启动的时候自动设置hook服务的回调
@@ -44,11 +46,15 @@ mysql:
   db: wechat
   schema: public # postgres 专用
 
+# 定时任务
 task:
-  enable: false
-  news:
   enable: true
+  news:
+    enable: false
     cron: '14 11 * * *' # 每天0:30
+  hotTop:
+    enable: true
+    cron: '0 */1 * * *' # 每小时一次
   syncFriends:
     enable: false
     cron: '*/5 * * * *' # 五分钟一次

config/system.go

@@ -20,4 +20,5 @@ type defaultRule struct {
 	Summary  bool `json:"summary" yaml:"summary"`   // 是否启用聊天总结
 	Welcome  bool `json:"welcome" yaml:"welcome"`   // 是否启用欢迎新成员
 	News     bool `json:"news" yaml:"news"`         // 是否启用每日早报
+	HotTop   bool `json:"hotTop" yaml:"hotTop"`     // 是否启用热门话题
 }

config/task.go

@@ -8,6 +8,7 @@ type task struct {
 	SyncFriends  syncFriends `json:"syncFriends" yaml:"syncFriends"`   // 同步好友
 	WaterGroup   waterGroup  `json:"waterGroup" yaml:"waterGroup"`     // 水群排行榜
 	GroupSummary syncFriends `json:"groupSummary" yaml:"groupSummary"` // 群聊总结
+	HotTop       syncFriends `json:"hotTop" yaml:"hotTop"`             // 热搜排行榜
 }
 
 // syncFriends

model/dto/news.go

@@ -16,3 +16,23 @@ type MorningPost struct {
 	Usage int    `json:"usage"`
 	LogId string `json:"log_id"`
 }
+
+// HotTop
+// @description: 热搜排行榜返回结构体
+type HotTop struct {
+	Success    bool             `json:"success"`     // 是否成功
+	Name       string           `json:"name"`        // 渠道
+	Subtitle   string           `json:"subtitle"`    // 副标题
+	UpdateTime string           `json:"update_time"` // 更新时间
+	Data       []HotTopDataItem `json:"data"`        // 数据
+}
+
+// HotTopDataItem
+// @description: 热搜排行榜数据项
+type HotTopDataItem struct {
+	Index    int    `json:"index"`    // 排行
+	Title    string `json:"title"`    // 标题
+	Hot      string `json:"hot"`      // 热度
+	Url      string `json:"url"`      // 链接
+	MobilUrl string `json:"mobilUrl"` // 手机端链接
+}

model/entity/friend.go

@@ -20,6 +20,7 @@ type Friend struct {
 	EnableWelcome  bool      `json:"enableWelcome" gorm:"type:tinyint(1) default 0 not null"`  // 是否启用迎新
 	EnableSummary  bool      `json:"enableSummary" gorm:"type:tinyint(1) default 0 not null"`  // 是否启用总结
 	EnableNews     bool      `json:"enableNews" gorm:"type:tinyint(1) default 0 not null"`     // 是否启用新闻
+	EnableHotTop   bool      `json:"enableHotTop" gorm:"type:tinyint(1) default 0 not null"`   // 是否启用热榜新闻
 	ClearMember    int       `json:"clearMember"`                                              // 清理成员配置（多少天未活跃的）
 	IsOk           bool      `json:"isOk" gorm:"type:tinyint(1) default 0 not null"`           // 是否正常
 	UsedTokens     int       `json:"usedTokens"`                                               // 已使用的AI Token数量

model/entity/hottop.go

@@ -0,0 +1,40 @@
+package entity
+
+import (
+	"github.com/google/uuid"
+	"go-wechat/common/types"
+	"gorm.io/gorm"
+	"strings"
+)
+
+// HotTop
+// @description: 热榜数据
+type HotTop struct {
+	Id        string         `json:"id" gorm:"type:varchar(32);primarykey"`
+	CreatedAt types.DateTime `json:"createdAt"`
+	Channel   string         `json:"channel"`   // 渠道
+	Title     string         `json:"title"`     // 标题
+	Hot       string         `json:"hot"`       // 热度
+	Url       string         `json:"url"`       // 链接
+	MobileUrl string         `json:"mobileUrl"` // 手机端链接
+}
+
+// TableName
+// @description: 表名
+// @receiver HotTop
+// @return string
+func (HotTop) TableName() string {
+	return "t_hot_top"
+}
+
+// BeforeCreate
+// @description: 创建数据库对象之前生成UUID
+// @receiver m
+// @param *gorm.DB
+// @return err
+func (m *HotTop) BeforeCreate(*gorm.DB) (err error) {
+	if m.Id == "" {
+		m.Id = strings.ReplaceAll(uuid.New().String(), "-", "")
+	}
+	return
+}

model/vo/friend.go

@@ -21,6 +21,7 @@ type FriendItem struct {
 	EnableCommand  bool           // 是否启用指令
 	EnableSummary  bool           // 是否启用总结
 	EnableNews     bool           // 是否启用新闻
+	EnableHotTop   bool           // 是否启用热搜
 	ClearMember    int            // 清理成员配置（多少天未活跃的）
 	IsOk           bool           // 是否还在通讯库(群聊是要还在群里也算)
 }

router/router.go

@@ -29,6 +29,7 @@ func Init(g *gin.Engine) {
 	api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus)     // 修改是否开启迎新状态
 	api.PUT("/command/status", app.ChangeEnableCommandStatus)     // 修改是否开启指令状态
 	api.PUT("/news/status", app.ChangeEnableNewsStatus)           // 修改是否开启早报状态
+	api.PUT("/hot-top/status", app.ChangeEnableHotTopStatus)      // 修改是否开启热榜状态
 	api.PUT("/grouprank/status", app.ChangeEnableGroupRankStatus) // 修改是否开启水群排行榜状态
 	api.PUT("/grouprank/skip", app.ChangeSkipGroupRankStatus)     // 修改是否跳过水群排行榜状态
 	api.GET("/group/users", app.GetGroupUsers)                    // 获取群成员列表

service/friend.go

@@ -88,6 +88,15 @@ func GetAllEnableNews() (records []entity.Friend, err error) {
 	return
 }
 
+// GetAllEnableHotTop
+// @description: 获取所有启用了热搜排行榜的好友或群组
+// @return records
+// @return err
+func GetAllEnableHotTop() (records []entity.Friend, err error) {
+	err = client.MySQL.Where("enable_hot_top = ?", 1).Where("is_ok IS TRUE").Find(&records).Error
+	return
+}
+
 // GetAllEnableClearGroup
 // @description: 获取所有需要清理成员的群组
 // @return records

tasks/friends/friends.go

@@ -75,6 +75,7 @@ func Sync() {
 				EnableSummary:  config.Conf.System.DefaultRule.Summary,
 				EnableWelcome:  config.Conf.System.DefaultRule.Welcome,
 				EnableNews:     config.Conf.System.DefaultRule.News,
+				EnableHotTop:   config.Conf.System.DefaultRule.HotTop,
 				ClearMember:    0,
 				LastActive:     time.Now().Local(),
 			}).Error

tasks/hottop/hottop.go

@@ -0,0 +1,88 @@
+package hottop
+
+import (
+	"fmt"
+	"go-wechat/client"
+	"go-wechat/common/types"
+	"go-wechat/model/entity"
+	"go-wechat/service"
+	"go-wechat/utils"
+	"log"
+	"slices"
+	"strings"
+	"time"
+)
+
+// HotTop
+// @description: 热搜排行榜
+func HotTop() {
+	// 发送到开启了的群
+	groups, err := service.GetAllEnableHotTop()
+	if err != nil {
+		log.Printf("获取启用了热榜的群组失败, 错误信息: %v", err)
+		return
+	}
+	// 获取热榜数据
+	news := getTopData()
+	if len(news) == 0 {
+		return
+	}
+
+	// 组装消息
+	msg := fmt.Sprintf("#热搜排行榜\n \n嘿，朋友，有新的新闻了喔，快来康康吧\n \n%s", strings.Join(news, "\n-------\n"))
+	for _, group := range groups {
+		utils.SendMessage(group.Wxid, "", msg, 0)
+		// 休眠一秒，防止频繁发送
+		time.Sleep(time.Second)
+	}
+}
+
+// getTopData
+// @description: 获取热榜数据
+// @return data
+func getTopData() (data []string) {
+	// 获取热榜数据
+	records := utils.NewsUtil().GetHotTop()
+	if len(records) == 0 {
+		log.Println("获取热榜数据失败")
+		return
+	}
+	var datas = make([]entity.HotTop, 0)
+	for _, item := range records {
+		var d = entity.HotTop{
+			CreatedAt: types.DateTime(time.Now().Local()),
+			Title:     item.Title,
+			Hot:       item.Hot,
+			Url:       item.Url,
+			MobileUrl: item.MobilUrl,
+		}
+		datas = append(datas, d)
+	}
+
+	// 获取缓存数据
+	var oldTitles []string
+	err := client.MySQL.Model(&entity.HotTop{}).Order("created_at DESC").Limit(len(datas)).Pluck("title", &oldTitles).Error
+	if err != nil {
+		log.Println("获取历史热榜数据失败", err)
+		return
+	}
+
+	// 筛选出新数据
+	var newDatas []entity.HotTop
+	for _, d := range datas {
+		if slices.Contains(oldTitles, d.Title) {
+			continue
+		}
+		d.Channel = "百度"
+		newDatas = append(newDatas, d)
+		data = append(data, fmt.Sprintf("标题: %s\n热度: %s\n详情: %s", d.Title, d.Hot, d.Url))
+	}
+	// 保存新数据到数据库
+	if len(newDatas) > 0 {
+		err = client.MySQL.Create(&newDatas).Error
+		if err != nil {
+			log.Println("保存新热榜数据失败", err)
+		}
+	}
+	return
+}

tasks/tasks.go

@@ -5,6 +5,7 @@ import (
 	"go-wechat/config"
 	"go-wechat/tasks/cleargroupuser"
 	"go-wechat/tasks/friends"
+	"go-wechat/tasks/hottop"
 	"go-wechat/tasks/news"
 	"go-wechat/tasks/summary"
 	"go-wechat/tasks/watergroup"
@@ -57,6 +58,11 @@ func InitTasks() {
 		_, _ = s.Cron(config.Conf.Task.News.Cron).Do(news.DailyNews)
 	}
 
+	// 热榜
+	if config.Conf.Task.HotTop.Enable {
+		_, _ = s.Cron(config.Conf.Task.HotTop.Cron).Do(hottop.HotTop)
+	}
+
 	// 每天0点检查一次处理清理群成员
 	_, _ = s.Cron("0 0 * * *").Do(cleargroupuser.ClearGroupUser)
 

utils/news.go

@@ -11,6 +11,7 @@ import (
 // @description: 新闻
 type News interface {
 	MorningPost() []string           // 早报
+	GetHotTop() []dto.HotTopDataItem // 获取热搜排行榜
 }
 
 type news struct{}
@@ -38,10 +39,34 @@ func (news) MorningPost() (records []string) {
 		SetResult(&newsResp).
 		Post("https://v2.alapi.cn/api/zaobao")
 	if err != nil {
-		log.Panicf("每日早报获取失败: %s", err.Error())
+		log.Printf("每日早报获取失败: %s", err.Error())
+		return
 	}
 	log.Printf("每日早报获取结果: %s", unicodeToText(resp.String()))
 
 	records = newsResp.Data.News
 	return
 }
+
+// GetHotTop
+// @description: 获取热搜排行榜
+// @receiver news
+// @return records
+func (news) GetHotTop() (records []dto.HotTopDataItem) {
+	var respData dto.HotTop
+	res := resty.New()
+	resp, err := res.R().
+		SetHeader("Content-Type", "application/json;chartset=utf-8").
+		SetResult(&respData).
+		Get("https://api.vvhan.com/api/hotlist/baiduRD")
+	if err != nil {
+		log.Printf("百度热榜获取失败: %s", err.Error())
+		return
+	}
+	log.Printf("百度热榜获取结果: %s", unicodeToText(resp.String()))
+	if !respData.Success {
+		log.Println("百度热榜获取失败")
+		return
+	}
+	return respData.Data
+}

views/components.html

@@ -192,6 +192,33 @@
 </button>
 {{end}}
 
+<!-- 热榜 -->
+{{define "hotTop"}}
+<button type="button"
+        class="{{ if eq .EnableHotTop true }}bg-green-600{{ else }}bg-gray-200{{ end }} relative inline-flex h-6 w-11 flex-shrink-0 cursor-pointer rounded-full border-2 border-transparent transition-colors duration-200 ease-in-out focus:outline-none focus:ring-2 focus:ring-green-600 focus:ring-offset-2"
+        role="switch" aria-checked="false" onclick="changeUserHotTopStatus({{.Wxid}})">
+  <span
+    class="{{ if eq .EnableHotTop true }}translate-x-5{{ else }}translate-x-0{{ end }} pointer-events-none relative inline-block h-5 w-5 transform rounded-full bg-white shadow ring-0 transition duration-200 ease-in-out">
+    <span
+      class="{{ if eq .EnableHotTop true }}opacity-0 duration-100 ease-out{{ else }}opacity-100 duration-200 ease-in{{ end }} absolute inset-0 flex h-full w-full items-center justify-center transition-opacity"
+      aria-hidden="true">
+      <svg class="h-3 w-3 text-gray-400" fill="none" viewBox="0 0 12 12">
+        <path d="M4 8l2-2m0 0l2-2M6 6L4 4m2 2l2 2" stroke="currentColor" stroke-width="2" stroke-linecap="round"
+              stroke-linejoin="round"/>
+      </svg>
+    </span>
+    <span
+      class="{{ if eq .EnableHotTop true }}opacity-100 duration-200 ease-in{{ else }}opacity-0 duration-100 ease-out{{ end }} absolute inset-0 flex h-full w-full items-center justify-center transition-opacity"
+      aria-hidden="true">
+      <svg class="h-3 w-3 text-green-600" fill="currentColor" viewBox="0 0 12 12">
+        <path
+          d="M3.707 5.293a1 1 0 00-1.414 1.414l1.414-1.414zM5 8l-.707.707a1 1 0 001.414 0L5 8zm4.707-3.293a1 1 0 00-1.414-1.414l1.414 1.414zm-7.414 2l2 2 1.414-1.414-2-2-1.414 1.414zm3.414 2l4-4-1.414-1.414-4 4 1.414 1.414z"/>
+      </svg>
+    </span>
+  </span>
+</button>
+{{end}}
+
 
 <!-- 是否tag -->
 {{define "flagTag"}}

views/friend.html

@@ -103,6 +103,13 @@
                 </dd>
               </div>
 
+              <div class="flex justify-between gap-x-4 py-3">
+                <dt class="text-gray-500">热榜</dt>
+                <dd class="flex items-start gap-x-2">
+                  {{ template "hotTop" . }}
+                </dd>
+              </div>
+
               <div class="flex justify-between gap-x-4 py-3">
                 <dt class="text-gray-500">指令</dt>
                 <dd class="flex items-start gap-x-2">

views/group.html

@@ -117,6 +117,13 @@
                   {{ template "news" . }}
                 </dd>
               </div>
+              <!-- 热榜 -->
+              <div class="flex justify-between gap-x-4 py-3 items-center">
+                <dt class="text-gray-500">热榜</dt>
+                <dd class="flex items-start gap-x-2">
+                  {{ template "hotTop" . }}
+                </dd>
+              </div>
               <!-- 指令 -->
               <div class="flex justify-between gap-x-4 py-3 items-center">
                 <dt class="text-gray-500">指令</dt>

views/groupuser.html

@@ -1,9 +1,19 @@
 <dialog id="groupUserModal" class="modal">
   <div class="modal-box w-11/12 max-w-7xl">
+    <div class="flex justify-between">
       <div class="flex">
         <h3 class="font-bold text-lg" id="groupUserModalName">我是群名称</h3>
         <h3 class="font-bold text-lg ml-5" id="groupUserCount">(健在成员100人)</h3>
       </div>
+
+      <button type="button" onclick="document.getElementById('groupUserModal').close();" class="rounded-md bg-white text-gray-400 hover:text-gray-500 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2">
+        <span class="sr-only">关闭</span>
+        <svg class="h-6 w-6" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor" aria-hidden="true">
+          <path stroke-linecap="round" stroke-linejoin="round" d="M6 18L18 6M6 6l12 12" />
+        </svg>
+      </button>
+    </div>
+
     <div class="divider divider-warning">成员列表</div>
     <table class="table table-zebra">
       <thead>

views/static/js/index.js

@@ -99,6 +99,25 @@ function changeUserNewsStatus(wxId) {
   })
 }
 
+// 修改用户热搜开启状态
+function changeUserHotTopStatus(wxId) {
+  axios({
+    method: 'put',
+    url: '/api/hot-top/status',
+    data: {
+      wxId: wxId
+    }
+  }).then(function (response) {
+    console.log(`返回结果: ${JSON.stringify(response)}`);
+    alert(`${response.data}`)
+  }).catch(function (error) {
+    console.log(`错误信息: ${error}`);
+    alert("修改失败")
+  }).finally(function () {
+    window.location.reload();
+  })
+}
+
 // 修改指令权限启用状态
 function changeCommandEnableStatus(wxId) {
   axios({