add color function for common.Player (#250)

see #248

Co-authored-by: Markus <[email protected]>

Author: Julien2313

pkg/demoinfocs/common/common.go

@@ -239,3 +239,33 @@ func ConvertSteamID32To64(steamID32 uint32) uint64 {
 func ConvertSteamID64To32(steamID64 uint64) uint32 {
 	return uint32(steamID64 - steamID64IndividualIdentifier)
 }
+
+// Color is the type for the various colors constants.
+type Color int
+
+// Color constants give information about which color a player has.
+const (
+	Grey Color = iota - 1
+	Yellow
+	Purple
+	Green
+	Blue
+	Orange
+)
+
+var strColors = map[Color]string{
+	Grey:   "Grey",
+	Yellow: "Yellow",
+	Purple: "Purple",
+	Green:  "Green",
+	Blue:   "Blue",
+	Orange: "Orange",
+}
+
+func (c Color) String() string {
+	if _, exists := strColors[c]; !exists {
+		return "Unknown-Color"
+	}
+
+	return strColors[c]
+}

pkg/demoinfocs/common/player.go

@@ -376,6 +376,11 @@ func (p *Player) Score() int {
 	return getInt(p.ResourceEntity(), "m_iScore."+p.entityIDStr())
 }
 
+// Color returns the players color as shown on the match.
+func (p *Player) Color() Color {
+	return Color(getInt(p.ResourceEntity(), "m_iCompTeammateColor."+p.entityIDStr()))
+}
+
 // Kills returns the amount of kills the player has as shown on the scoreboard.
 func (p *Player) Kills() int {
 	return getInt(p.ResourceEntity(), "m_iKills."+p.entityIDStr())

pkg/demoinfocs/common/player_test.go

@@ -389,6 +389,12 @@ func TestPlayer_Score(t *testing.T) {
 	assert.Equal(t, 10, pl.Score())
 }
 
+func TestPlayer_Color(t *testing.T) {
+	pl := playerWithResourceProperty("m_iCompTeammateColor", st.PropertyValue{IntVal: int(Yellow)})
+
+	assert.Equal(t, Yellow, pl.Color())
+}
+
 func TestPlayer_Kills(t *testing.T) {
 	pl := playerWithResourceProperty("m_iKills", st.PropertyValue{IntVal: 5})