1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use mithril_common::entities::{Epoch, PartyId, SignerWithStake, Stake};
use serde::{Deserialize, Serialize};

/// Message structure of signer registrations for an epoch.
#[derive(Clone, Debug, PartialEq, Eq, Default, Serialize, Deserialize)]
pub struct SignerRegistrationsMessage {
    /// The epoch at which the registration was sent.
    pub registered_at: Epoch,

    /// The epoch at which the registration was able to send signatures.
    pub signing_at: Epoch,

    /// The signer registrations
    pub registrations: Vec<SignerRegistrationsListItemMessage>,
}

/// Message structure of a signer registration
#[derive(Clone, Debug, PartialEq, Eq, Default, Serialize, Deserialize)]
pub struct SignerRegistrationsListItemMessage {
    /// The registered signer party id
    pub party_id: PartyId,

    /// The registered signer stake
    pub stake: Stake,
}

impl SignerRegistrationsMessage {
    /// Build a [SignerRegistrationsMessage] from a list of signers with stake.
    pub fn new(registered_at: Epoch, signers_with_stake: Vec<SignerWithStake>) -> Self {
        let registrations: Vec<SignerRegistrationsListItemMessage> = signers_with_stake
            .into_iter()
            .map(|signer| SignerRegistrationsListItemMessage {
                party_id: signer.party_id,
                stake: signer.stake,
            })
            .collect();

        Self {
            registered_at,
            signing_at: registered_at.offset_to_signer_signing_offset(),
            registrations,
        }
    }
}