Added supply getters for each horse level

MDJ-Labs · Aug 25, 2022 · 2a5ae4e · 2a5ae4e
1 parent 98e759d
commit 2a5ae4e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/contracts/StallionNFT.sol b/contracts/StallionNFT.sol
@@ -144,4 +144,18 @@ contract StallionNFT is ERC721, ERC721Enumerable, ERC721URIStorage, Ownable {
             return 'none';
         }
     }
+
+    function bulletSupply() external view returns(uint256){
+        return _levels[0].supply;
+
+    }
+
+    function hopeSupply() external view returns(uint256){
+        return _levels[1].supply;
+    }
+
+    function flashSupply() external view returns(uint256){
+        return _levels[2].supply;
+    }
+
 }