From 03268a5575004e59662baf93ebfea908bbfccf87 Mon Sep 17 00:00:00 2001
From: Milinai <41624505+Milinai@users.noreply.github.com>
Date: Sat, 17 Nov 2018 18:57:10 +0100
Subject: [PATCH] Update ServerControl.java (#339)

* Update ServerControl.java

Hopefully made core fill for teams.

* Fixed compile errors and incorrect team handling
---
 .../anuke/mindustry/server/ServerControl.java   | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)

diff --git a/server/src/io/anuke/mindustry/server/ServerControl.java b/server/src/io/anuke/mindustry/server/ServerControl.java
index c0a43aae57..e4be13ed40 100644
--- a/server/src/io/anuke/mindustry/server/ServerControl.java
+++ b/server/src/io/anuke/mindustry/server/ServerControl.java
@@ -355,13 +355,20 @@ public class ServerControl extends Module{
                 err("Not playing. Host first.");
                 return;
             }
-
-            for(Item item : content.items()){
-                if(item.type == ItemType.material){
-                    state.teams.get(Team.blue).cores.first().entity.items.add(item, 2000);
+            
+            try{
+                Team team = Team.valueOf(arg[0]);
+                
+                for(Item item : content.items()){
+                    if(item.type == ItemType.material){
+                        state.teams.get(team).cores.first().entity.items.add(item, 2000);
+                    }
                 }
+                
+                info("Core filled.");
+            }catch(IllegalArgumentException ignored){
+                err("No such team exists.");
             }
-            info("Core filled.");
         });
 
         handler.register("crashreport", "<on/off>", "Disables or enables automatic crash reporting", arg -> {