add sotrage writes to set_src20_data

FuelLabs · Sep 16, 2024 · 17809b0 · 17809b0
1 parent 29986f1
commit 17809b0
Showing 1 changed file with 25 additions and 2 deletions.
diff --git a/examples/src20-native-asset/multi_asset/src/multi_asset.sw b/examples/src20-native-asset/multi_asset/src/multi_asset.sw
@@ -199,9 +199,32 @@ impl SetSRC20Data for Contract {
         }
         let sender = msg_sender().unwrap();
 
-        SetNameEvent::new(asset, name, sender).log();
-        SetSymbolEvent::new(asset, symbol, sender).log();
+        match name {
+            Some(unwrapped_name) => {
+                storage.name.get(asset).write_slice(unwrapped_name);
+                SetNameEvent::new(asset, name, sender).log();
+            },
+            None => {
+                let _ = storage.name.get(asset).clear();
+                SetNameEvent::new(asset, name, sender).log();
+            }
+        }
+
+        match symbol {
+            Some(unwrapped_symbol) => {
+                storage.symbol.get(asset).write_slice(unwrapped_symbol);
+                SetSymbolEvent::new(asset, symbol, sender).log();
+            },
+            None => {
+                let _ = storage.symbol.get(asset).clear();
+                SetSymbolEvent::new(asset, symbol, sender).log();
+            }
+        }
+
+        storage.decimals.get(asset).write(decimals);
         SetDecimalsEvent::new(asset, decimals, sender).log();
+
+        storage.total_supply.get(asset).write(supply);
         TotalSupplyEvent::new(asset, supply, sender).log();
     }
 }