From 7ee50f5b8e4cc2ca88a532054e03d7a60f30000b Mon Sep 17 00:00:00 2001
From: ZMiguel Alves <zmiguel.alves@gmail.com>
Date: Thu, 7 Mar 2019 17:42:18 +0000
Subject: [PATCH] Rev 1.1 Teams

---
 api/Controllers/TeamsController.cs | 37 ++++++++++++++++++++++++------
 1 file changed, 30 insertions(+), 7 deletions(-)

diff --git a/api/Controllers/TeamsController.cs b/api/Controllers/TeamsController.cs
index 4aa50c2c..5734f578 100644
--- a/api/Controllers/TeamsController.cs
+++ b/api/Controllers/TeamsController.cs
@@ -96,14 +96,33 @@ namespace api.Controllers
 
           Team tEdit = await context.Teams.Include(t=>t.Membros).FirstOrDefaultAsync(t=>t.Id == MemberToAdd.id);
 
-          tEdit.NMembros++;
-          tEdit.Membros.Add(newMember);
+          List<Team> allTeams = await context.Teams.Include(t=>t.Membros).Include(t=>t.Cap).ToListAsync();
 
-          context.Update(tEdit);
+          var valido = true;
 
-          var result = context.SaveChanges();
-          
-          return StatusCode(201);
+          allTeams.ForEach(delegate(Team t){
+            if(newMember == t.Cap){
+              valido = false;
+            }
+            t.Membros.ForEach(delegate(User m){
+              if(newMember == m){
+                valido = false;
+              }
+            });
+          });
+
+          if(valido){
+            tEdit.NMembros++;
+            tEdit.Membros.Add(newMember);
+
+            context.Update(tEdit);
+
+            var result = context.SaveChanges();
+            
+            return StatusCode(201);
+          }
+
+          return StatusCode(403);
           
         }
 
@@ -156,7 +175,11 @@ namespace api.Controllers
 
           User rmMember = await context.Users.FirstOrDefaultAsync(u=>u.QRcode == MemberToRemove.UserToRemoveQR);
 
-          Team tEdit = await context.Teams.Include(t=>t.Membros).FirstOrDefaultAsync(t=>t.Id == MemberToRemove.TeamID);
+          Team tEdit = await context.Teams.Include(t=>t.Membros).Include(t=>t.Cap).FirstOrDefaultAsync(t=>t.Id == MemberToRemove.TeamID);
+
+          if(rmMember == tEdit.Cap){
+            return StatusCode(403);
+          }
 
           tEdit.NMembros--;
           tEdit.Membros.Remove(rmMember);