Store derivation_path in RegisteredInfo

entropyxyz · Aug 7, 2024 · 77ed456 · 77ed456
1 parent 016ab57
commit 77ed456
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pallets/registry/src/lib.rs b/pallets/registry/src/lib.rs
@@ -119,6 +119,9 @@ pub mod pallet {
     pub struct RegisteredInfo<T: Config> {
         pub programs_data: BoundedVec<ProgramInstance<T>, T::MaxProgramHashes>,
         pub program_modification_account: T::AccountId,
+        /// TODO (Nando): We're just going to store the `count` for now, but we should consider
+        /// storing the full derivation path here in the future (as a `Vec<u8>`).
+        pub derivation_path: Option<u32>,
         pub version_number: u8,
     }
 
@@ -140,6 +143,7 @@ pub mod pallet {
                     RegisteredInfo {
                         programs_data: BoundedVec::default(),
                         program_modification_account: account_id.clone(),
+                        derivation_path: None,
                         version_number: T::KeyVersionNumber::get(),
                     },
                 );
@@ -149,6 +153,7 @@ pub mod pallet {
                     RegisteredInfo {
                         programs_data: BoundedVec::default(),
                         program_modification_account: account_id.clone(),
+                        derivation_path: None, // TODO (Nando): What `count` should we use here?
                         version_number: T::KeyVersionNumber::get(),
                     },
                 );
@@ -665,6 +670,7 @@ pub mod pallet {
                     RegisteredInfo {
                         programs_data: registering_info.programs_data,
                         program_modification_account: registering_info.program_modification_account,
+                        derivation_path: None,
                         version_number: registering_info.version_number,
                     },
                 );
@@ -778,6 +784,7 @@ pub mod pallet {
                     programs_data,
                     program_modification_account: program_modification_account.clone(),
                     version_number: T::KeyVersionNumber::get(),
+                    derivation_path: Some(count),
                 },
             );