extend KeyContribution struct

Author: Mr-Leshiy

snapshot-lib/src/lib.rs

@@ -1,6 +1,7 @@
 pub use fraction::Fraction;
 use jormungandr_lib::{crypto::account::Identifier, interfaces::Value};
 use proptest::prelude::Arbitrary;
+use registration::MainnetStakeAddress;
 use registration::{Delegations, MainnetRewardAddress, VotingRegistration};
 use serde::{Deserialize, Serialize};
 use std::{borrow::Borrow, collections::BTreeMap, iter::Iterator, num::NonZeroU64};
@@ -36,6 +37,7 @@ pub enum Error {
 /// Contribution to a voting key for some registration
 #[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
 pub struct KeyContribution {
+    pub stake_public_key: MainnetStakeAddress,
     pub reward_address: MainnetRewardAddress,
     pub value: u64,
 }
@@ -79,12 +81,14 @@ impl Snapshot {
                     reward_address,
                     delegations,
                     voting_power,
+                    stake_public_key,
                     ..
                 } = reg;
 
                 match delegations {
                     Delegations::Legacy(vk) => {
                         acc.entry(vk).or_default().push(KeyContribution {
+                            stake_public_key,
                             reward_address,
                             value: voting_power.into(),
                         });
@@ -103,6 +107,7 @@ impl Snapshot {
                                 })
                                 .map(|(vk, value)| {
                                     acc.entry(vk).or_default().push(KeyContribution {
+                                        stake_public_key: stake_public_key.clone(),
                                         reward_address: reward_address.clone(),
                                         value: value.get(),
                                     });
@@ -111,6 +116,7 @@ impl Snapshot {
                                 .sum::<u64>()
                         });
                         acc.entry(last.0).or_default().push(KeyContribution {
+                            stake_public_key,
                             reward_address,
                             value: voting_power - others_total_vp,
                         });