Custom Card Completed

aps2019project · Jul 9, 2019 · 90ba52e · 90ba52e
1 parent 910d548
commit 90ba52e
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/src/controllers/graphical/UniversalShopController.java b/src/controllers/graphical/UniversalShopController.java
@@ -169,13 +169,19 @@ public void initializeIds() {
     }
 
 
-    public void addId(Card card){
-        if (card instanceof Spell)
-            spellIds.add("s" + card.getCardId());
-        if (card instanceof Hero)
-            heroIds.add("h" + card.getCardId());
-        if (card instanceof Minion)
-            minionIds.add("m" + card.getCardId());
+    public void addId(Card card) {
+        if (card instanceof Spell) {
+            spellIds.add("s" + card.getCode());
+            card.setCardId("s" + card.getCode());
+        }
+        if (card instanceof Hero) {
+            heroIds.add("h" + card.getCode());
+            card.setCardId("h" + card.getCode());
+        }
+        if (card instanceof Minion) {
+            minionIds.add("m" + card.getCode());
+            card.setCardId("m" + card.getCode());
+        }
     }
 
     public void initializeCards() {