token: add common write_params fn

shielded_token/src/lib.rs

@@ -1,6 +1,8 @@
 //! Namada shielded token.
 
 pub mod conversion;
-pub mod storage;
+mod storage;
 pub mod storage_key;
 pub mod utils;
+
+pub use storage::*;

token/src/lib.rs

@@ -8,3 +8,20 @@ pub mod storage_key {
     pub use namada_shielded_token::storage_key::*;
     pub use namada_trans_token::storage_key::*;
 }
+
+use namada_core::types::address::Address;
+use namada_storage::{Result, StorageRead, StorageWrite};
+use namada_trans_token::Parameters;
+
+/// Initialize parameters for the token in storage during the genesis block.
+pub fn write_params<S>(
+    params: &Parameters,
+    storage: &mut S,
+    address: &Address,
+) -> Result<()>
+where
+    S: StorageRead + StorageWrite,
+{
+    namada_trans_token::write_params(params, storage, address)?;
+    namada_shielded_token::write_params(params, storage, address)
+}