Add points per challenge and overall scoring

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
 }