format User display

eatmoreapple · Dec 30, 2022 · e171c1f · e171c1f
1 parent 326f200
commit e171c1f
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/user.go b/user.go
@@ -54,7 +54,17 @@ type User struct {
 
 // implement fmt.Stringer
 func (u *User) String() string {
-	return fmt.Sprintf("<User:%s>", u.NickName)
+	format := "User"
+	if u.IsFriend() {
+		format = "Friend"
+	} else if u.IsGroup() {
+		format = "Group"
+	} else if u.IsMP() {
+		format = "MP"
+	} else if u.IsSelf() {
+		format = "Self"
+	}
+	return fmt.Sprintf("<%s:%s>", format, u.NickName)
 }
 
 // GetAvatarResponse 获取用户头像