feat(equipment): Add equipment to entity snapshots (#39)

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

csminify.go

@@ -9,6 +9,7 @@ import (
 
 	r3 "github.com/golang/geo/r3"
 	dem "github.com/markus-wa/demoinfocs-golang/v2/pkg/demoinfocs"
+	common "github.com/markus-wa/demoinfocs-golang/v2/pkg/demoinfocs/common"
 	events "github.com/markus-wa/demoinfocs-golang/v2/pkg/demoinfocs/events"
 
 	rep "github.com/markus-wa/cs-demo-minifier/replay"
@@ -163,6 +164,7 @@ func (m *minifier) snapshot() rep.Snapshot {
 				AngleY:        int(pl.ViewDirectionY()),
 				HasHelmet:     pl.HasHelmet(),
 				HasDefuseKit:  pl.HasDefuseKit(),
+				Equipment:     toEntityEquipment(pl.Weapons()),
 			}
 
 			// FIXME: Smoothify Positions
@@ -201,3 +203,15 @@ func r3VectorToPoint(v r3.Vector) rep.Point {
 func roundTo(x, precision float64) float64 {
 	return math.Round(x/precision) * precision
 }
+
+func toEntityEquipment(eq []*common.Equipment) []rep.EntityEquipment {
+	var equipmentForPlayer = make([]rep.EntityEquipment, 0, len(eq))
+
+	for _, equipment := range eq {
+		equipmentForPlayer = append(equipmentForPlayer, rep.EntityEquipment{
+			Type: int(equipment.Type),
+		})
+	}
+
+	return equipmentForPlayer
+}