go-wxhelper/utils/friend.go

106 lines
2.5 KiB
Go
Raw Normal View History

package utils
import (
"encoding/json"
"github.com/go-resty/resty/v2"
"go-wechat/common/constant"
"go-wechat/config"
"go-wechat/model/dto"
"log"
"slices"
"strings"
)
// http客户端
var hc = resty.New()
// GetFriendList
// @description: 获取好友列表
// @return friends
// @return err
func GetFriendList() (friends []dto.FriendItem, err error) {
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())
// 循环获取到的好友列表
for _, item := range base.Data {
// 跳过特殊账号
// 跳过公众号和企业微信好友
if strings.Contains(item.Wxid, "gh_") || strings.Contains(item.Wxid, "@openim") {
continue
}
// 特殊Id跳过
if slices.Contains(constant.SpecialId, item.Wxid) {
continue
}
// 添加到待返回列表
friends = append(friends, item)
}
return
}
// GetGroupMembers
// @description: 获取指定群成员
// @param wxId 群Id
// @return ids 群成员id数组
// @return members 群成员信息
// @return err 错误信息
func GetGroupMembers(wxId string) (ids []string, admin string, err error) {
var base dto.Response[dto.GroupUser]
// 组装参数
param := map[string]any{
"chatRoomId": wxId, // 群Id
}
pbs, _ := json.Marshal(param)
_, err = hc.R().
SetHeader("Content-Type", "application/json;chartset=utf-8").
SetBody(string(pbs)).
SetResult(&base).
Post(config.Conf.Wechat.GetURL("/api/getMemberFromChatRoom"))
if err != nil {
log.Printf("获取群成员信息失败: %s", err.Error())
return
}
// 昵称Id
ids = strings.Split(base.Data.Members, "^G")
admin = base.Data.Admin
return
}
// GetContactProfile
// @description: 获取联系人信息
// @param wxId 微信Id
// @return info 联系人信息
// @return err 错误信息
func GetContactProfile(wxId string) (info 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
}
info = baseResp.Data
return
}