2016-01-13 19:38:45 +00:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"flag"
|
|
|
|
"io/ioutil"
|
|
|
|
"log"
|
2016-01-14 17:17:35 +00:00
|
|
|
"math/rand"
|
2016-01-13 19:38:45 +00:00
|
|
|
"os"
|
|
|
|
"path"
|
2018-01-21 13:07:44 +00:00
|
|
|
"strconv"
|
2016-01-13 19:38:45 +00:00
|
|
|
"strings"
|
2016-01-14 17:17:35 +00:00
|
|
|
"time"
|
2016-01-13 19:38:45 +00:00
|
|
|
|
|
|
|
"srs.epita.fr/fic-server/libfic"
|
2016-12-30 11:45:14 +00:00
|
|
|
"srs.epita.fr/fic-server/settings"
|
|
|
|
|
|
|
|
"gopkg.in/fsnotify.v1"
|
2016-01-13 19:38:45 +00:00
|
|
|
)
|
|
|
|
|
2016-01-23 11:28:31 +00:00
|
|
|
var TeamsDir string
|
2016-01-13 19:38:45 +00:00
|
|
|
var SubmissionDir string
|
|
|
|
|
2016-12-08 08:14:42 +00:00
|
|
|
func watchsubdir(watcher *fsnotify.Watcher, pathname string) error {
|
2016-01-13 19:38:45 +00:00
|
|
|
log.Println("Watch new directory:", pathname)
|
2016-12-08 08:14:42 +00:00
|
|
|
if err := watcher.Add(pathname); err != nil {
|
2016-01-13 19:38:45 +00:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if ds, err := ioutil.ReadDir(pathname); err != nil {
|
|
|
|
return err
|
|
|
|
} else {
|
|
|
|
for _, d := range ds {
|
|
|
|
p := path.Join(pathname, d.Name())
|
2018-01-21 13:03:55 +00:00
|
|
|
if d.IsDir() && d.Name() != ".tmp" && d.Mode() & os.ModeSymlink == 0 {
|
2016-01-13 19:38:45 +00:00
|
|
|
if err := watchsubdir(watcher, p); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
2016-01-23 11:28:31 +00:00
|
|
|
} else if d.Mode().IsRegular() {
|
|
|
|
go treat(p)
|
2016-01-13 19:38:45 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-01-24 01:14:28 +00:00
|
|
|
var lastRegeneration time.Time
|
|
|
|
|
2016-12-30 11:45:14 +00:00
|
|
|
func reloadSettings(config settings.FICSettings) {
|
2017-01-24 01:14:28 +00:00
|
|
|
if lastRegeneration != config.Generation || fic.PartialValidation != config.PartialValidation || fic.UnlockedChallenges != !config.EnableExerciceDepend || fic.FirstBlood != config.FirstBlood || fic.SubmissionCostBase != config.SubmissionCostBase {
|
|
|
|
lastRegeneration = config.Generation
|
|
|
|
|
2017-01-16 12:12:57 +00:00
|
|
|
fic.PartialValidation = config.PartialValidation
|
|
|
|
fic.UnlockedChallenges = !config.EnableExerciceDepend
|
2016-12-30 11:45:14 +00:00
|
|
|
|
2017-01-16 12:12:57 +00:00
|
|
|
fic.FirstBlood = config.FirstBlood
|
|
|
|
fic.SubmissionCostBase = config.SubmissionCostBase
|
2016-12-30 11:45:14 +00:00
|
|
|
|
2017-01-16 12:12:57 +00:00
|
|
|
log.Println("Generating files...")
|
|
|
|
go func() {
|
|
|
|
genAll()
|
|
|
|
log.Println("Full generation done")
|
|
|
|
}()
|
|
|
|
} else {
|
|
|
|
log.Println("No change found. Skipping regeneration.")
|
|
|
|
}
|
2016-12-30 11:45:14 +00:00
|
|
|
}
|
|
|
|
|
2016-01-13 19:38:45 +00:00
|
|
|
func main() {
|
2017-10-17 04:47:10 +00:00
|
|
|
var dsn = flag.String("dsn", fic.DSNGenerator(), "DSN to connect to the MySQL server")
|
2017-11-25 15:05:03 +00:00
|
|
|
flag.StringVar(&settings.SettingsDir, "settings", settings.SettingsDir, "Base directory where load and save settings")
|
2016-01-13 19:38:45 +00:00
|
|
|
flag.StringVar(&SubmissionDir, "submission", "./submissions", "Base directory where save submissions")
|
2016-12-30 11:45:14 +00:00
|
|
|
flag.StringVar(&TeamsDir, "teams", "./TEAMS", "Base directory where save teams JSON files")
|
2016-11-19 14:54:32 +00:00
|
|
|
flag.StringVar(&fic.FilesDir, "files", "/files", "Request path prefix to reach files")
|
2017-11-22 01:39:52 +00:00
|
|
|
var debugINotify = flag.Bool("debuginotify", false, "Show skipped inotofy events")
|
2016-01-13 19:38:45 +00:00
|
|
|
flag.Parse()
|
|
|
|
|
2016-10-13 17:02:41 +00:00
|
|
|
log.SetPrefix("[backend] ")
|
|
|
|
|
2017-11-25 15:05:03 +00:00
|
|
|
settings.SettingsDir = path.Clean(settings.SettingsDir)
|
2016-01-13 19:38:45 +00:00
|
|
|
SubmissionDir = path.Clean(SubmissionDir)
|
2016-12-30 11:45:14 +00:00
|
|
|
TeamsDir = path.Clean(TeamsDir)
|
2016-01-13 19:38:45 +00:00
|
|
|
|
2016-01-14 17:17:35 +00:00
|
|
|
rand.Seed(time.Now().UnixNano())
|
|
|
|
|
2016-01-13 19:38:45 +00:00
|
|
|
log.Println("Creating submission directory...")
|
|
|
|
if _, err := os.Stat(SubmissionDir); os.IsNotExist(err) {
|
|
|
|
if err := os.MkdirAll(SubmissionDir, 0777); err != nil {
|
|
|
|
log.Fatal("Unable to create submission directory: ", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
log.Println("Opening DB...")
|
2017-10-17 04:47:10 +00:00
|
|
|
if err := fic.DBInit(*dsn); err != nil {
|
2016-01-13 19:38:45 +00:00
|
|
|
log.Fatal("Cannot open the database: ", err)
|
|
|
|
}
|
|
|
|
defer fic.DBClose()
|
|
|
|
|
2016-12-30 11:45:14 +00:00
|
|
|
// Load configuration
|
2017-11-25 15:05:03 +00:00
|
|
|
settings.LoadAndWatchSettings(path.Join(settings.SettingsDir, settings.SettingsFile), reloadSettings)
|
2016-12-30 11:45:14 +00:00
|
|
|
|
2016-01-13 19:38:45 +00:00
|
|
|
log.Println("Registering directory events...")
|
2016-12-08 08:14:42 +00:00
|
|
|
watcher, err := fsnotify.NewWatcher()
|
2016-01-13 19:38:45 +00:00
|
|
|
if err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
2016-12-08 08:14:42 +00:00
|
|
|
defer watcher.Close()
|
2016-01-13 19:38:45 +00:00
|
|
|
|
|
|
|
if err := watchsubdir(watcher, SubmissionDir); err != nil {
|
|
|
|
log.Fatal(err)
|
|
|
|
}
|
|
|
|
|
2017-11-22 01:57:24 +00:00
|
|
|
watchedNotify := fsnotify.Create
|
|
|
|
|
2016-01-13 19:38:45 +00:00
|
|
|
for {
|
|
|
|
select {
|
2016-12-08 08:14:42 +00:00
|
|
|
case ev := <-watcher.Events:
|
2018-01-21 13:03:55 +00:00
|
|
|
if d, err := os.Lstat(ev.Name); err == nil && ev.Op & fsnotify.Create == fsnotify.Create && d.Mode().IsDir() && d.Mode() & os.ModeSymlink == 0 && d.Name() != ".tmp" {
|
2016-01-13 19:38:45 +00:00
|
|
|
// Register new subdirectory
|
2017-04-05 00:02:10 +00:00
|
|
|
if err := watchsubdir(watcher, ev.Name); err != nil {
|
|
|
|
log.Println(err)
|
2016-01-13 19:38:45 +00:00
|
|
|
}
|
2018-01-21 13:03:55 +00:00
|
|
|
} else if ev.Op & watchedNotify == watchedNotify && d.Mode().IsRegular() {
|
2018-01-16 22:59:51 +00:00
|
|
|
if *debugINotify {
|
|
|
|
log.Println("Treating event:", ev, "for", ev.Name)
|
|
|
|
}
|
2016-01-23 11:28:31 +00:00
|
|
|
go treat(ev.Name)
|
2017-11-22 01:57:24 +00:00
|
|
|
} else if ev.Op & fsnotify.Write == fsnotify.Write {
|
|
|
|
log.Println("FSNOTIFY WRITE SEEN. Prefer looking at them, as it appears files are not atomically moved.")
|
|
|
|
watchedNotify = fsnotify.Write
|
2017-11-22 01:39:52 +00:00
|
|
|
} else if *debugINotify {
|
|
|
|
log.Println("Skipped event:", ev, "for", ev.Name)
|
2016-01-13 19:38:45 +00:00
|
|
|
}
|
2016-12-08 08:14:42 +00:00
|
|
|
case err := <-watcher.Errors:
|
2016-01-13 19:38:45 +00:00
|
|
|
log.Println("error:", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-01-23 11:28:31 +00:00
|
|
|
|
|
|
|
func treat(raw_path string) {
|
|
|
|
// Extract
|
|
|
|
spath := strings.Split(strings.TrimPrefix(raw_path, SubmissionDir), "/")
|
|
|
|
|
2016-03-06 17:59:33 +00:00
|
|
|
if len(spath) == 3 {
|
|
|
|
if spath[1] == "_registration" {
|
2017-12-21 21:18:18 +00:00
|
|
|
treatRegistration(raw_path, spath[2])
|
2018-01-21 13:18:26 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
var team fic.Team
|
|
|
|
|
|
|
|
if strings.HasPrefix(spath[1], "_AUTH_ID_") {
|
|
|
|
if serial, err := strconv.ParseInt(strings.TrimPrefix(spath[1], "_AUTH_ID_"), 16, 64); err != nil {
|
|
|
|
log.Println("[ERR]", err)
|
|
|
|
return
|
|
|
|
} else if team, err = fic.GetTeamBySerial(serial); err != nil {
|
|
|
|
log.Println("[ERR]", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
} else if teamid, err := strconv.ParseInt(spath[1], 10, 64); err != nil {
|
2018-01-21 13:07:44 +00:00
|
|
|
log.Println("[ERR]", err)
|
2018-01-21 13:18:26 +00:00
|
|
|
return
|
|
|
|
} else if team, err = fic.GetTeam(teamid); err != nil {
|
2016-03-06 17:59:33 +00:00
|
|
|
log.Println("[ERR]", err)
|
2018-01-21 13:18:26 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
switch spath[2] {
|
|
|
|
case "name":
|
2016-03-06 17:59:33 +00:00
|
|
|
treatRename(raw_path, team)
|
2018-01-21 13:18:26 +00:00
|
|
|
case "hint":
|
2016-12-09 10:49:29 +00:00
|
|
|
treatOpeningHint(raw_path, team)
|
2018-01-21 13:18:26 +00:00
|
|
|
default:
|
2016-03-06 17:59:33 +00:00
|
|
|
treatSubmission(raw_path, team, spath[2])
|
|
|
|
}
|
2016-01-23 11:28:31 +00:00
|
|
|
} else {
|
|
|
|
log.Println("Invalid new file:", raw_path)
|
|
|
|
}
|
|
|
|
}
|