server/evdist/settings.go

101 lines
2.9 KiB
Go

package main
import (
"fmt"
"io"
"io/ioutil"
"log"
"os"
"path"
"strconv"
"strings"
"time"
"srs.epita.fr/fic-server/settings"
)
// NewDistList creates a distList from the given src directory
func NewSettingsDistList(src string) (*distList, error) {
list, err := settings.ListNextSettingsFiles()
if err != nil {
return nil, err
}
var dlist []DistEvent
for _, e := range list {
dlist = append(dlist, e)
}
return &distList{List: dlist, Timer: time.NewTimer(time.Minute)}, nil
}
func parseSettingsFilename(fname string) (int64, error) {
return strconv.ParseInt(strings.TrimSuffix(fname, ".json"), 10, 64)
}
func newSettingsFile(l *distList, raw_path string) {
bpath := path.Base(raw_path)
if bpath == "challenge.json" || bpath == "settings.json" {
log.Printf("Copying %s to SETTINGDIST...", bpath)
// Copy content through tmp file
fd, err := os.Open(raw_path)
if err != nil {
log.Printf("ERROR: Unable to open %s: %s", raw_path, err.Error())
return
}
defer fd.Close()
tmpfile, err := ioutil.TempFile(TmpSettingsDistDirectory, "")
if err != nil {
log.Printf("ERROR: Unable to create temporary file for %s: %s", bpath, err.Error())
return
}
_, err = io.Copy(tmpfile, fd)
tmpfile.Close()
if err != nil {
log.Printf("ERROR: Unable to copy content to temporary file (%s): %s", bpath, err.Error())
return
}
os.Chmod(tmpfile.Name(), 0644)
if err = os.Rename(tmpfile.Name(), path.Join(SettingsDistDir, bpath)); err != nil {
log.Println("ERROR: Unable to move file:", err)
return
}
} else if ts, err := parseSettingsFilename(bpath); err == nil {
activateTime := time.Unix(ts, 0)
log.Printf("Preparing %s: activation time at %s", bpath, activateTime)
l.AddEvent(&settings.NextSettingsFile{
Id: ts,
Date: activateTime,
})
} else {
log.Println("WARNING: Unknown file to treat: not a valid timestamp:", err.Error())
}
}
func treatSettingsFile(e DistEvent) {
v, err := settings.ReadNextSettingsFile(path.Join(settings.SettingsDir, fmt.Sprintf("%d.json", e.GetId())), e.GetId())
if err != nil {
log.Printf("Unable to read json: %s", err.Error())
} else if cur_settings, err := settings.ReadSettings(path.Join(settings.SettingsDir, settings.SettingsFile)); err != nil {
log.Printf("Unable to read settings.json: %s", err.Error())
} else {
cur_settings = settings.MergeSettings(*cur_settings, v.Values)
if err = settings.SaveSettings(path.Join(TmpSettingsDirectory, "settings.json"), cur_settings); err != nil {
log.Printf("Unable to save settings.json to tmp dir: %s", err.Error())
} else if err = os.Rename(path.Join(TmpSettingsDirectory, "settings.json"), path.Join(settings.SettingsDir, "settings.json")); err != nil {
log.Printf("Unable to move settings.json to dest dir: %s", err.Error())
} else if err = os.Remove(path.Join(settings.SettingsDir, fmt.Sprintf("%d.json", v.Id))); err != nil {
log.Printf("Unable to remove initial diff file (%d.json): %s", v.Id, err.Error())
}
}
}