comqx/go-wxhelper
1
0
Fork 0
forked from lxh/go-wxhelper
Code Pull Requests Activity

🆕 新增热榜功能

Browse Source
This commit is contained in:
李寻欢 2024-07-15 14:14:24 +08:00
parent 24ddb1befe
commit 3c92f83745
15 changed files with 283 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
app/friend.go
View File

@ -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

10
config.yaml
View File

@ -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 * * * *' # 五分钟一次

1
config/task.go
View File

@ -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

20
model/dto/news.go
View File

@ -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"` // 手机端链接
}

40
model/entity/hottop.go Normal file
View File

@ -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
}

1
model/vo/friend.go
View File

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

1
router/router.go
View File

@ -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) // 获取群成员列表

9
service/friend.go
View File

@ -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

88
tasks/hottop/hottop.go Normal file
View File

@ -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
}

6
tasks/tasks.go
View File

@ -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)

29
utils/news.go
View File

@ -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
}

27
views/components.html
View File

@ -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"}}

7
views/friend.html
View File

@ -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">

7
views/group.html
View File

@ -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>

19
views/static/js/index.js
View File

@ -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({