protobuf: equipment mapping

markus-wa · Jun 8, 2020 · 066b9e0 · 066b9e0
1 parent 41c04d8
commit 066b9e0
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 0 deletions.
diff --git a/examples/minimal.pb b/examples/minimal.pb
diff --git a/protobuf/marshal.go b/protobuf/marshal.go
@@ -84,8 +84,21 @@ func mapToEntityUpdates(entityUpdates []rep.EntityUpdate) []*gen.Replay_Snapshot
 			Team:          mapToTeam(u.Team),
 			HasHelmet:     u.HasHelmet,
 			HasDefuseKit:  u.HasDefuseKit,
+			Equipment:     mapToEquipment(u.Equipment),
 		})
 	}
+
+	return result
+}
+
+func mapToEquipment(equipment []rep.EntityEquipment) []*gen.Replay_Snapshot_EntityEquipment {
+	result := make([]*gen.Replay_Snapshot_EntityEquipment, 0)
+	for _, eq := range equipment {
+		result = append(result, &gen.Replay_Snapshot_EntityEquipment{
+			Type: int32(eq.Type),
+		})
+	}
+
 	return result
 }
 

diff --git a/protobuf/unmarshal.go b/protobuf/unmarshal.go
@@ -108,6 +108,22 @@ func mapFromEntityUpdates(entityUpdates []*gen.Replay_Snapshot_EntityUpdate) []r
 			Team:          mapFromTeam(u.Team),
 			HasHelmet:     u.HasHelmet,
 			HasDefuseKit:  u.HasDefuseKit,
+			Equipment:     mapFromEquipment(u.Equipment),
+		}
+	}
+
+	return result
+}
+
+func mapFromEquipment(equipment []*gen.Replay_Snapshot_EntityEquipment) []rep.EntityEquipment {
+	if equipment == nil {
+		return nil
+	}
+
+	result := make([]rep.EntityEquipment, len(equipment))
+	for i, eq := range equipment {
+		result[i] = rep.EntityEquipment{
+			Type: int(eq.Type),
 		}
 	}