change ship unlocker to check both score and coin values before unlocking ships

Author: AutumnQiu99

src/main/java/com/dinosaur/dinosaurexploder/utils/WeaponUnlockChecker.java

@@ -1,7 +1,10 @@
 package com.dinosaur.dinosaurexploder.utils;
 
+import com.dinosaur.dinosaurexploder.exception.LockedShipException;
 import com.dinosaur.dinosaurexploder.exception.LockedWeaponException;
 import com.dinosaur.dinosaurexploder.model.HighScore;
+import com.dinosaur.dinosaurexploder.model.TotalCoins;
+
 import java.io.FileInputStream;
 import java.io.IOException;
 import java.io.ObjectInputStream;
@@ -14,11 +17,18 @@ public class WeaponUnlockChecker {
             1, 0,
             2, 50,
             3, 100);
+
+    private static final Map<Integer, Integer> coinMap = Map.of( // key: weaponNumber, value: lower limit total coins
+            1, 0,
+            2, 5,
+            3, 10);
+
     private HighScore highScore = new HighScore();
+    private TotalCoins totalCoins = new TotalCoins();
 
     public int check(int weaponNumber) {
         highScore = getHighScore();
-        checkScore(weaponNumber);
+        checkScoreAndCoins(weaponNumber);
         return weaponNumber;
     }
 
@@ -31,11 +41,44 @@ public HighScore getHighScore() {
         }
     }
 
-    private void checkScore(int weaponNumber) {
-        int lowerLimit = scoreMap.getOrDefault(weaponNumber, 0);
-        if (lowerLimit <= highScore.getHigh())
+    public TotalCoins getTotalCoins() {
+        try (FileInputStream file = new FileInputStream("totalCoins.ser");
+                ObjectInputStream in = new ObjectInputStream(file)) {
+            return (TotalCoins) in.readObject();
+        } catch (IOException | ClassNotFoundException e) {
+            return new TotalCoins();
+        }
+    }
+
+    // private void checkScore(int weaponNumber) {
+    // int lowerLimit = scoreMap.getOrDefault(weaponNumber, 0);
+    // if (lowerLimit <= highScore.getHigh())
+    // return;
+    // throw new
+    // LockedWeaponException(languageManager.getTranslation("weapon_locked") + "\n"
+    // +
+    // languageManager.getTranslation("unlock_highScore").replace("##",
+    // String.valueOf(lowerLimit)));
+    // }
+
+    private void checkScoreAndCoins(int weaponNumber) {
+        int lowerScoreLimit = scoreMap.getOrDefault(weaponNumber, 0);
+        int lowerCoinLimit = coinMap.getOrDefault(weaponNumber, 0);
+
+        if (lowerScoreLimit <= highScore.getHigh() && lowerCoinLimit <= totalCoins.getTotal())
             return;
-        throw new LockedWeaponException(languageManager.getTranslation("weapon_locked") + "\n" +
-                languageManager.getTranslation("unlock_highScore").replace("##", String.valueOf(lowerLimit)));
+        else if (lowerScoreLimit > highScore.getHigh() && lowerCoinLimit <= totalCoins.getTotal()) {
+            throw new LockedWeaponException(languageManager.getTranslation("weapon_locked") + "\n" +
+                    languageManager.getTranslation("unlock_highScore").replace("##", String.valueOf(lowerScoreLimit)));
+        } else if (lowerScoreLimit <= highScore.getHigh() && lowerCoinLimit > totalCoins.getTotal()) {
+            throw new LockedWeaponException(languageManager.getTranslation("weapon_locked") + "\n" +
+                    languageManager.getTranslation("unlock_totalCoins").replace("##", String.valueOf(lowerCoinLimit)));
+        } else {
+            throw new LockedWeaponException(languageManager.getTranslation("weapon_locked") + "\n"
+                    + languageManager.getTranslation("unlock_highScore").replace("##", String.valueOf(lowerScoreLimit))
+                    + "\n" +
+                    languageManager.getTranslation("unlock_totalCoins").replace("##", String.valueOf(lowerCoinLimit)));
+        }
+
     }
 }