sync: add dependency on flag to download file

libfic/db.go

@@ -163,6 +163,16 @@ CREATE TABLE IF NOT EXISTS exercice_keys(
   cksum BINARY(64) NOT NULL,
   FOREIGN KEY(id_exercice) REFERENCES exercices(id_exercice)
 ) DEFAULT CHARACTER SET = utf8mb4 COLLATE = utf8mb4_unicode_ci;
+`); err != nil {
+		return err
+	}
+	if _, err := db.Exec(`
+CREATE TABLE IF NOT EXISTS exercice_files_deps(
+  id_file INTEGER NOT NULL,
+  id_key INTEGER NOT NULL,
+  FOREIGN KEY(id_file) REFERENCES exercice_files(id_file),
+  FOREIGN KEY(id_key) REFERENCES exercice_keys(id_key)
+) DEFAULT CHARACTER SET = utf8 COLLATE = utf8_bin;
 `); err != nil {
 		return err
 	}

libfic/file.go

@@ -84,6 +84,15 @@ func GetFileByPath(path string) (EFile, error) {
 	return f, nil
 }
 
+// GetFileByFilename retrieves the file that should be called so.
+func (e Exercice) GetFileByFilename(filename string) (f EFile, err error) {
+	filename = path.Base(filename)
+
+	err = DBQueryRow("SELECT id_file, origin, path, id_exercice, name, cksum, size FROM exercice_files WHERE id_exercice = ? AND origin LIKE ?", e.Id, "%/" + filename).Scan(&f.Id, &f.origin, &f.Path, &f.IdExercice, &f.Name, &f.Checksum, &f.Size)
+
+	return
+}
+
 // GetFiles returns a list of files coming with the challenge.
 func (e Exercice) GetFiles() ([]EFile, error) {
 	if rows, err := DBQuery("SELECT id_file, origin, path, name, cksum, size FROM exercice_files WHERE id_exercice = ?", e.Id); err != nil {
@@ -262,6 +271,35 @@ func (f EFile) GetOrigin() string {
 	return f.origin
 }
 
+// AddDepend insert a new dependency to a given flag.
+func (f EFile) AddDepend(k Key) (err error) {
+	_, err = DBExec("INSERT INTO exercice_files_deps (id_file, id_key) VALUES (?, ?)", f.Id, k.Id)
+	return
+}
+
+// GetDepends retrieve the flag's dependency list.
+func (f EFile) GetDepends() ([]Key, error) {
+	if rows, err := DBQuery("SELECT id_key FROM exercice_files_deps WHERE id_file = ?", f.Id); err != nil {
+		return nil, err
+	} else {
+		defer rows.Close()
+
+		var deps = make([]Key, 0)
+		for rows.Next() {
+			var d int64
+			if err := rows.Scan(&d); err != nil {
+				return nil, err
+			}
+			deps = append(deps, Key{d, f.IdExercice, "", []byte{}})
+		}
+		if err := rows.Err(); err != nil {
+			return nil, err
+		}
+
+		return deps, nil
+	}
+}
+
 // CheckFileOnDisk recalculates the hash of the file on disk.
 func (f EFile) CheckFileOnDisk() error {
 	if _, size, err := checkFileHash(path.Join(FilesDir, f.Path), f.Checksum); err != nil {

libfic/key.go

@@ -79,7 +79,9 @@ func (k Key) Update() (int64, error) {
 
 // Delete the flag from the database.
 func (k Key) Delete() (int64, error) {
-	if res, err := DBExec("DELETE FROM exercice_keys WHERE id_key = ?", k.Id); err != nil {
+	if _, err := DBExec("DELETE FROM exercice_files_deps WHERE id_key = ?", k.Id); err != nil {
+		return 0, err
+	} else if res, err := DBExec("DELETE FROM exercice_keys WHERE id_key = ?", k.Id); err != nil {
 		return 0, err
 	} else if nb, err := res.RowsAffected(); err != nil {
 		return 0, err
@@ -90,7 +92,9 @@ func (k Key) Delete() (int64, error) {
 
 // WipeKeys deletes keys coming with the challenge.
 func (e Exercice) WipeKeys() (int64, error) {
-	if res, err := DBExec("DELETE FROM exercice_keys WHERE id_exercice = ?", e.Id); err != nil {
+	if _, err := DBExec("DELETE FROM exercice_files_deps WHERE id_key IN (SELECT id_key FROM exercice_keys WHERE id_exercice = ?)", e.Id); err != nil {
+		return 0, err
+	} else if res, err := DBExec("DELETE FROM exercice_keys WHERE id_exercice = ?", e.Id); err != nil {
 		return 0, err
 	} else if nb, err := res.RowsAffected(); err != nil {
 		return 0, err

libfic/reset.go

@@ -34,7 +34,7 @@ func ResetGame() (error) {
 
 // ResetExercices wipes out all challenges (both attempts and statements).
 func ResetExercices() (error) {
-	return truncateTable("team_hints", "exercice_files", "key_found", "exercice_keys", "exercice_solved", "exercice_tries", "exercice_hints", "mcq_found", "mcq_entries", "exercice_mcq", "exercices", "themes")
+	return truncateTable("team_hints", "exercice_files_deps", "exercice_files", "key_found", "exercice_keys", "exercice_solved", "exercice_tries", "exercice_hints", "mcq_found", "mcq_entries", "exercice_mcq", "exercices", "themes")
 }
 
 // ResetTeams wipes out all teams, incluings members and attempts.

libfic/team.go

@@ -1,6 +1,7 @@
 package fic
 
 import (
+	"log"
 	"time"
 )
 
@@ -104,6 +105,25 @@ func (t Team) HasAccess(e Exercice) bool {
 	}
 }
 
+// CanDownload checks if the Team has access to the given file.
+func (t Team) CanDownload(f EFile) bool {
+	if deps, err := f.GetDepends(); err != nil {
+		log.Printf("Unable to retrieve file dependencies: %s\n", err)
+		return false
+	} else {
+		res := true
+
+		for _, dep := range deps {
+			if t.HasPartiallySolved(dep) == nil {
+				res = false
+				break
+			}
+		}
+
+		return res
+	}
+}
+
 // NbTry retrieves the number of attempts made by the Team to the given challenge.
 func NbTry(t *Team, e Exercice) int {
 	var cnt *int

libfic/team_my.go

@@ -115,7 +115,9 @@ func MyJSONTeam(t *Team, started bool) (interface{}, error) {
 					return nil, err
 				} else {
 					for _, f := range files {
-						exercice.Files = append(exercice.Files, myTeamFile{path.Join(FilesDir, f.Path), f.Name, hex.EncodeToString(f.Checksum), f.Size})
+						if t == nil || t.CanDownload(f) {
+							exercice.Files = append(exercice.Files, myTeamFile{path.Join(FilesDir, f.Path), f.Name, hex.EncodeToString(f.Checksum), f.Size})
+						}
 					}
 				}