forked from lxh/go-wxhelper
267 lines
7.2 KiB
Go
267 lines
7.2 KiB
Go
package friends
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/go-resty/resty/v2"
|
|
"go-wechat/client"
|
|
"go-wechat/common/constant"
|
|
"go-wechat/config"
|
|
"go-wechat/model/dto"
|
|
"go-wechat/model/entity"
|
|
"go-wechat/utils"
|
|
"gorm.io/gorm"
|
|
"log"
|
|
"slices"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// 同步群成员
|
|
|
|
// http客户端
|
|
var hc = resty.New()
|
|
|
|
// Sync
|
|
// @description: 同步好友列表
|
|
func Sync() {
|
|
var base dto.Response[[]dto.FriendItem]
|
|
|
|
resp, err := hc.R().
|
|
SetHeader("Content-Type", "application/json;chartset=utf-8").
|
|
SetResult(&base).
|
|
Post(config.Conf.Wechat.GetURL("/api/getContactList"))
|
|
if err != nil {
|
|
log.Printf("获取好友列表失败: %s", err.Error())
|
|
return
|
|
}
|
|
log.Printf("获取好友列表结果: %s", resp.String())
|
|
|
|
tx := client.MySQL.Begin()
|
|
defer tx.Commit()
|
|
|
|
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
|
|
}
|
|
// 特殊Id跳过
|
|
if slices.Contains(constant.SpecialId, friend.Wxid) {
|
|
continue
|
|
}
|
|
//log.Printf("昵称: %s -> 类型: %d -> 微信号: %s -> 微信原始Id: %s", friend.Nickname, friend.Type, friend.CustomAccount, friend.Wxid)
|
|
nowIds = append(nowIds, friend.Wxid)
|
|
|
|
// 判断是否存在，不存在的话就新增，存在就修改一下名字
|
|
var count int64
|
|
err = tx.Model(&entity.Friend{}).Where("wxid = ?", friend.Wxid).Count(&count).Error
|
|
if err != nil {
|
|
continue
|
|
}
|
|
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,
|
|
EnableAi: config.Conf.System.DefaultRule.Ai,
|
|
EnableChatRank: config.Conf.System.DefaultRule.ChatRank,
|
|
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
|
|
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 {
|
|
case "text":
|
|
// 文字类型
|
|
utils.SendMessage(friend.Wxid, "", conf.Path, 0)
|
|
case "image":
|
|
// 图片类型
|
|
utils.SendImage(friend.Wxid, conf.Path, 0)
|
|
case "emotion":
|
|
// 表情类型
|
|
utils.SendEmotion(friend.Wxid, conf.Path, 0)
|
|
}
|
|
}
|
|
} else {
|
|
pm := map[string]any{
|
|
"nickname": friend.Nickname,
|
|
"custom_account": friend.CustomAccount,
|
|
"pinyin": friend.Pinyin,
|
|
"pinyin_all": friend.PinyinAll,
|
|
"is_ok": true,
|
|
}
|
|
err = tx.Model(&entity.Friend{}).Where("wxid = ?", friend.Wxid).Updates(pm).Error
|
|
if err != nil {
|
|
log.Printf("修改好友失败: %s", err.Error())
|
|
continue
|
|
}
|
|
}
|
|
|
|
// 群成员，同步一下成员信息
|
|
if strings.Contains(friend.Wxid, "@chatroom") {
|
|
syncGroupUsers(tx, friend.Wxid)
|
|
}
|
|
}
|
|
|
|
// 通知有新成员
|
|
if len(newItmes) > 0 && config.Conf.System.NewFriendNotify.Enable {
|
|
// 组装成一句话
|
|
msg := []string{"#新好友通知\n"}
|
|
for wxId, nickname := range newItmes {
|
|
msg = append(msg, "微信Id: "+wxId+"\n昵称: "+nickname)
|
|
}
|
|
for _, user := range config.Conf.System.NewFriendNotify.ToUser {
|
|
if user != "" {
|
|
// 发送一条新消息
|
|
utils.SendMessage(user, "", strings.Join(msg, "\n-------\n"), 0)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 清理不在列表中的好友
|
|
clearPm := map[string]any{
|
|
"is_ok": false,
|
|
"enable_chat_rank": false,
|
|
"enable_welcome": false,
|
|
"enable_summary": false,
|
|
"enable_news": false,
|
|
"clear_member": false,
|
|
"enable_ai": false,
|
|
}
|
|
err = tx.Model(&entity.Friend{}).Where("wxid NOT IN (?)", nowIds).Updates(clearPm).Error
|
|
if err != nil {
|
|
log.Printf("清理好友失败: %s", err.Error())
|
|
}
|
|
|
|
log.Println("同步好友列表完成")
|
|
}
|
|
|
|
// syncGroupUsers
|
|
// @description: 同步群成员
|
|
// @param gid
|
|
func syncGroupUsers(tx *gorm.DB, gid string) {
|
|
var baseResp dto.Response[dto.GroupUser]
|
|
|
|
// 组装参数
|
|
param := map[string]any{
|
|
"chatRoomId": gid, // 群Id
|
|
}
|
|
pbs, _ := json.Marshal(param)
|
|
|
|
_, err := hc.R().
|
|
SetHeader("Content-Type", "application/json;chartset=utf-8").
|
|
SetBody(string(pbs)).
|
|
SetResult(&baseResp).
|
|
Post(config.Conf.Wechat.GetURL("/api/getMemberFromChatRoom"))
|
|
if err != nil {
|
|
log.Printf("获取群成员信息失败: %s", err.Error())
|
|
return
|
|
}
|
|
|
|
// 昵称Id
|
|
wxIds := strings.Split(baseResp.Data.Members, "^G")
|
|
//log.Printf(" 群成员数: %d", len(wxIds))
|
|
|
|
// 修改不在数组的群成员状态为不在
|
|
pm := map[string]any{
|
|
"is_member": false,
|
|
"leave_time": time.Now().Local(),
|
|
}
|
|
err = tx.Model(&entity.GroupUser{}).Where("group_id = ?", gid).Where("is_member IS TRUE").Where("wxid NOT IN (?)", wxIds).Updates(pm).Error
|
|
if err != nil {
|
|
log.Printf("修改群成员状态失败: %s", err.Error())
|
|
return
|
|
}
|
|
|
|
for _, wxid := range wxIds {
|
|
// 获取成员信息
|
|
cp, _ := getContactProfile(wxid)
|
|
if cp.Wxid != "" {
|
|
//log.Printf(" 微信Id: %s -> 昵称: %s -> 微信号: %s", wxid, cp.Nickname, cp.Account)
|
|
// 查询成员是否存在，不在就新增，否则修改
|
|
var count int64
|
|
err = tx.Model(&entity.GroupUser{}).Where("group_id = ?", gid).Where("wxid = ?", wxid).Count(&count).Error
|
|
if err != nil {
|
|
log.Printf("查询群成员失败: %s", err.Error())
|
|
continue
|
|
}
|
|
if count == 0 {
|
|
// 新增
|
|
err = tx.Create(&entity.GroupUser{
|
|
GroupId: gid,
|
|
Account: cp.Account,
|
|
HeadImage: cp.HeadImage,
|
|
Nickname: cp.Nickname,
|
|
Wxid: cp.Wxid,
|
|
IsMember: true,
|
|
IsAdmin: wxid == baseResp.Data.Admin,
|
|
JoinTime: time.Now().Local(),
|
|
LastActive: time.Now().Local(),
|
|
}).Error
|
|
if err != nil {
|
|
log.Printf("新增群成员失败: %s", err.Error())
|
|
continue
|
|
}
|
|
} else {
|
|
// 修改
|
|
pm := map[string]any{
|
|
"account": cp.Account,
|
|
"head_image": cp.HeadImage,
|
|
"nickname": cp.Nickname,
|
|
"is_member": true,
|
|
"is_admin": wxid == baseResp.Data.Admin,
|
|
"leave_time": nil,
|
|
}
|
|
err = tx.Model(&entity.GroupUser{}).Where("group_id = ?", gid).Where("wxid = ?", wxid).Updates(pm).Error
|
|
if err != nil {
|
|
log.Printf("修改群成员失败: %s", err.Error())
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// getContactProfile
|
|
// @description: 获取成员详情
|
|
// @param wxid
|
|
// @return ent
|
|
// @return err
|
|
func getContactProfile(wxid string) (ent dto.ContactProfile, err error) {
|
|
var baseResp dto.Response[dto.ContactProfile]
|
|
|
|
// 组装参数
|
|
param := map[string]any{
|
|
"wxid": wxid, // 群Id
|
|
}
|
|
pbs, _ := json.Marshal(param)
|
|
|
|
_, err = hc.R().
|
|
SetHeader("Content-Type", "application/json;chartset=utf-8").
|
|
SetBody(string(pbs)).
|
|
SetResult(&baseResp).
|
|
Post(config.Conf.Wechat.GetURL("/api/getContactProfile"))
|
|
if err != nil {
|
|
log.Printf("获取成员详情失败: %s", err.Error())
|
|
return
|
|
}
|
|
ent = baseResp.Data
|
|
return
|
|
}
|