add supply

amegakure-studio · Nov 4, 2024 · 89c4121 · 89c4121
1 parent ff0f73b
commit 89c4121
Showing 1 changed file with 22 additions and 6 deletions.
diff --git a/src/lib.cairo b/src/lib.cairo
@@ -48,6 +48,7 @@ mod MyToken {
         #[substorage(v0)]
         ownable: OwnableComponent::Storage,
         beasts_stats: Map<u256, BeastStats>,
+        total_supply: u256,
     }
 
     #[event]
@@ -102,12 +103,15 @@ mod MyToken {
         fn safe_mint(
             ref self: ContractState,
             recipient: ContractAddress,
-            token_id: u256,
             beast_stats: BeastStats,
-            data: Span<felt252>,
         ) {
             self.ownable.assert_only_owner();
-            self.erc721.safe_mint(recipient, token_id, data);
+
+            let total_supply = self.total_supply.read();
+            let token_id = total_supply + 1;
+            self.total_supply.write(token_id);
+
+            self.erc721.safe_mint(recipient, token_id, array![].span());
             self.beasts_stats.entry(token_id).write(BeastStats {
                 tier: beast_stats.tier,
                 level: beast_stats.level,
@@ -121,11 +125,9 @@ mod MyToken {
         fn safeMint(
             ref self: ContractState,
             recipient: ContractAddress,
-            tokenId: u256,
             beast_stats: BeastStats,
-            data: Span<felt252>,
         ) {
-            self.safe_mint(recipient, tokenId, beast_stats, data);
+            self.safe_mint(recipient, beast_stats);
         }
 
         #[external(v0)]
@@ -143,5 +145,19 @@ mod MyToken {
         ) -> BeastStats {
             self.beasts_stats.entry(tokenId).read()
         }
+
+        #[external(v0)]
+        fn total_supply(
+            self: @ContractState,
+        ) -> u256 {
+            self.total_supply.read()
+        }
+
+        #[external(v0)]
+        fn totalSupply(
+            self: @ContractState,
+        ) -> u256 {
+            self.total_supply.read()
+        }
     }
 }