From 5b88a91903ba2d139aea364ab7f4ccb68c1f4bce Mon Sep 17 00:00:00 2001
From: Ronnie Friman <L4168@student.jamk.fi>
Date: Tue, 16 Jul 2019 19:53:58 +0300
Subject: [PATCH] add icon based on group class

---
 src/tracking/tracking.service.ts | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/tracking/tracking.service.ts b/src/tracking/tracking.service.ts
index ff97aae..58d64b9 100644
--- a/src/tracking/tracking.service.ts
+++ b/src/tracking/tracking.service.ts
@@ -15,6 +15,13 @@ export class TrackingService {
     private gamepersonrepository: Repository<Game_PersonEntity>,
   ) {}
 
+  private icons = {
+    null: 'infantry.svg',
+    infantry: 'infantry.svg',
+    recon: 'recon.svg',
+    mechanized: 'mechanized.svg',
+  };
+
   async trackLocation(
     gameperson: Game_PersonEntity,
     gameId,
@@ -37,7 +44,7 @@ export class TrackingService {
       trackdata['time'] = Date.now();
       // initialize data
       trackedperson = await this.trackingrepository.create(trackedperson);
-      trackedperson.icon = 'infantry.svg';
+      trackedperson.icon = this.icons[gameperson.group.class];
       trackedperson.data = [trackdata];
       trackedperson.faction = gameperson.faction;
       trackedperson.game = gameId;
-- 
GitLab