Merge pull request #14 from ShanghaitechGeekPie/tyy

Implement personal reward display

Author: Prince213

internal/controllers/comments/course_group.go

@@ -6,8 +6,9 @@ import (
 	"coursebench-backend/pkg/database"
 	"coursebench-backend/pkg/errors"
 	"coursebench-backend/pkg/models"
-	"github.com/gofiber/fiber/v2"
 	"strconv"
+
+	"github.com/gofiber/fiber/v2"
 )
 
 func CourseGroupComment(c *fiber.Ctx) (err error) {

internal/controllers/comments/user.go

@@ -7,8 +7,9 @@ import (
 	"coursebench-backend/pkg/errors"
 	"coursebench-backend/pkg/models"
 	"coursebench-backend/pkg/queries"
-	"github.com/gofiber/fiber/v2"
 	"strconv"
+
+	"github.com/gofiber/fiber/v2"
 )
 
 type CommentResponse struct {
@@ -43,6 +44,7 @@ type CommentResponse struct {
 	CoverTitle   string `json:"cover_title"`
 	CoverContent string `json:"cover_content"`
 	CoverReason  string `json:"cover_reason"`
+	Reward       int    `json:"reward"`
 }
 
 type CommentLikeResult struct {
@@ -109,8 +111,16 @@ func GenerateResponse(comments []models.Comment, uid uint, likeResult []CommentL
 			CoverTitle:   v.CoverTitle,
 			CoverContent: v.CoverContent,
 			CoverReason:  v.CoverReason,
+			Reward:       v.Reward,
 		}
 		// 该评论未设置匿名，或者是自己的评论，则显示用户信息
+		currentUser, err := queries.GetUserByID(nil, uid)
+		if err != nil {
+			c.Reward = -2
+		}
+		if !currentUser.IsAdmin && !currentUser.IsCommunityAdmin {
+			c.Reward = -1
+		}
 		if !anonymous || v.User.ID == uid {
 			t, _ := queries.GetProfile(nil, v.UserID, uid)
 			c.User = &t

internal/controllers/users/profile.go

@@ -5,8 +5,9 @@ import (
 	"coursebench-backend/pkg/errors"
 	"coursebench-backend/pkg/models"
 	"coursebench-backend/pkg/queries"
-	"github.com/gofiber/fiber/v2"
 	"strconv"
+
+	"github.com/gofiber/fiber/v2"
 )
 
 func Profile(c *fiber.Ctx) error {

pkg/models/comment.go

@@ -2,6 +2,7 @@ package models
 
 import (
 	"coursebench-backend/pkg/modelRegister"
+
 	"github.com/lib/pq"
 	"gorm.io/gorm"
 )
@@ -30,6 +31,7 @@ type Comment struct {
 	CoverTitle          string
 	CoverContent        string
 	CoverReason         string
+	Reward              int
 }
 
 type CommentLike struct {

pkg/models/user.go

@@ -48,4 +48,5 @@ type ProfileResponse struct {
 	IsAdmin          bool      `json:"is_admin"`
 	IsCommunityAdmin bool      `json:"is_community_admin"`
 	InvitationCode   string    `json:"invitation_code"`
+	Reward           int       `json:"reward"`
 }

pkg/queries/user.go

@@ -350,6 +350,11 @@ func GetProfile(db *gorm.DB, id uint, uid uint) (models.ProfileResponse, error)
 		if id == uid {
 			r.InvitationCode = user.InvitationCode
 		}
+		if id == uid || user.IsAdmin || user.IsCommunityAdmin {
+			r.Reward = user.Reward
+		} else {
+			r.Reward = -1
+		}
 		return r, nil
 	}
 }