2022-10-22 17:38:39 +08:00
|
|
|
package status
|
|
|
|
|
|
|
|
import (
|
2023-07-14 21:47:11 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/e"
|
2023-12-22 15:19:13 +08:00
|
|
|
"git.0x7f.app/WOJ/woj-server/internal/model"
|
|
|
|
"git.0x7f.app/WOJ/woj-server/pkg/utils"
|
2022-10-22 17:38:39 +08:00
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
)
|
|
|
|
|
|
|
|
type queryRequest struct {
|
2023-12-23 15:35:09 +08:00
|
|
|
Pid uint `form:"pid" json:"pid"`
|
|
|
|
Uid uint `form:"uid" json:"uid"`
|
|
|
|
Offset int `form:"offset" json:"offset"`
|
|
|
|
Limit int `form:"limit" json:"limit" binding:"required"`
|
2023-12-22 15:19:13 +08:00
|
|
|
}
|
|
|
|
|
2023-12-23 15:36:42 +08:00
|
|
|
type submissionWithScore struct {
|
2023-12-22 15:19:13 +08:00
|
|
|
Submission model.Submission `json:"submission"`
|
|
|
|
Point int32 `json:"point"`
|
2022-10-22 17:38:39 +08:00
|
|
|
}
|
|
|
|
|
2022-10-23 17:29:35 +08:00
|
|
|
// Query
|
2023-12-22 15:19:13 +08:00
|
|
|
// @Summary query status via problem id or user id
|
|
|
|
// @Description Batch query judgement status based on either the question or user.
|
2023-12-18 21:21:15 +08:00
|
|
|
// @Tags status
|
2022-10-23 17:29:35 +08:00
|
|
|
// @Accept application/x-www-form-urlencoded
|
|
|
|
// @Produce json
|
2023-12-22 15:19:13 +08:00
|
|
|
// @Param pid formData uint false "problem id"
|
|
|
|
// @Param uid formData uint false "user id"
|
|
|
|
// @Param offset formData int false "start position"
|
|
|
|
// @Param limit formData int true "limit number of records"
|
2023-12-23 15:36:42 +08:00
|
|
|
// @Response 200 {object} e.Response[e.WithCount[submissionWithScore]] "status"
|
|
|
|
// @Security Authentication
|
2022-10-23 17:29:35 +08:00
|
|
|
// @Router /v1/status/query [post]
|
2022-10-22 17:38:39 +08:00
|
|
|
func (h *handler) Query(c *gin.Context) {
|
2023-12-22 15:19:13 +08:00
|
|
|
claim, exist := c.Get("claim")
|
|
|
|
if !exist {
|
|
|
|
e.Pong[any](c, e.UserUnauthenticated, nil)
|
|
|
|
return
|
|
|
|
}
|
2023-07-16 16:27:12 +08:00
|
|
|
|
2022-10-22 17:38:39 +08:00
|
|
|
req := new(queryRequest)
|
|
|
|
if err := c.ShouldBind(req); err != nil {
|
|
|
|
e.Pong(c, e.InvalidParameter, err.Error())
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2023-12-22 15:19:13 +08:00
|
|
|
if req.Pid == 0 && req.Uid == 0 {
|
|
|
|
e.Pong[any](c, e.InvalidParameter, nil)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
submissions, status := h.submissionService.Query(req.Pid, req.Uid, req.Offset, req.Limit)
|
|
|
|
|
|
|
|
uid := claim.(*model.Claim).UID
|
|
|
|
role := claim.(*model.Claim).Role
|
|
|
|
var response []*queryResponse
|
|
|
|
|
2023-12-23 15:36:42 +08:00
|
|
|
var count int64
|
|
|
|
submissions, status := h.submissionService.Query(req.Pid, req.Uid, req.Offset, req.Limit, &count)
|
|
|
|
|
|
|
|
var response []*submissionWithScore
|
2023-12-22 15:19:13 +08:00
|
|
|
for _, submission := range submissions {
|
|
|
|
cur, _ := h.statusService.Query(submission.ID, false)
|
|
|
|
point := utils.If(cur == nil, -1, cur.Point)
|
2023-12-23 15:36:42 +08:00
|
|
|
resp := &submissionWithScore{
|
2023-12-22 15:19:13 +08:00
|
|
|
Submission: *submission,
|
|
|
|
Point: point,
|
|
|
|
}
|
|
|
|
|
|
|
|
if role < model.RoleAdmin || uid != submission.UserID {
|
|
|
|
// strip out code
|
|
|
|
resp.Submission.Code = ""
|
|
|
|
}
|
|
|
|
|
|
|
|
response = append(response, resp)
|
|
|
|
}
|
2022-10-22 17:38:39 +08:00
|
|
|
|
2023-12-23 15:36:42 +08:00
|
|
|
e.Pong(c, status, e.WithCount[*submissionWithScore]{Count: count, Data: response})
|
2022-10-22 17:38:39 +08:00
|
|
|
}
|