Skip to content

Commit 44afaa5

Browse files
committed
feat(queries.getprofile): add condition to get comment reward
1 parent 88ce7f3 commit 44afaa5

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

pkg/queries/user.go

+10-1
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ import (
66
"coursebench-backend/pkg/errors"
77
"coursebench-backend/pkg/mail"
88
"coursebench-backend/pkg/models"
9+
"coursebench-backend/pkg/queries"
910
"fmt"
1011
"math/rand"
1112
"strings"
@@ -346,9 +347,17 @@ func GetProfile(db *gorm.DB, id uint, uid uint) (models.ProfileResponse, error)
346347
if user.IsAnonymous && id != uid {
347348
return models.ProfileResponse{ID: id, NickName: user.NickName, Avatar: avatar, IsAnonymous: user.IsAnonymous, IsAdmin: user.IsAdmin, IsCommunityAdmin: user.IsCommunityAdmin}, nil
348349
} else {
349-
r := models.ProfileResponse{ID: id, Email: user.Email, Year: user.Year, Grade: user.Grade, NickName: user.NickName, RealName: user.RealName, IsAnonymous: user.IsAnonymous, Avatar: avatar, IsAdmin: user.IsAdmin, IsCommunityAdmin: user.IsCommunityAdmin, Reward: user.Reward}
350+
r := models.ProfileResponse{ID: id, Email: user.Email, Year: user.Year, Grade: user.Grade, NickName: user.NickName, RealName: user.RealName, IsAnonymous: user.IsAnonymous, Avatar: avatar, IsAdmin: user.IsAdmin, IsCommunityAdmin: user.IsCommunityAdmin}
350351
if id == uid {
351352
r.InvitationCode = user.InvitationCode
353+
r.Reward = user.Reward
354+
}
355+
currentUser, err := queries.GetUserByID(nil, uid)
356+
if err != nil {
357+
return err
358+
}
359+
if currentUser.IsAdmin || !currentUser.IsCommunityAdmin {
360+
r.Reward = user.Reward
352361
}
353362
return r, nil
354363
}

0 commit comments

Comments
 (0)