frontend: refactor and dispatch in many routes

frontend/save.go

@@ -0,0 +1,81 @@
+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
+}