From 0db2be86253a2bf7c0e30d6c3fa98e5e2431fc77 Mon Sep 17 00:00:00 2001
From: Aldin Duraki <aldin.duraki@gmail.com>
Date: Mon, 18 May 2020 02:00:13 +0200
Subject: [PATCH] Test: trying to add the TeamComponent to the new captured
 base  [unsafe-state]

---
 GoldWars/GoldWars/Entities/EntityManager.swift | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/GoldWars/GoldWars/Entities/EntityManager.swift b/GoldWars/GoldWars/Entities/EntityManager.swift
index 1dbda90..9ebfc7a 100644
--- a/GoldWars/GoldWars/Entities/EntityManager.swift
+++ b/GoldWars/GoldWars/Entities/EntityManager.swift
@@ -135,8 +135,18 @@ class EntityManager {
                 getOwnerBySnapBase = MatchmakingHelper.sharedInstance.getGKPlayerByUsername(displayName: snapBase.ownership!)
             }
             if getOwnerBySnapBase != nil {
+                base.changeOwnership = true
                 base.ownershipPlayer = getOwnerBySnapBase
+                entity.addComponent(TeamComponent(
+                        team: getTeamByPlayer(playerName: snapBase.ownership!),
+                        player: getOwnerBySnapBase!,
+                        position: (entity.component(ofType: DefaultBaseComponent.self)?.spriteNode.position)!
+                    )
+                )
+                print(entity)
+                print(getTeamByBase(base: base))
             }
+            
             print("nach updateSnap -> Entity \(base)")
         }
     }
@@ -189,6 +199,10 @@ class EntityManager {
         }
         return nil
     }
+    
+    func getTeamByPlayer(playerName: String) -> Team {
+        return entities.filter { $0 is Base && ($0 as! Base).component(ofType: TeamComponent.self)?.player.displayName == playerName }[0].component(ofType: TeamComponent.self)!.team
+    }
 
 	func getBasebyID(id: Int) -> Base?{
 		for entity in entities {