82 lines
1.9 KiB
Go
82 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path"
|
|
)
|
|
|
|
func saveTeamFile(dirname string, team string, filename string, w http.ResponseWriter, r *http.Request) bool {
|
|
if _, err := os.Stat(path.Join(dirname, team)); os.IsNotExist(err) || len(team) < 1 || team[0] == '_' {
|
|
http.Error(w, "{\"errmsg\":\"Requête invalide.\"}", http.StatusBadRequest)
|
|
return false
|
|
} else {
|
|
if len(filename) <= 0 {
|
|
log.Println("EMPTY $EXERCICE RECEIVED:", filename)
|
|
http.Error(w, "{\"errmsg\":\"Internal server error. Please retry in few seconds.\"}", http.StatusInternalServerError)
|
|
return false
|
|
}
|
|
|
|
// Previous submission not treated
|
|
if _, err := os.Stat(path.Join(SubmissionDir, team, filename)); !os.IsNotExist(err) {
|
|
http.Error(w, "{\"errmsg\":\"Du calme ! une requête est déjà en cours de traitement.\"}", http.StatusPaymentRequired)
|
|
return false
|
|
}
|
|
|
|
if err := saveFile(path.Join(SubmissionDir, team), filename, r); err != nil {
|
|
log.Println("Unable to handle submission file:", err)
|
|
http.Error(w, "{\"errmsg\":\"Internal server error. Please retry in few seconds.\"}", http.StatusInternalServerError)
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
}
|
|
|
|
func saveFile(dirname string, filename string, r *http.Request) error {
|
|
if _, err := os.Stat(dirname); os.IsNotExist(err) {
|
|
if err := os.MkdirAll(dirname, 0777); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
var f *os.File
|
|
|
|
if filename == "" {
|
|
if fd, err := ioutil.TempFile(dirname, ""); err != nil {
|
|
return err
|
|
} else {
|
|
defer f.Close()
|
|
f = fd
|
|
}
|
|
} else {
|
|
if fd, err := os.Create(path.Join(dirname, filename)); err != nil {
|
|
return err
|
|
} else {
|
|
defer f.Close()
|
|
f = fd
|
|
}
|
|
}
|
|
|
|
// Read request body
|
|
var body []byte
|
|
if r.ContentLength > 0 {
|
|
tmp := make([]byte, 1024)
|
|
for {
|
|
n, err := r.Body.Read(tmp)
|
|
for j := 0; j < n; j++ {
|
|
body = append(body, tmp[j])
|
|
}
|
|
if err != nil || n <= 0 {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
f.Write(body)
|
|
|
|
return nil
|
|
}
|