challenge-sync-airbus: Ready for 2024
This commit is contained in:
parent
5a6d9047c2
commit
ac966f9023
5 changed files with 273 additions and 115 deletions
|
|
@ -2,13 +2,16 @@ package main
|
|||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"path"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strconv"
|
||||
"strings"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
|
|
@ -27,6 +30,8 @@ func main() {
|
|||
flag.BoolVar(&dryRun, "dry-run", dryRun, "Don't perform any write action, just display")
|
||||
flag.BoolVar(&skipInitialSync, "skipinitialsync", skipInitialSync, "Skip the initial synchronization")
|
||||
flag.BoolVar(&noValidateChallenge, "no-validate-challenge", noValidateChallenge, "Consider challenge validation as a standard award (if each exercice hasn't been imported on their side)")
|
||||
skipVerify := flag.Bool("skip-tls-verify", false, "Allow not verified certificates (INSECURE!)")
|
||||
tsFromFile := flag.Bool("timestamp-from-file", false, "Load timestamp matching for score from filesystem instead of the API")
|
||||
daemon := flag.Bool("watch", false, "Enable daemon mode by watching the directory")
|
||||
tspath := flag.String("timestamp-file", "./REMOTE/timestamp", "Path to the file storing the last timestamp")
|
||||
exercicespath := flag.String("exercices-file", "./REMOTE/exercices-bindings.json", "Path to the file containing the ID bindings")
|
||||
|
|
@ -34,12 +39,20 @@ func main() {
|
|||
flag.Parse()
|
||||
|
||||
api := AirbusAPI{
|
||||
BaseURL: "https://portal.european-cybercup.lan/api",
|
||||
BaseURL: "https://portal.european-cybercup.lan/api",
|
||||
InsecureSkipVerify: *skipVerify,
|
||||
}
|
||||
|
||||
if v, exists := os.LookupEnv("AIRBUS_BASEURL"); exists {
|
||||
api.BaseURL = v
|
||||
}
|
||||
if v, exists := os.LookupEnv("AIRBUS_SKIP_TLS_VERIFY"); exists {
|
||||
var err error
|
||||
api.InsecureSkipVerify, err = strconv.ParseBool(v)
|
||||
if err != nil {
|
||||
log.Fatal("Unable to parse boolean value in AIRBUS_SKIP_TLS_VERIFY:", err)
|
||||
}
|
||||
}
|
||||
if v, exists := os.LookupEnv("AIRBUS_TOKEN"); exists {
|
||||
api.Token = v
|
||||
}
|
||||
|
|
@ -68,39 +81,156 @@ func main() {
|
|||
} else {
|
||||
log.Println("Session ID discovered: ", api.SessionID)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
sessions, err := api.GetSessions()
|
||||
if err != nil {
|
||||
log.Fatal("Unable to retrieve session (check your credentials!): ", err)
|
||||
}
|
||||
|
||||
if v, exists := os.LookupEnv("AIRBUS_SESSIONUUID"); exists {
|
||||
api.SessionUUID = v
|
||||
log.Println("Please define your AIRBUS_SESSIONID or AIRBUS_SESSION_NAME.")
|
||||
log.Println("Existing sessions are:")
|
||||
for _, session := range sessions {
|
||||
log.Printf(" - %d: %q", session.ID, session.Name)
|
||||
}
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
log.SetPrefix("[challenge-sync-airbus] ")
|
||||
|
||||
if flag.NArg() > 0 {
|
||||
args := flag.Args()
|
||||
switch args[0] {
|
||||
case "list":
|
||||
teams, err := api.GetTeams()
|
||||
if err != nil {
|
||||
log.Println("Unable to retrieve teams:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
fmt.Println("## Airbus' registered teams:")
|
||||
fmt.Println("-------------------------------------------")
|
||||
fmt.Println(" ID | Name | Nb. | Score | Rank")
|
||||
fmt.Println("-------------------------------------------")
|
||||
for _, team := range teams {
|
||||
fmt.Printf("% 2d | % 15s | % 3d | % 5d | % 3d\n", team.ID, team.Name, len(team.Members), team.Score, team.Rank)
|
||||
}
|
||||
case "rank":
|
||||
teams, err := api.GetTeams()
|
||||
if err != nil {
|
||||
log.Println("Unable to retrieve teams:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
ranking := []*AirbusTeam{}
|
||||
for _, team := range teams {
|
||||
tmp := team
|
||||
ranking = append(ranking, &tmp)
|
||||
}
|
||||
|
||||
sort.Sort(sort.Reverse(ByScore(ranking)))
|
||||
|
||||
fmt.Println("## Airbus' ranking:")
|
||||
fmt.Println("-------------------------------------")
|
||||
fmt.Println(" Rank | Name | ID | Score")
|
||||
fmt.Println("-------------------------------------")
|
||||
for _, team := range ranking {
|
||||
fmt.Printf("% 5d | % 15s |% 3d | % 5d\n", team.Rank, team.Name, team.ID, team.Score)
|
||||
}
|
||||
case "get":
|
||||
teams, err := api.GetTeams()
|
||||
if err != nil {
|
||||
log.Println("Unable to retrieve teams:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
teamid, err := strconv.ParseInt(args[1], 10, 64)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
for _, team := range teams {
|
||||
if team.ID == teamid {
|
||||
fmt.Printf("## Airbus' registered team %d:\n\nID: %d\nName: %s\nScore: %d\nRank: %d\nMembers:\n", teamid, team.ID, team.Name, team.Score, team.Rank)
|
||||
for _, member := range team.Members {
|
||||
fmt.Printf(" - ID: %d\n Name: %s\n Nickname: %s\n E-mail: %s\n", member.ID, member.Name, member.Nickname, member.EMail)
|
||||
}
|
||||
os.Exit(0)
|
||||
}
|
||||
}
|
||||
fmt.Printf("Team %d not found. Use 'list' to view all existing teams\n", teamid)
|
||||
case "award":
|
||||
if len(args) < 3 {
|
||||
fmt.Println("award <TEAM_ID> <VALUE> <MESSAGE>")
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
teams, err := api.GetTeams()
|
||||
if err != nil {
|
||||
log.Println("Unable to retrieve teams:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
teamid, err := strconv.ParseInt(args[1], 10, 64)
|
||||
if err != nil {
|
||||
log.Println("Invalid team id", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
value, err := strconv.ParseInt(args[2], 10, 64)
|
||||
if err != nil {
|
||||
log.Println("Invalid award value:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
for _, team := range teams {
|
||||
if team.ID == teamid {
|
||||
err = api.AwardUser(&team, value, strings.Join(args[3:], " "))
|
||||
if err != nil {
|
||||
log.Println("Unable to award team:", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
fmt.Println("Team awarded")
|
||||
os.Exit(0)
|
||||
}
|
||||
}
|
||||
fmt.Printf("Team %d not found. Use 'list' to view all existing teams\n", teamid)
|
||||
}
|
||||
|
||||
os.Exit(0)
|
||||
}
|
||||
|
||||
var err error
|
||||
TeamsDir = path.Clean(TeamsDir)
|
||||
|
||||
// Load the timestamp
|
||||
ts, err := loadTS(*tspath)
|
||||
if err != nil {
|
||||
log.Fatal("Unable to open timestamp file: ", err.Error())
|
||||
w := Walker{
|
||||
API: api,
|
||||
Coeff: *coeff,
|
||||
}
|
||||
|
||||
// Load teams.json
|
||||
teamsbindings, err := getTeams(filepath.Join(TeamsDir, "teams.json"))
|
||||
w.Teams, err = getTeams(filepath.Join(TeamsDir, "teams.json"))
|
||||
if err != nil {
|
||||
log.Fatal("Unable to open teams bindings file: ", err.Error())
|
||||
}
|
||||
log.Println("Team bindings loaded: ", len(teamsbindings))
|
||||
log.Println("Team bindings loaded: ", len(w.Teams))
|
||||
|
||||
w := Walker{
|
||||
LastSync: ts,
|
||||
Teams: teamsbindings,
|
||||
API: api,
|
||||
Coeff: *coeff,
|
||||
}
|
||||
// Fetch teams from Airbus API
|
||||
if err = w.fetchTeams(); err != nil {
|
||||
log.Fatal("Unable to fetch Airbus teams: ", err.Error())
|
||||
}
|
||||
|
||||
// Load the timestamp
|
||||
if *tsFromFile {
|
||||
w.LastSync, err = loadTS(*tspath)
|
||||
} else {
|
||||
w.LastSync, err = loadTSFromAPI(w.TeamBindings)
|
||||
}
|
||||
if err != nil {
|
||||
log.Fatal("Unable to open timestamp file: ", err.Error())
|
||||
}
|
||||
|
||||
if !noValidateChallenge {
|
||||
// Load exercices bindings
|
||||
w.Exercices, err = ReadExercicesBindings(*exercicespath)
|
||||
|
|
@ -110,6 +240,7 @@ func main() {
|
|||
}
|
||||
|
||||
if !skipInitialSync {
|
||||
log.Println("Doing initial score balance")
|
||||
err = w.BalanceScores()
|
||||
if err != nil {
|
||||
log.Println("Something goes wrong during score balance: ", err.Error())
|
||||
|
|
@ -120,9 +251,9 @@ func main() {
|
|||
if err != nil {
|
||||
log.Fatal("Unable to save timestamp file: ", err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
log.Println("initial sync done")
|
||||
log.Println("initial sync done")
|
||||
}
|
||||
|
||||
if daemon != nil && *daemon {
|
||||
// Watch teams.json and scores.json
|
||||
|
|
@ -145,7 +276,12 @@ func main() {
|
|||
interrupt3 := make(chan os.Signal, 1)
|
||||
signal.Notify(interrupt3, syscall.SIGUSR2)
|
||||
|
||||
ticker := time.NewTicker(5 * time.Second)
|
||||
var ticker *time.Ticker
|
||||
if *tsFromFile {
|
||||
ticker = time.NewTicker(5 * time.Second)
|
||||
} else {
|
||||
ticker = time.NewTicker(5 * time.Minute)
|
||||
}
|
||||
|
||||
watchedNotify := fsnotify.Create
|
||||
|
||||
|
|
@ -156,17 +292,27 @@ func main() {
|
|||
teamsbindings, err := getTeams(filepath.Join(TeamsDir, "teams.json"))
|
||||
if err != nil {
|
||||
log.Println("Unable to open teams bindings file: ", err.Error())
|
||||
return
|
||||
} else {
|
||||
w.Teams = teamsbindings
|
||||
}
|
||||
w.Teams = teamsbindings
|
||||
|
||||
if err = w.fetchTeams(); err != nil {
|
||||
log.Fatal("Unable to fetch teams: ", err.Error())
|
||||
log.Println("Unable to fetch teams: ", err.Error())
|
||||
}
|
||||
|
||||
if !*tsFromFile {
|
||||
ts, err := loadTSFromAPI(w.TeamBindings)
|
||||
if err != nil {
|
||||
log.Println("Unable to refresh timestamp: ", err.Error())
|
||||
} else {
|
||||
w.LastSync = ts
|
||||
}
|
||||
}
|
||||
|
||||
// save current timestamp for teams
|
||||
err = saveTS(*tspath, w.LastSync)
|
||||
if err != nil {
|
||||
log.Fatal("Unable to save timestamp file: ", err.Error())
|
||||
log.Println("Unable to save timestamp file: ", err.Error())
|
||||
}
|
||||
log.Println("SIGHUP treated.")
|
||||
case <-interrupt2:
|
||||
|
|
@ -180,7 +326,7 @@ func main() {
|
|||
// save current timestamp for teams
|
||||
err = saveTS(*tspath, w.LastSync)
|
||||
if err != nil {
|
||||
log.Fatal("Unable to save timestamp file: ", err.Error())
|
||||
log.Println("Unable to save timestamp file: ", err.Error())
|
||||
}
|
||||
log.Println("SIGUSR1 treated.")
|
||||
case <-interrupt3:
|
||||
|
|
@ -189,17 +335,40 @@ func main() {
|
|||
if err != nil {
|
||||
log.Println("Unable to open teams bindings file: ", err.Error())
|
||||
return
|
||||
}
|
||||
w.Teams = teamsbindings
|
||||
if err = w.fetchTeams(); err != nil {
|
||||
log.Fatal("Unable to fetch teams: ", err.Error())
|
||||
} else {
|
||||
w.Teams = teamsbindings
|
||||
}
|
||||
|
||||
// FIXME
|
||||
if err = w.fetchTeams(); err != nil {
|
||||
log.Println("Unable to fetch teams: ", err.Error())
|
||||
return
|
||||
}
|
||||
|
||||
if !*tsFromFile {
|
||||
ts, err := loadTSFromAPI(w.TeamBindings)
|
||||
if err != nil {
|
||||
log.Println("Unable to refresh timestamp: ", err.Error())
|
||||
return
|
||||
} else {
|
||||
w.LastSync = ts
|
||||
}
|
||||
}
|
||||
|
||||
err = w.BalanceScores()
|
||||
if err != nil {
|
||||
log.Println("Unable to balance scores: ", err.Error())
|
||||
}
|
||||
|
||||
log.Println("SIGUSR2 treated.")
|
||||
case ev := <-watcher.Events:
|
||||
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" {
|
||||
d, err := os.Lstat(ev.Name)
|
||||
|
||||
if err == nil && watchedNotify == fsnotify.Create && 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
|
||||
}
|
||||
|
||||
if err == nil && ev.Op&fsnotify.Create == fsnotify.Create && d.Mode().IsDir() && d.Mode()&os.ModeSymlink == 0 && d.Name() != ".tmp" {
|
||||
// Register new subdirectory
|
||||
if err := watchsubdir(watcher, ev.Name); err != nil {
|
||||
log.Println(err)
|
||||
|
|
@ -208,22 +377,24 @@ func main() {
|
|||
if *debugINotify {
|
||||
log.Println("Treating event:", ev, "for", ev.Name)
|
||||
}
|
||||
|
||||
if watchedNotify == fsnotify.Write {
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
}
|
||||
|
||||
if filepath.Base(ev.Name) == "scores.json" {
|
||||
go w.treat(ev.Name)
|
||||
} else if filepath.Base(ev.Name) == "teams.json" {
|
||||
teamsbindings, err := getTeams(filepath.Join(TeamsDir, "teams.json"))
|
||||
if err != nil {
|
||||
log.Println("Unable to open teams bindings file: ", err.Error())
|
||||
return
|
||||
} else {
|
||||
w.Teams = teamsbindings
|
||||
}
|
||||
w.Teams = teamsbindings
|
||||
if err = w.fetchTeams(); err != nil {
|
||||
log.Fatal("Unable to fetch teams: ", err.Error())
|
||||
log.Println("Unable to fetch teams: ", err.Error())
|
||||
}
|
||||
}
|
||||
} else if err == nil && 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
|
||||
} else if err == nil && *debugINotify {
|
||||
log.Println("Skipped event:", ev, "for", ev.Name)
|
||||
}
|
||||
|
|
@ -233,7 +404,7 @@ func main() {
|
|||
// save current timestamp for teams
|
||||
err = saveTS(*tspath, w.LastSync)
|
||||
if err != nil {
|
||||
log.Fatal("Unable to save timestamp file: ", err.Error())
|
||||
log.Println("Unable to save timestamp file: ", err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Reference in a new issue