backend: treat file found in directory discovered + handle multiple keys + generate my.json

backend/main.go

@@ -14,6 +14,7 @@ import (
 	"srs.epita.fr/fic-server/libfic"
 )
 
+var TeamsDir string
 var SubmissionDir string
 var BaseURL string
 
@@ -32,6 +33,8 @@ func watchsubdir(watcher *inotify.Watcher, pathname string) error {
 				if err := watchsubdir(watcher, p); err != nil {
 					return err
 				}
+			} else if d.Mode().IsRegular() {
+				go treat(p)
 			}
 		}
 		return nil
@@ -42,6 +45,7 @@ func main() {
 	var dbfile = flag.String("db", "fic.db", "Path to the DB")
 	flag.StringVar(&BaseURL, "baseurl", "http://fic.srs.epita.fr/", "URL prepended to each URL")
 	flag.StringVar(&SubmissionDir, "submission", "./submissions", "Base directory where save submissions")
+	flag.StringVar(&TeamsDir, "teams", "../TEAMS", "Base directory where save teams JSON files")
 	flag.Parse()
 
 	SubmissionDir = path.Clean(SubmissionDir)
@@ -84,17 +88,21 @@ func main() {
 					}
 				}
 			} else if ev.Mask&inotify.IN_CLOSE_WRITE == inotify.IN_CLOSE_WRITE {
-				// Extract
-				spath := strings.Split(strings.TrimPrefix(ev.Name, SubmissionDir), "/")
-
-				if len(spath) > 2 {
-					go treatSubmission(ev.Name, spath[1], spath[2])
-				} else {
-					log.Println("Invalid new file:", ev.Name)
-				}
+				go treat(ev.Name)
 			}
 		case err := <-watcher.Error:
 			log.Println("error:", err)
 		}
 	}
 }
+
+func treat(raw_path string) {
+	// Extract
+	spath := strings.Split(strings.TrimPrefix(raw_path, SubmissionDir), "/")
+
+	if len(spath) > 2 {
+		treatSubmission(raw_path, spath[1], spath[2])
+	} else {
+		log.Println("Invalid new file:", raw_path)
+	}
+}

backend/submission.go

@@ -1,12 +1,16 @@
 package main
 
 import (
-	"bufio"
 	"encoding/base64"
 	"encoding/binary"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"io/ioutil"
 	"log"
 	"math/rand"
 	"os"
+	"path"
 	"strconv"
 
 	"srs.epita.fr/fic-server/libfic"
@@ -18,6 +22,8 @@ func treatSubmission(pathname string, team_id string, exercice_id string) {
 	id := "[" + base64.StdEncoding.EncodeToString(bid) + "]"
 	log.Println(id, "New submission receive", pathname)
 
+	var keys map[string]string
+
 	if tid, err := strconv.Atoi(team_id); err != nil {
 		log.Println(id, "[ERR]", err)
 	} else if eid, err := strconv.Atoi(exercice_id); err != nil {
@@ -26,30 +32,47 @@ func treatSubmission(pathname string, team_id string, exercice_id string) {
 		log.Println(id, "[ERR]", err)
 	} else if exercice, err := fic.GetExercice(eid); err != nil {
 		log.Println(id, "[ERR]", err)
-	} else if file, err := os.Open(pathname); err != nil {
+	} else if theme, err := exercice.GetTheme(); err != nil {
+		log.Println(id, "[ERR]", err)
+	} else if cnt_raw, err := ioutil.ReadFile(pathname); err != nil {
+		log.Println(id, "[ERR]", err)
+	} else if err := json.Unmarshal(cnt_raw, &keys); err != nil {
 		log.Println(id, "[ERR]", err)
 	} else {
-		defer file.Close()
-
-		cnt := ""
-		scanner := bufio.NewScanner(file)
-		for scanner.Scan() {
-			cnt += scanner.Text()
-		}
-
-		if solved, err := exercice.CheckResponse(cnt, team); err != nil {
+		if solved, err := exercice.CheckResponse(keys, team); err != nil {
 			log.Println(id, "[ERR]", err)
 		} else if solved {
 			exercice.Solved(team)
-			log.Printf("%s Team %d SOLVED exercice %d\n", id, team.Id, exercice.Id)
+			log.Printf("%s Team %d SOLVED exercice %d (%s : %s)\n", id, team.Id, exercice.Id, theme.Name, exercice.Title)
 			if err := os.Remove(pathname); err != nil {
 				log.Println(id, "[ERR]", err)
 			}
 		} else {
-			log.Printf("%s Team %d submit an invalid solution for exercice %d\n", id, team.Id, exercice.Id)
+			log.Printf("%s Team %d submit an invalid solution for exercice %d (%s : %s)\n", id, team.Id, exercice.Id, theme.Name, exercice.Title)
 			if err := os.Remove(pathname); err != nil {
 				log.Println(id, "[ERR]", err)
 			}
 		}
+		genTeamMyFile(team)
 	}
 }
+
+func genTeamMyFile(team fic.Team) error {
+	dirPath := path.Join(TeamsDir, fmt.Sprintf("%d", team.Id))
+
+	if s, err := os.Stat(dirPath); os.IsNotExist(err) {
+		os.MkdirAll(dirPath, 0777)
+	} else if ! s.IsDir() {
+		return errors.New("dirPath is not a directory")
+	}
+
+	if my, err := fic.MyJSONTeam(&team, true); err != nil {
+		return err
+	} else if j, err := json.Marshal(my); err != nil {
+		return err
+	} else if err := ioutil.WriteFile(path.Join(dirPath, "my.json"), j, 0666); err != nil {
+		return err
+	}
+
+	return nil
+}

libfic/exercice.go

@@ -150,7 +150,7 @@ func (e Exercice) SolvedCount() int64 {
 	}
 }
 
-func (e Exercice) CheckResponse(response string, t Team) (bool, error) {
+func (e Exercice) CheckResponse(resps map[string]string, t Team) (bool, error) {
 	s, _, _ := t.HasSolved(e)
 	if s {
 		return true, nil
@@ -165,12 +165,18 @@ func (e Exercice) CheckResponse(response string, t Team) (bool, error) {
 			return true, errors.New("Exercice with no key registered")
 		}
 
-		for _, k := range keys {
-			if !k.Check(response) {
-				return false, nil
+		valid := true
+		for _, key := range keys {
+			if _, ok := resps[key.Type]; !ok {
+				valid = false
+				break
+			}
+			if !key.Check(resps[key.Type]) {
+				valid = false
+				break
 			}
 		}
 
-		return true, nil
+		return valid, nil
 	}
 }