[feat]: 添加 User 转换函数 (#199)

This commit is contained in:
多吃点苹果 2023-01-12 18:24:14 +08:00 committed by GitHub
parent 26abfd6e49
commit 0af4b51d93
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

24
user.go
View File

@ -130,16 +130,40 @@ func (u *User) IsFriend() bool {
return !u.IsGroup() && strings.HasPrefix(u.UserName, "@") && u.VerifyFlag == 0
}
// AsFriend 将当前用户转换为好友类型
func (u *User) AsFriend() (*Friend, bool) {
if u.IsFriend() {
return &Friend{User: u}, true
}
return nil, false
}
// IsGroup 判断是否为群组
func (u *User) IsGroup() bool {
return strings.HasPrefix(u.UserName, "@@") && u.VerifyFlag == 0
}
// AsGroup 将当前用户转换为群组类型
func (u *User) AsGroup() (*Group, bool) {
if u.IsGroup() {
return &Group{User: u}, true
}
return nil, false
}
// IsMP 判断是否为公众号
func (u *User) IsMP() bool {
return u.VerifyFlag == 8 || u.VerifyFlag == 24 || u.VerifyFlag == 136
}
// AsMP 将当前用户转换为公众号类型
func (u *User) AsMP() (*Mp, bool) {
if u.IsMP() {
return &Mp{User: u}, true
}
return nil, false
}
// Pin 将联系人置顶
func (u *User) Pin() error {
req := u.self.bot.Storage.Request