p2p 0.5.2

use priv_prelude::*;
use rust_sodium::crypto::box_::{gen_keypair, PublicKey};

/// Information necessary to connect to peer.
// NOTE: this structure is very similar to the one in Crust. Except the one in Crust stores
// PaAddr instead of SocketAddr. But it's worth considering moving PaAddr to p2p to reuse code.
#[derive(PartialEq, Eq, Hash, Clone, Debug, Serialize, Deserialize)]
pub struct PeerInfo {
    /// Peer IP address.
    pub addr: SocketAddr,
    /// Peer public key.
    pub pub_key: PublicKey,
}

impl PeerInfo {
    /// Constructs peer info.
    pub fn new(addr: SocketAddr, pub_key: PublicKey) -> Self {
        Self { addr, pub_key }
    }

    /// Constructs peer info with random generated public key.
    pub fn with_rand_key(addr: SocketAddr) -> Self {
        let (pub_key, _) = gen_keypair();
        Self::new(addr, pub_key)
    }
}

impl fmt::Display for PeerInfo {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "({}, {:?})", self.addr, self.pub_key)
    }
}