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)
}