diff --git a/config.yaml b/config.yaml index e4505ba..9358b3b 100644 --- a/config.yaml +++ b/config.yaml @@ -1,3 +1,20 @@ +system: + # 添加新好友或群之后通知给指定的人 + newFriendNotify: + enable: true + toUser: + - "wxid_xxx" + # 默认AI等配置 + defaultRule: + # 默认是否开启AI + ai: true + # 默认是否开启水群排行榜 + chatRank: true + # 默认是否开启聊天记录总结 + summary: true + # 默认是否开启新成员加群欢迎 + welcome: true + # 微信HOOK配置 wechat: # 微信HOOK接口地址 diff --git a/config/config.go b/config/config.go index 9cbae53..04192c9 100644 --- a/config/config.go +++ b/config/config.go @@ -6,6 +6,7 @@ var Conf conf // Config // @description: 配置 type conf struct { + System system `json:"system" yaml:"system"` // 系统配置 Task task `json:"task" yaml:"task"` // 定时任务配置 MySQL mysql `json:"mysql" yaml:"mysql"` // MySQL 配置 Wechat wechat `json:"wechat" yaml:"wechat"` // 微信助手 diff --git a/config/system.go b/config/system.go new file mode 100644 index 0000000..97d9341 --- /dev/null +++ b/config/system.go @@ -0,0 +1,21 @@ +package config + +// 系统配置 +type system struct { + NewFriendNotify newFriendNotify `json:"newFriendNotify" yaml:"newFriendNotify"` // 新好友通知 + DefaultRule defaultRule `json:"defaultRule" yaml:"defaultRule"` // 默认规则 +} + +// 添加新好友或群之后通知给指定的人 +type newFriendNotify struct { + Enable bool `json:"enable" yaml:"enable"` // 是否启用 + ToUser []string `json:"toUser" yaml:"toUser"` // 通知给谁 +} + +// 默认规则 +type defaultRule struct { + Ai bool `json:"ai" yaml:"ai"` // 是否启用AI + ChatRank bool `json:"chatRank" yaml:"chatRank"` // 是否启用聊天排行榜 + Summary bool `json:"summary" yaml:"summary"` // 是否启用聊天总结 + Welcome bool `json:"welcome" yaml:"welcome"` // 是否启用欢迎新成员 +} diff --git a/tasks/friends/friends.go b/tasks/friends/friends.go index a58eeb2..b5e96ee 100644 --- a/tasks/friends/friends.go +++ b/tasks/friends/friends.go @@ -41,6 +41,9 @@ func Sync() { nowIds := []string{} + // 新增的成员,用于通知给指定的人 + var newItmes = make(map[string]string) + for _, friend := range base.Data { if strings.Contains(friend.Wxid, "gh_") || strings.Contains(friend.Wxid, "@openim") { continue @@ -61,18 +64,23 @@ func Sync() { if count == 0 { // 新增 err = tx.Create(&entity.Friend{ - CustomAccount: friend.CustomAccount, - Nickname: friend.Nickname, - Pinyin: friend.Pinyin, - PinyinAll: friend.PinyinAll, - Wxid: friend.Wxid, - IsOk: true, - LastActive: time.Now().Local(), + CustomAccount: friend.CustomAccount, + Nickname: friend.Nickname, + Pinyin: friend.Pinyin, + PinyinAll: friend.PinyinAll, + Wxid: friend.Wxid, + IsOk: true, + EnableAi: config.Conf.System.DefaultRule.Ai, + EnableChatRank: config.Conf.System.DefaultRule.ChatRank, + EnableSummary: config.Conf.System.DefaultRule.Summary, + EnableWelcome: config.Conf.System.DefaultRule.Welcome, + LastActive: time.Now().Local(), }).Error if err != nil { log.Printf("新增好友失败: %s", err.Error()) continue } + newItmes[friend.Wxid] = friend.Nickname if conf, ok := config.Conf.Resource["introduce"]; ok { // 发送一条新消息 switch conf.Type { @@ -107,6 +115,21 @@ func Sync() { } } + // 通知有新成员 + if len(newItmes) > 0 && config.Conf.System.NewFriendNotify.Enable { + // 组装成一句话 + msg := []string{"#新好友通知\n"} + for wxId, nickname := range newItmes { + msg = append(msg, "微信Id: "+wxId+" -> 昵称: "+nickname) + } + for _, user := range config.Conf.System.NewFriendNotify.ToUser { + if user != "" { + // 发送一条新消息 + utils.SendMessage(user, "", strings.Join(msg, "\n"), 0) + } + } + } + // 清理不在列表中的好友 err = tx.Model(&entity.Friend{}).Where("wxid NOT IN (?)", nowIds).Update("is_ok", false).Error