tmp #99
@ -182,3 +182,26 @@ func ChangeSkipGroupRankStatus(ctx *gin.Context) {
|
||||
|
||||
ctx.String(http.StatusOK, "操作成功")
|
||||
}
|
||||
|
||||
// ChangeEnableNewsStatus
|
||||
// @description: 修改是否开启新闻
|
||||
// @param ctx
|
||||
func ChangeEnableNewsStatus(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_news`", gorm.Expr(" !`enable_news`")).Error
|
||||
if err != nil {
|
||||
log.Printf("修改早报启用状态失败:%s", err)
|
||||
ctx.String(http.StatusInternalServerError, "操作失败: %s", err)
|
||||
return
|
||||
}
|
||||
|
||||
ctx.String(http.StatusOK, "操作成功")
|
||||
}
|
||||
|
@ -14,6 +14,8 @@ system:
|
||||
summary: true
|
||||
# 默认是否开启新成员加群欢迎
|
||||
welcome: true
|
||||
# 每日早报
|
||||
news: true
|
||||
|
||||
# 微信HOOK配置
|
||||
wechat:
|
||||
@ -40,12 +42,15 @@ mysql:
|
||||
schema: public # postgres 专用
|
||||
|
||||
task:
|
||||
enable: false
|
||||
enable: true
|
||||
news:
|
||||
enable: true
|
||||
cron: '14 11 * * *' # 每天0:30
|
||||
syncFriends:
|
||||
enable: false
|
||||
cron: '*/5 * * * *' # 五分钟一次
|
||||
groupSummary:
|
||||
enable: true
|
||||
enable: false
|
||||
cron: '30 0 * * *' # 每天0:30
|
||||
waterGroup:
|
||||
enable: false
|
||||
|
@ -18,4 +18,5 @@ type defaultRule struct {
|
||||
ChatRank bool `json:"chatRank" yaml:"chatRank"` // 是否启用聊天排行榜
|
||||
Summary bool `json:"summary" yaml:"summary"` // 是否启用聊天总结
|
||||
Welcome bool `json:"welcome" yaml:"welcome"` // 是否启用欢迎新成员
|
||||
News bool `json:"news" yaml:"news"` // 是否启用每日早报
|
||||
}
|
||||
|
@ -4,6 +4,7 @@ package config
|
||||
// @description: 定时任务
|
||||
type task struct {
|
||||
Enable bool `json:"enable" yaml:"enable"` // 是否启用
|
||||
News syncFriends `json:"news" yaml:"news"` // 每日早报
|
||||
SyncFriends syncFriends `json:"syncFriends" yaml:"syncFriends"` // 同步好友
|
||||
WaterGroup waterGroup `json:"waterGroup" yaml:"waterGroup"` // 水群排行榜
|
||||
GroupSummary syncFriends `json:"groupSummary" yaml:"groupSummary"` // 群聊总结
|
||||
|
@ -19,6 +19,7 @@ type Friend struct {
|
||||
EnableChatRank bool `json:"enableChatRank" gorm:"type:tinyint(1) default 0 not null"` // 是否使用聊天排行
|
||||
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"` // 是否启用新闻
|
||||
IsOk bool `json:"isOk" gorm:"type:tinyint(1) default 0 not null"` // 是否正常
|
||||
}
|
||||
|
||||
|
18
model/news.go
Normal file
18
model/news.go
Normal file
@ -0,0 +1,18 @@
|
||||
package model
|
||||
|
||||
// MorningPost
|
||||
// @description: 每日早报返回结构体
|
||||
type MorningPost struct {
|
||||
Code int `json:"code"`
|
||||
Msg string `json:"msg"`
|
||||
Data struct {
|
||||
Date string `json:"date"` // 新闻日期
|
||||
News []string `json:"news"` // 新闻标题文字版
|
||||
WeiYu string `json:"weiyu"` // 微语,就是一句屁话
|
||||
Image string `json:"image"` // 早报完整图片
|
||||
HeadImage string `json:"head_image"` // 早报头部图片
|
||||
} `json:"data"`
|
||||
Time int `json:"time"`
|
||||
Usage int `json:"usage"`
|
||||
LogId string `json:"log_id"`
|
||||
}
|
@ -54,7 +54,7 @@ func AI(m *plugin.MessageContext) {
|
||||
|
||||
// 组装消息体
|
||||
messages := make([]openai.ChatCompletionMessage, 0)
|
||||
if config.Conf.Ai.Personality != "" {
|
||||
if prompt != "" {
|
||||
// 填充人设
|
||||
messages = append(messages, openai.ChatCompletionMessage{
|
||||
Role: openai.ChatMessageRoleSystem,
|
||||
|
@ -26,6 +26,7 @@ func Init(g *gin.Engine) {
|
||||
api.POST("/ai/model", app.ChangeUseAiModel) // 修改使用的AI模型
|
||||
api.PUT("/welcome/status", app.ChangeEnableWelcomeStatus) // 修改是否开启迎新状态
|
||||
api.PUT("/command/status", app.ChangeEnableCommandStatus) // 修改是否开启指令状态
|
||||
api.PUT("/news/status", app.ChangeEnableNewsStatus) // 修改是否开启早报状态
|
||||
api.PUT("/grouprank/status", app.ChangeEnableGroupRankStatus) // 修改是否开启水群排行榜状态
|
||||
api.PUT("/grouprank/skip", app.ChangeSkipGroupRankStatus) // 修改是否跳过水群排行榜状态
|
||||
api.GET("/group/users", app.GetGroupUsers) // 获取群成员列表
|
||||
|
@ -49,7 +49,10 @@ func GetAllEnableAI() (records []entity.Friend, err error) {
|
||||
// @return records
|
||||
// @return err
|
||||
func GetAllEnableChatRank() (records []entity.Friend, err error) {
|
||||
err = client.MySQL.Where("enable_chat_rank = ?", 1).Where("wxid LIKE '%@chatroom'").Find(&records).Error
|
||||
err = client.MySQL.Where("enable_chat_rank = ?", 1).
|
||||
Where("is_ok IS TRUE").
|
||||
Where("wxid LIKE '%@chatroom'").
|
||||
Find(&records).Error
|
||||
return
|
||||
}
|
||||
|
||||
@ -58,7 +61,19 @@ func GetAllEnableChatRank() (records []entity.Friend, err error) {
|
||||
// @return records
|
||||
// @return err
|
||||
func GetAllEnableSummary() (records []entity.Friend, err error) {
|
||||
err = client.MySQL.Where("enable_summary = ?", 1).Where("wxid LIKE '%@chatroom'").Find(&records).Error
|
||||
err = client.MySQL.Where("enable_summary = ?", 1).
|
||||
Where("is_ok IS TRUE").
|
||||
Where("wxid LIKE '%@chatroom'").
|
||||
Find(&records).Error
|
||||
return
|
||||
}
|
||||
|
||||
// GetAllEnableNews
|
||||
// @description: 取出所有启用了新闻的好友或群组
|
||||
// @return records
|
||||
// @return err
|
||||
func GetAllEnableNews() (records []entity.Friend, err error) {
|
||||
err = client.MySQL.Where("enable_news = ?", 1).Where("is_ok IS TRUE").Find(&records).Error
|
||||
return
|
||||
}
|
||||
|
||||
|
@ -74,6 +74,7 @@ func Sync() {
|
||||
EnableChatRank: config.Conf.System.DefaultRule.ChatRank,
|
||||
EnableSummary: config.Conf.System.DefaultRule.Summary,
|
||||
EnableWelcome: config.Conf.System.DefaultRule.Welcome,
|
||||
EnableNews: config.Conf.System.DefaultRule.News,
|
||||
LastActive: time.Now().Local(),
|
||||
}).Error
|
||||
if err != nil {
|
||||
|
36
tasks/news/news.go
Normal file
36
tasks/news/news.go
Normal file
@ -0,0 +1,36 @@
|
||||
package news
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"go-wechat/service"
|
||||
"go-wechat/utils"
|
||||
"log"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// DailyNews
|
||||
// @description: 每日新闻
|
||||
func DailyNews() {
|
||||
groups, err := service.GetAllEnableNews()
|
||||
if err != nil {
|
||||
log.Printf("获取启用了聊天排行榜的群组失败, 错误信息: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
news := utils.NewsUtil().MorningPost()
|
||||
if len(news) == 0 {
|
||||
log.Println("每日早报获取失败")
|
||||
return
|
||||
}
|
||||
|
||||
newsStr := fmt.Sprintf("#每日早报#\n\n又是新的一天了,让我们康康一觉醒来世界又发生了哪些变化~\n\n%s", strings.Join(news, "\n"))
|
||||
|
||||
// 循环发送新闻
|
||||
for _, group := range groups {
|
||||
// 发送消息
|
||||
utils.SendMessage(group.Wxid, "", newsStr, 0)
|
||||
// 休眠一秒,防止频繁发送
|
||||
time.Sleep(time.Second)
|
||||
}
|
||||
}
|
@ -4,6 +4,7 @@ import (
|
||||
"github.com/go-co-op/gocron"
|
||||
"go-wechat/config"
|
||||
"go-wechat/tasks/friends"
|
||||
"go-wechat/tasks/news"
|
||||
"go-wechat/tasks/summary"
|
||||
"go-wechat/tasks/watergroup"
|
||||
"log"
|
||||
@ -42,7 +43,6 @@ func InitTasks() {
|
||||
if config.Conf.Task.GroupSummary.Enable {
|
||||
log.Printf("群聊总结任务已启用,执行表达式: %s", config.Conf.Task.GroupSummary.Cron)
|
||||
_, _ = s.Cron(config.Conf.Task.GroupSummary.Cron).Do(summary.AiSummary)
|
||||
|
||||
}
|
||||
|
||||
// 更新好友列表
|
||||
@ -51,6 +51,11 @@ func InitTasks() {
|
||||
_, _ = s.Cron(config.Conf.Task.SyncFriends.Cron).Do(friends.Sync)
|
||||
}
|
||||
|
||||
// 每日早报
|
||||
if config.Conf.Task.News.Enable {
|
||||
_, _ = s.Cron(config.Conf.Task.News.Cron).Do(news.DailyNews)
|
||||
}
|
||||
|
||||
// 开启定时任务
|
||||
s.StartAsync()
|
||||
log.Println("定时任务初始化成功")
|
||||
|
46
utils/news.go
Normal file
46
utils/news.go
Normal file
@ -0,0 +1,46 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"github.com/go-resty/resty/v2"
|
||||
"go-wechat/model"
|
||||
"log"
|
||||
)
|
||||
|
||||
// News
|
||||
// @description: 新闻
|
||||
type News interface {
|
||||
MorningPost() []string // 早报
|
||||
}
|
||||
|
||||
type news struct{}
|
||||
|
||||
// NewsUtil
|
||||
// @description: 新闻工具
|
||||
// @param account
|
||||
// @param password
|
||||
// @return LeiGod
|
||||
func NewsUtil() News {
|
||||
return &news{}
|
||||
}
|
||||
|
||||
// MorningPost
|
||||
// @description: 早报
|
||||
// @receiver news
|
||||
// @return records
|
||||
func (news) MorningPost() (records []string) {
|
||||
var newsResp model.MorningPost
|
||||
|
||||
res := resty.New()
|
||||
resp, err := res.R().
|
||||
SetHeader("Content-Type", "application/json;chartset=utf-8").
|
||||
SetQueryParam("token", "cFoMZNNBxT4jQovS").
|
||||
SetResult(&newsResp).
|
||||
Post("https://v2.alapi.cn/api/zaobao")
|
||||
if err != nil {
|
||||
log.Panicf("每日早报获取失败: %s", err.Error())
|
||||
}
|
||||
log.Printf("每日早报获取结果: %s", unicodeToText(resp.String()))
|
||||
|
||||
records = newsResp.Data.News
|
||||
return
|
||||
}
|
@ -165,6 +165,33 @@
|
||||
</button>
|
||||
{{end}}
|
||||
|
||||
<!-- 早报 -->
|
||||
{{define "news"}}
|
||||
<button type="button"
|
||||
class="{{ if eq .EnableNews 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="changeUserNewsStatus({{.Wxid}})">
|
||||
<span
|
||||
class="{{ if eq .EnableNews 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 .EnableNews 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 .EnableNews 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"}}
|
||||
|
@ -78,6 +78,13 @@
|
||||
</div>
|
||||
{{ end }}
|
||||
|
||||
<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 "news" . }}
|
||||
</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">
|
||||
|
@ -33,19 +33,22 @@
|
||||
是否在通讯录
|
||||
</th>
|
||||
<th scope="col" class="px-3 py-3.5 text-left text-sm font-semibold text-gray-900">
|
||||
是否启用AI
|
||||
AI
|
||||
</th>
|
||||
<th scope="col" class="px-3 py-3.5 text-left text-sm font-semibold text-gray-900">
|
||||
是否启用水群排行榜
|
||||
排行榜
|
||||
</th>
|
||||
<th scope="col" class="px-3 py-3.5 text-left text-sm font-semibold text-gray-900">
|
||||
是否启用聊天记录总结
|
||||
总结
|
||||
</th>
|
||||
<th scope="col" class="px-3 py-3.5 text-left text-sm font-semibold text-gray-900">
|
||||
是否启用迎新
|
||||
迎新
|
||||
</th>
|
||||
<th scope="col" class="px-3 py-3.5 text-left text-sm font-semibold text-gray-900">
|
||||
是否启用指令
|
||||
早报
|
||||
</th>
|
||||
<th scope="col" class="px-3 py-3.5 text-left text-sm font-semibold text-gray-900">
|
||||
指令
|
||||
</th>
|
||||
<th scope="col" class="px-3 py-3.5 text-left text-sm font-semibold text-gray-900">
|
||||
操作
|
||||
@ -98,6 +101,9 @@
|
||||
<td class="whitespace-nowrap px-3 py-4 text-sm text-gray-500">
|
||||
{{ template "welcome" . }}
|
||||
</td>
|
||||
<td class="whitespace-nowrap px-3 py-4 text-sm text-gray-500">
|
||||
{{ template "news" . }}
|
||||
</td>
|
||||
<td class="whitespace-nowrap px-3 py-4 text-sm text-gray-500">
|
||||
{{ template "command" . }}
|
||||
</td>
|
||||
|
@ -80,6 +80,25 @@ function changeWelcomeEnableStatus(wxId) {
|
||||
})
|
||||
}
|
||||
|
||||
// 修改用户新闻开启状态
|
||||
function changeUserNewsStatus(wxId) {
|
||||
axios({
|
||||
method: 'put',
|
||||
url: '/api/news/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({
|
||||
|
@ -19,6 +19,7 @@ type FriendItem struct {
|
||||
EnableWelcome bool // 是否使用迎新
|
||||
EnableCommand bool // 是否启用指令
|
||||
EnableSummary bool // 是否启用总结
|
||||
EnableNews bool // 是否启用新闻
|
||||
IsOk bool // 是否还在通讯库(群聊是要还在群里也算)
|
||||
}
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user