server/backend/main.go

package main

import (
	"flag"
	"fmt"
	"io/ioutil"
	"log"
	"math/rand"
	"os"
	"path"
	"strings"
	"time"

	"golang.org/x/exp/inotify"
	"srs.epita.fr/fic-server/libfic"
)

var TeamsDir string
var SubmissionDir string

func watchsubdir(watcher *inotify.Watcher, pathname string) error {
	log.Println("Watch new directory:", pathname)
	if err := watcher.AddWatch(pathname, inotify.IN_CLOSE_WRITE|inotify.IN_CREATE); err != nil {
		return err
	}

	if ds, err := ioutil.ReadDir(pathname); err != nil {
		return err
	} else {
		for _, d := range ds {
			p := path.Join(pathname, d.Name())
			if d.IsDir() {
				if err := watchsubdir(watcher, p); err != nil {
					return err
				}
			} else if d.Mode().IsRegular() {
				go treat(p)
			}
		}
		return nil
	}
}

func main() {
	var dsn = flag.String("dsn", "fic:fic@/fic", "DSN to connect to the MySQL server")
	flag.StringVar(&SubmissionDir, "submission", "./submissions", "Base directory where save submissions")
	flag.StringVar(&TeamsDir, "teams", "../TEAMS", "Base directory where save teams JSON files")
	var skipFullGeneration = flag.Bool("skipFullGeneration", false, "Skip initial full generation (safe to skip after start)")
	flag.Parse()

	log.Prefix("[backend] ")

	SubmissionDir = path.Clean(SubmissionDir)

	rand.Seed(time.Now().UnixNano())

	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...")
	if err := fic.DBInit(fmt.Sprintf("%s?parseTime=true", *dsn)); err != nil {
		log.Fatal("Cannot open the database: ", err)
	}
	defer fic.DBClose()

	log.Println("Registering directory events...")
	watcher, err := inotify.NewWatcher()
	if err != nil {
		log.Fatal(err)
	}

	if err := watchsubdir(watcher, SubmissionDir); err != nil {
		log.Fatal(err)
	}

	if !*skipFullGeneration {
		log.Println("Generating files...")
		go genAll()
	}

	for {
		select {
		case ev := <-watcher.Event:
			if ev.Mask&inotify.IN_CREATE == inotify.IN_CREATE {
				// Register new subdirectory
				if d, err := os.Stat(ev.Name); err == nil && d.IsDir() {
					if err := watchsubdir(watcher, ev.Name); err != nil {
						log.Println(err)
					}
				}
			} else if ev.Mask&inotify.IN_CLOSE_WRITE == inotify.IN_CLOSE_WRITE {
				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) == 3 {
		if spath[1] == "_registration" {
			treatRegistration(raw_path)
		} else if team, err := fic.GetTeamByInitialName(spath[1]); err != nil {
			log.Println("[ERR]", err)
		} else if spath[2] == "name" {
			treatRename(raw_path, team)
		} else {
			treatSubmission(raw_path, team, spath[2])
		}
	} else {
		log.Println("Invalid new file:", raw_path)
	}
}