Update CastleSiegeEvents.java

jac18042 · Mar 26, 2021 · c1d4ce7 · c1d4ce7
1 parent 94b8d4e
commit c1d4ce7
Showing 1 changed file with 21 additions and 13 deletions.
diff --git a/src/com/cramer/castlesiege/events/CastleSiegeEvents.java b/src/com/cramer/castlesiege/events/CastleSiegeEvents.java
@@ -61,9 +61,18 @@ public class CastleSiegeEvents implements Listener
         o.setDisplayName(ChatColor.DARK_GREEN + "Zombie Siege");
 
         Score mobCount = o.getScore("Zombies:");
-        mobCount.setScore(CastleSiege.totalMobs);
         Score villagerCount = o.getScore("Villagers:");
-        villagerCount.setScore(CastleSiege.totalVillagers);
+
+        if (CastleSiege.siegeOngoing)
+        {
+            mobCount.setScore(CastleSiege.totalMobs);
+            villagerCount.setScore(CastleSiege.totalVillagers);
+        }
+        else
+        {
+            mobCount.setScore(0);
+            villagerCount.setScore(12);
+        }
 
         p.setScoreboard(b);
     }
@@ -72,19 +81,18 @@ public class CastleSiegeEvents implements Listener
     @EventHandler
     public void onEntityDeath(EntityDeathEvent e)
     {
-        if (e.getEntityType() != EntityType.PLAYER)
-        {
-            if (e.getEntityType() == EntityType.VILLAGER) {
-                CastleSiege.totalVillagers -= 1;
-            }
-            else {
-                CastleSiege.totalMobs -= 1;
-                if (e.getEntity().getEquipment().getHelmet().getType() == Material.WITHER_SKELETON_SKULL)
-                {
-                    CastleSiege.WinGame();
+        if (CastleSiege.siegeOngoing) {
+            if (e.getEntityType() != EntityType.PLAYER) {
+                if (e.getEntityType() == EntityType.VILLAGER) {
+                    CastleSiege.totalVillagers -= 1;
+                } else {
+                    CastleSiege.totalMobs -= 1;
+                    if (e.getEntity().getEquipment().getHelmet().getType() == Material.WITHER_SKELETON_SKULL) {
+                        CastleSiege.WinGame();
+                    }
                 }
+                CastleSiege.UpdateScore();
             }
-            CastleSiege.UpdateScore();
         }
     }