common: add function TeamState.Team()

Useful for when we have a TeamState but don't know which team's it is
from context.

Fixes #76

Author: markus-wa

common/common.go

@@ -105,6 +105,8 @@ func (b Bomb) Position() r3.Vector {
 
 // TeamState contains a team's ID, score, clan name & country flag.
 type TeamState struct {
+	team Team
+
 	// ID stays the same even after switching sides.
 	ID int
 
@@ -116,3 +118,12 @@ type TeamState struct {
 	// Watch out, in some demos this is upper-case and in some lower-case.
 	Flag string
 }
+
+// Team returns the team for which the TeamState contains data.
+func (ts TeamState) Team() Team {
+	return ts.team
+}
+
+func NewTeamState(team Team) TeamState {
+	return TeamState{team: team}
+}

common/common_test.go

@@ -38,3 +38,8 @@ func TestDemoHeader(t *testing.T) {
 	assert.Equal(t, float64(128.0), header.TickRate(), "TickRate should be 128")
 	assert.Equal(t, time.Second/128, header.TickTime(), "TickTime should be 1/128")
 }
+
+func TestTeamState(t *testing.T) {
+	assert.Equal(t, TeamTerrorists, NewTeamState(TeamTerrorists).Team())
+	assert.Equal(t, TeamCounterTerrorists, NewTeamState(TeamCounterTerrorists).Team())
+}

game_state.go

@@ -110,6 +110,8 @@ func newGameState() GameState {
 		grenadeProjectiles: make(map[int]*common.GrenadeProjectile),
 		infernos:           make(map[int]*common.Inferno),
 		entities:           make(map[int]*st.Entity),
+		tState:             common.NewTeamState(common.TeamTerrorists),
+		ctState:            common.NewTeamState(common.TeamCounterTerrorists),
 	}
 }
 

game_state_test.go

@@ -0,0 +1,16 @@
+package demoinfocs
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+
+	"github.com/markus-wa/demoinfocs-golang/common"
+)
+
+func TestNewGameState(t *testing.T) {
+	gs := newGameState()
+
+	assert.Equal(t, common.TeamTerrorists, gs.tState.Team())
+	assert.Equal(t, common.TeamCounterTerrorists, gs.ctState.Team())
+}