woj-server/internal/api/submission/create.go
2024-04-28 23:19:00 +08:00

88 lines
2.3 KiB
Go

package submission
import (
"git.0x7f.app/WOJ/woj-server/internal/e"
"git.0x7f.app/WOJ/woj-server/internal/model"
"git.0x7f.app/WOJ/woj-server/internal/service/problem"
"git.0x7f.app/WOJ/woj-server/internal/service/submission"
"github.com/gin-gonic/gin"
)
type createRequest struct {
Pid uint `form:"pid" json:"pid" binding:"required"`
Language string `form:"language" json:"language" binding:"required"`
Code string `form:"code" json:"code" binding:"required"`
}
// Create
// @Summary submit for judgement
// @Description Submit the code for judgement.
// @Tags submission
// @Accept application/x-www-form-urlencoded
// @Produce json
// @Param pid formData int true "problem id"
// @Param language formData string true "language"
// @Param code formData string true "code"
// @Response 200 {object} e.Response[uint] "submission id"
// @Security Authentication
// @Router /v1/submission/create [post]
func (h *handler) Create(c *gin.Context) {
claim, exist := c.Get("claim")
if !exist {
e.Pong[any](c, e.UserUnauthenticated, nil)
return
}
uid := claim.(*model.Claim).UID
role := claim.(*model.Claim).Role
// guest can not submit
if role < model.RoleGeneral {
e.Pong[any](c, e.UserUnauthorized, nil)
return
}
req := new(createRequest)
if err := c.ShouldBind(req); err != nil {
e.Pong(c, e.InvalidParameter, err.Error())
return
}
// check problem enabled
_, status := h.problemService.Query(&problem.QueryData{ID: req.Pid, Associations: false, ShouldEnable: role < model.RoleAdmin})
if status != e.Success {
e.Pong[any](c, status, nil)
return
}
// query latest version
pv, status := h.problemService.QueryLatestVersion(req.Pid)
if status != e.Success {
e.Pong[any](c, status, nil)
return
}
// create submission
createData := &submission.CreateData{
ProblemID: req.Pid,
UserID: uid,
Language: req.Language,
Code: req.Code,
}
res, status := h.submissionService.Create(createData)
if status != e.Success {
e.Pong[any](c, status, nil)
return
}
// submit judge
payload := &model.SubmitJudgePayload{
ProblemVersionID: pv.ID,
StorageKey: pv.StorageKey,
Submission: *res,
}
_, status = h.taskService.SubmitJudge(payload)
e.Pong[any](c, status, res.ID)
}