diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/challenge.go | 5 | ||||
| -rw-r--r-- | src/seed.go | 17 |
2 files changed, 19 insertions, 3 deletions
diff --git a/src/challenge.go b/src/challenge.go index 1d0a210..3fb26d7 100644 --- a/src/challenge.go +++ b/src/challenge.go @@ -10,6 +10,7 @@ type Challenge struct { FlagTries uint Container string // this could, but is not required as well Category string + Points int } // TODO: The name "Stripped" is a bit inaccurate now. Rename. @@ -21,12 +22,14 @@ type StrippedChallenge struct { FlagTries uint `json:"flagTries"` ContainsLaunchable bool `json:"ContainsLaunchable"` IPAddress string `json:"IPAddress"` + Points int `json:"points"` } type StatsStrippedChallenge struct { Name string `json:"name"` FoundFlag int64 `json:"foundFlag"` FlagTries uint `json:"flagTries"` + Points int `json:"points"` } func stripChallenge(c Challenge) (StrippedChallenge) { @@ -38,6 +41,7 @@ func stripChallenge(c Challenge) (StrippedChallenge) { FlagTries: c.FlagTries, ContainsLaunchable: c.Container != "", IPAddress: getAddressForChallengeContainer(c.Container), + Points: c.Points, } } @@ -46,5 +50,6 @@ func stripChallengeForStatistics(c Challenge) (StatsStrippedChallenge) { Name: c.Name, FoundFlag: c.FoundFlag.Unix(), FlagTries: c.FlagTries, + Points: c.Points, } } diff --git a/src/seed.go b/src/seed.go index 805ab20..d876940 100644 --- a/src/seed.go +++ b/src/seed.go @@ -6,6 +6,7 @@ import ( "fmt" "io/ioutil" "log" + "strconv" "time" "github.com/pkg/errors" @@ -51,8 +52,10 @@ func readSeedFile(path string) ([]Challenge, error) { flag, flagOK := challenge["flag"].(string) cont, contOK := challenge["container"].(string) category, categoryOK := challenge["category"].(string) + pointsStr, pointsOK := challenge["points"].(string) + points, _ := strconv.Atoi(pointsStr) - if nameOK && descOK && flagOK && contOK && categoryOK { + if nameOK && descOK && flagOK && contOK && categoryOK && pointsOK { tmpChallenges = append(tmpChallenges, Challenge{ Name: name, Description: desc, @@ -61,9 +64,10 @@ func readSeedFile(path string) ([]Challenge, error) { FlagTries: 0, Container: cont, Category: category, + Points: points, }) } else { - log.Printf("Ignoring challenge at position %d: Not all values are parseable (name: %b, desc: %b, flag: %b, container: %b, category: %b", index, nameOK, descOK, flagOK, contOK, categoryOK) + log.Printf("Ignoring challenge at position %d: Not all values are parseable (name: %b, desc: %b, flag: %b, container: %b, category: %b, points: %b", index, nameOK, descOK, flagOK, contOK, categoryOK, pointsOK) } } @@ -116,18 +120,25 @@ func generateJSONFromChallenges() (string, error) { func generateJSONFromChallengesForStats() (string, error) { // To include only required information for statistics, we need to strip the challenge var strippedChallenges []StatsStrippedChallenge + score := 0 categories := map[string]int{} for _, challenge := range challenges { // Append challenge to list strippedChallenges = append(strippedChallenges, stripChallengeForStatistics(challenge)) - + // Raise category counter categories[challenge.Category]++ + + // Raise score if flag was found + if challenge.FoundFlag.Unix() > 0 { + score = score + challenge.Points + } } marshalled, marshalError := json.Marshal(map[string]interface{}{ "user": user, "challenges": strippedChallenges, + "score": score, }) return string(marshalled), marshalError } |
