91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"datarush/internal/gw/domain"
|
|
"datarush/internal/gw/grpc_client"
|
|
"datarush/internal/gw/storage"
|
|
"datarush/internal/gw/utils"
|
|
)
|
|
|
|
type SubmissionHandler struct {
|
|
submissionClient *grpc_client.SubmissionClient
|
|
s3Storage *storage.S3Storage
|
|
}
|
|
|
|
func NewSubmissionHandler(
|
|
submissionClient *grpc_client.SubmissionClient,
|
|
s3Storage *storage.S3Storage,
|
|
) *SubmissionHandler {
|
|
return &SubmissionHandler{
|
|
submissionClient: submissionClient,
|
|
s3Storage: s3Storage,
|
|
}
|
|
}
|
|
|
|
func (h *SubmissionHandler) SubmitTask(w http.ResponseWriter, r *http.Request) {
|
|
userID, err := getUserIDFromContext(r.Context())
|
|
if err != nil {
|
|
utils.RespondError(w, domain.NewUnauthorizedError("unauthorized"))
|
|
return
|
|
}
|
|
|
|
competitionID := getPathParam(r, "competition_id")
|
|
taskID := getPathParam(r, "task_id")
|
|
|
|
if err := r.ParseMultipartForm(32 << 20); err != nil { // 32 MB max
|
|
utils.RespondError(w, domain.NewBadRequestError("failed to parse form"))
|
|
return
|
|
}
|
|
|
|
file, header, err := r.FormFile("content")
|
|
if err != nil {
|
|
utils.RespondError(w, domain.NewBadRequestError("missing or invalid file"))
|
|
return
|
|
}
|
|
defer file.Close()
|
|
|
|
fileURL, err := h.s3Storage.UploadFile(r.Context(), file, header)
|
|
if err != nil {
|
|
utils.RespondError(w, domain.NewInternalServerError("failed to upload file"))
|
|
return
|
|
}
|
|
|
|
submission, err := h.submissionClient.SubmitTask(r.Context(), userID, competitionID, taskID, fileURL)
|
|
if err != nil {
|
|
utils.RespondError(w, domain.NewInternalServerError("failed to submit task"))
|
|
return
|
|
}
|
|
|
|
utils.RespondJSON(w, http.StatusCreated, &domain.SubmitTaskResponse{
|
|
SubmissionID: submission.Id,
|
|
})
|
|
}
|
|
|
|
func (h *SubmissionHandler) GetSubmissionHistory(w http.ResponseWriter, r *http.Request) {
|
|
userID, err := getUserIDFromContext(r.Context())
|
|
if err != nil {
|
|
utils.RespondError(w, domain.NewUnauthorizedError("unauthorized"))
|
|
return
|
|
}
|
|
|
|
competitionID := getPathParam(r, "competition_id")
|
|
taskID := getPathParam(r, "task_id")
|
|
|
|
submissions, err := h.submissionClient.GetSubmissionsHistory(r.Context(), userID, competitionID, taskID)
|
|
if err != nil {
|
|
utils.RespondError(w, domain.NewInternalServerError("failed to get submission history"))
|
|
return
|
|
}
|
|
|
|
response := make([]domain.SubmissionResponse, len(submissions))
|
|
for i, sub := range submissions {
|
|
response[i] = *utils.SubmissionProtoToHTTP(sub)
|
|
}
|
|
|
|
utils.RespondJSON(w, http.StatusOK, &domain.SubmissionHistoryResponse{
|
|
Submissions: response,
|
|
})
|
|
}
|