rusk-recovery: support stake owner in genesis

dusk-network · Dec 22, 2024 · ad8748b · ad8748b
1 parent 3b60ac2
commit ad8748b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/rusk-recovery/src/state/snapshot/stake.rs b/rusk-recovery/src/state/snapshot/stake.rs
@@ -14,7 +14,8 @@ use super::wrapper::Wrapper;
 
 #[derive(Serialize, Deserialize, PartialEq, Eq)]
 pub struct GenesisStake {
-    pub(crate) address: Wrapper<BlsPublicKey, { BlsPublicKey::SIZE }>,
+    address: Wrapper<BlsPublicKey, { BlsPublicKey::SIZE }>,
+    owner: Option<Wrapper<BlsPublicKey, { BlsPublicKey::SIZE }>>,
     pub amount: Dusk,
     pub eligibility: Option<u64>,
     pub reward: Option<Dusk>,
@@ -26,6 +27,7 @@ impl GenesisStake {
     }
 
     pub fn to_stake_keys(&self) -> StakeKeys {
-        StakeKeys::single_key(*self.address())
+        let owner: &BlsPublicKey = self.owner.as_ref().unwrap_or(&self.address);
+        StakeKeys::new(*self.address(), *owner)
     }
 }