Adds prelim match stats tracking

Author: denverquane

discord/discordGameState.go

@@ -53,6 +53,11 @@ type DiscordGameState struct {
 	Running    bool `json:"running"`
 	Subscribed bool `json:"subscribed"`
 
+	MatchID        int64             `json:"matchID"`
+	MatchStartUnix int64             `json:"matchStartUnix"`
+	Winners        []game.GameWinner `json:"winners"`
+	GameResult     game.GameResult   `json:"gameResult"`
+
 	UserData UserDataSet     `json:"userData"`
 	Tracking TrackingChannel `json:"tracking"`
 
@@ -62,23 +67,21 @@ type DiscordGameState struct {
 }
 
 func NewDiscordGameState(guildID string) *DiscordGameState {
-	return &DiscordGameState{
-		GuildID:      guildID,
-		ConnectCode:  "",
-		Linked:       false,
-		Running:      false,
-		Subscribed:   false,
-		UserData:     UserDataSet{},
-		Tracking:     TrackingChannel{},
-		GameStateMsg: MakeGameStateMessage(),
-		AmongUsData:  game.NewAmongUsData(),
-	}
+	dgs := DiscordGameState{GuildID: guildID}
+	dgs.Reset()
+	return &dgs
 }
 
 func (dgs *DiscordGameState) Reset() {
+	//Explicitly does not reset the GuildID!
 	dgs.ConnectCode = ""
 	dgs.Linked = false
 	dgs.Running = false
+	dgs.Subscribed = false
+	dgs.MatchID = -1
+	dgs.MatchStartUnix = -1
+	dgs.Winners = []game.GameWinner{}
+	dgs.GameResult = -1
 	dgs.UserData = map[string]UserData{}
 	dgs.Tracking = TrackingChannel{}
 	dgs.GameStateMsg = MakeGameStateMessage()

discord/eventHandler.go

@@ -239,6 +239,22 @@ func (bot *Bot) processTransition(phase game.Phase, dgsRequest GameStateRequest)
 		lock.Release(ctx)
 		return
 	}
+	//if we started a new game
+	if oldPhase == game.LOBBY && phase == game.TASKS {
+		matchID := bot.RedisInterface.GetAndIncrementMatchID()
+		matchStart := time.Now().Unix()
+		dgs.MatchStartUnix = matchStart
+		dgs.MatchID = matchID
+		log.Printf("New match has begun. ID %d and starttime %d\n", matchID, matchStart)
+		//if we went to lobby from anywhere else but the menu, assume the game is over
+	} else if phase == game.LOBBY && oldPhase != game.MENU {
+		//TODO process the game's completion and send to Postgres
+		//only process games that actually receive the end-game event from the capture! Might need to start a worker
+		//to listen for this
+
+		dgs.MatchID = -1
+		dgs.MatchStartUnix = -1
+	}
 
 	bot.RedisInterface.SetDiscordGameState(dgs, lock)
 	switch phase {

discord/redis.go

@@ -67,6 +67,18 @@ func cacheHash(guildID string) string {
 	return "automuteus:discord:" + guildID + ":cache"
 }
 
+func matchIDKey() string {
+	return "automuteus:match:counter"
+}
+
+func (redisInterface *RedisInterface) GetAndIncrementMatchID() int64 {
+	num, err := redisInterface.client.Incr(ctx, matchIDKey()).Result()
+	if err != nil {
+		log.Println(err)
+	}
+	return num
+}
+
 func (redisInterface *RedisInterface) SetVersion(version string) {
 	err := redisInterface.client.Set(ctx, versionKey(), version, 0).Err()
 	if err != nil {

game/state.go

@@ -32,6 +32,21 @@ const (
 	EXILED       PlayerAction = iota
 )
 
+type GameResult int
+
+const (
+	HumansByVote GameResult = iota
+	HumansByTask
+	ImpostorByVote
+	ImpostorByKill
+	ImpostorBySabotage
+	ImpostorDisconnect
+	HumansDisconnect
+)
+
+type GameWinner struct {
+}
+
 type PhaseNameString string
 
 // PhaseNames for lowercase, possibly for translation if needed