From 3c92f83745f3e20beb21504e358f540e406d919d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=AF=BB=E6=AC=A2?= Date: Mon, 15 Jul 2024 14:14:24 +0800 Subject: [PATCH] =?UTF-8?q?:new:=20=E6=96=B0=E5=A2=9E=E7=83=AD=E6=A6=9C?= =?UTF-8?q?=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/friend.go | 23 +++++++++++ config.yaml | 10 +++-- config/task.go | 1 + model/dto/news.go | 20 +++++++++ model/entity/hottop.go | 40 ++++++++++++++++++ model/vo/friend.go | 1 + router/router.go | 1 + service/friend.go | 9 ++++ tasks/hottop/hottop.go | 88 ++++++++++++++++++++++++++++++++++++++++ tasks/tasks.go | 6 +++ utils/news.go | 29 ++++++++++++- views/components.html | 27 ++++++++++++ views/friend.html | 7 ++++ views/group.html | 7 ++++ views/static/js/index.js | 19 +++++++++ 15 files changed, 283 insertions(+), 5 deletions(-) create mode 100644 model/entity/hottop.go create mode 100644 tasks/hottop/hottop.go diff --git a/app/friend.go b/app/friend.go index 7f65080e..42b549c1 100644 --- a/app/friend.go +++ b/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 diff --git a/config.yaml b/config.yaml index 15793edf..86476839 100644 --- a/config.yaml +++ b/config.yaml @@ -23,7 +23,7 @@ system: # 微信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 +44,15 @@ mysql: db: wechat schema: public # postgres 专用 +# 定时任务 task: - enable: false + enable: true news: - enable: true + enable: false cron: '14 11 * * *' # 每天0:30 + hotTop: + enable: true + cron: '0 */1 * * *' # 每小时一次 syncFriends: enable: false cron: '*/5 * * * *' # 五分钟一次 diff --git a/config/task.go b/config/task.go index f29ad57f..eb1c0408 100644 --- a/config/task.go +++ b/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 diff --git a/model/dto/news.go b/model/dto/news.go index bf2b10bd..94d24fc8 100644 --- a/model/dto/news.go +++ b/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"` // 手机端链接 +} diff --git a/model/entity/hottop.go b/model/entity/hottop.go new file mode 100644 index 00000000..63ed3d07 --- /dev/null +++ b/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 +} diff --git a/model/vo/friend.go b/model/vo/friend.go index c948a4e7..d51420d8 100644 --- a/model/vo/friend.go +++ b/model/vo/friend.go @@ -21,6 +21,7 @@ type FriendItem struct { EnableCommand bool // 是否启用指令 EnableSummary bool // 是否启用总结 EnableNews bool // 是否启用新闻 + EnableHotTop bool // 是否启用热搜 ClearMember int // 清理成员配置（多少天未活跃的） IsOk bool // 是否还在通讯库(群聊是要还在群里也算) } diff --git a/router/router.go b/router/router.go index d2163608..2b40dfde 100644 --- a/router/router.go +++ b/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) // 获取群成员列表 diff --git a/service/friend.go b/service/friend.go index d10cbed2..93a67736 100644 --- a/service/friend.go +++ b/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 diff --git a/tasks/hottop/hottop.go b/tasks/hottop/hottop.go new file mode 100644 index 00000000..6de3175a --- /dev/null +++ b/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 +} diff --git a/tasks/tasks.go b/tasks/tasks.go index 9627c010..f441b321 100644 --- a/tasks/tasks.go +++ b/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) diff --git a/utils/news.go b/utils/news.go index db2317cd..7b46a1c7 100644 --- a/utils/news.go +++ b/utils/news.go @@ -10,7 +10,8 @@ import ( // News // @description: 新闻 type News interface { - MorningPost() []string // 早报 + 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 +} diff --git a/views/components.html b/views/components.html index 96ebc1ea..02b74132 100644 --- a/views/components.html +++ b/views/components.html @@ -192,6 +192,33 @@ {{end}} + +{{define "hotTop"}} + +{{end}} + {{define "flagTag"}} diff --git a/views/friend.html b/views/friend.html index e97c7194..04c660db 100644 --- a/views/friend.html +++ b/views/friend.html @@ -103,6 +103,13 @@ +
+
热榜
+
+ {{ template "hotTop" . }} +
+
+
指令
diff --git a/views/group.html b/views/group.html index add70ccf..04990f40 100644 --- a/views/group.html +++ b/views/group.html @@ -117,6 +117,13 @@ {{ template "news" . }}
+ +
+
热榜
+
+ {{ template "hotTop" . }} +
+
指令
diff --git a/views/static/js/index.js b/views/static/js/index.js index 95d6d85a..aad91700 100644 --- a/views/static/js/index.js +++ b/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({