Add bounds checking on resources

Assignment 2/src/ca/cosc3p91/a2/gameobjects/CasaDeNarino.java

@@ -1,16 +1,14 @@
 package ca.cosc3p91.a2.gameobjects;
 
-import java.util.ArrayList;
-
 public class CasaDeNarino extends Building {
 
     private int goldCapacity = 0;
     private int ironCapacity = 0;
     private int woodCapacity = 0;
 
-    private int currentGold;
+    private int currentGold = 0;
-    private int currentIron;
+    private int currentIron = 0;
-    private int currentWood;
+    private int currentWood = 50;
 
     public CasaDeNarino(int lvl, VillageStage baseStage) {
         setLevel(lvl);
@@ -41,23 +39,38 @@ public class CasaDeNarino extends Building {
         return currentGold;
     }
 
-    public void addCurrentGold(int currentGold) {
+    public boolean addGold(int amount) {
-        this.currentGold += currentGold;
+        int newGold = this.currentGold + amount;
+        if (newGold <= goldCapacity && this.currentGold + amount >= 0){
+            this.currentGold += amount;
+            return true;
+        }
+        return false;
     }
 
     public int getCurrentIron() {
         return currentIron;
     }
 
-    public void addCurrentIron(int currentIron) {
+    public boolean addIron(int amount) {
-        this.currentIron += currentIron;
+        int newIron = this.currentIron + amount;
+        if (newIron <= ironCapacity && newIron >= 0) {
+            this.currentIron += amount;
+            return true;
+        }
+        return false;
     }
 
     public int getCurrentWood() {
         return currentWood;
     }
 
-    public void addCurrentWood(int currentWood) {
+    public boolean addWood(int amount) {
-        this.currentWood += currentWood;
+        int newWood = this.currentWood + amount;
+        if (newWood <= woodCapacity && newWood >= 0) {
+            this.currentWood += amount;
+            return true;
+        }
+        return false;
     }
 }

Assignment 2/src/ca/cosc3p91/a2/gameobjects/IronMine.java

@@ -11,6 +11,6 @@ public class IronMine extends ResourceBuilding {
 
     @Override
     protected ResourceHarvestHandler getHarvestHandler() {
-        return hall -> hall.addCurrentIron(getHarvestRate());
+        return hall -> hall.addIron(getHarvestRate());
     }
 }

Assignment 2/src/ca/cosc3p91/a2/gameobjects/LumberMine.java

@@ -11,7 +11,7 @@ public class LumberMine extends ResourceBuilding {
 
     @Override
     protected ResourceHarvestHandler getHarvestHandler() {
-        return hall -> hall.addCurrentWood(getHarvestRate());
+        return hall -> hall.addWood(getHarvestRate());
     }
 
 }

Assignment 2/src/ca/cosc3p91/a2/gameobjects/SaulGoodMine.java

@@ -11,6 +11,6 @@ public class SaulGoodMine extends ResourceBuilding {
 
     @Override
     protected ResourceHarvestHandler getHarvestHandler() {
-        return hall -> hall.addCurrentGold(getHarvestRate());
+        return hall -> hall.addGold(getHarvestRate());
     }
 }

Assignment 2/src/ca/cosc3p91/a2/gameobjects/VillageHallStages.java

@@ -0,0 +1,4 @@
+package ca.cosc3p91.a2.gameobjects;
+
+public class VillageHallStages {
+}