package friends import ( "encoding/json" "github.com/go-resty/resty/v2" "go-wechat/client" "go-wechat/common/constant" "go-wechat/config" "go-wechat/model/entity" model2 "go-wechat/model/model" "go-wechat/utils" "gorm.io/gorm" "log" "slices" "strings" "time" ) // 同步群成员 // http客户端 var hc = resty.New() // Sync // @description: 同步好友列表 func Sync() { var base model2.Response[[]model2.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, 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 model2.Response[model2.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 model2.ContactProfile, err error) { var baseResp model2.Response[model2.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 }