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
45
46

/*
    Appellation: mainnet <module>
    Creator: FL03 <jo3mccain@icloud.com>
    Description:
        ... Summary ...
*/
use crate::KademliaMS;
use libp2p::{
    floodsub::{Floodsub, FloodsubEvent},
    kad::KademliaEvent,
    mdns::{MdnsEvent, TokioMdns},
    NetworkBehaviour,
};

/// Create the standard behaviour for blockchain networks, building on top of Kademlia and MDNS
#[derive(NetworkBehaviour)]
#[behaviour(out_event = "MainnetBehaviourEvent")]
pub struct MainnetBehaviour {
    pub floodsub: Floodsub,
    pub kademlia: KademliaMS,
    pub mdns: TokioMdns,
}

pub enum MainnetBehaviourEvent {
    Fsub(FloodsubEvent),
    Kad(KademliaEvent),
    Mdns(MdnsEvent),
}

impl From<FloodsubEvent> for MainnetBehaviourEvent {
    fn from(event: FloodsubEvent) -> Self {
        Self::Fsub(event)
    }
}

impl From<KademliaEvent> for MainnetBehaviourEvent {
    fn from(event: KademliaEvent) -> Self {
        Self::Kad(event)
    }
}

impl From<MdnsEvent> for MainnetBehaviourEvent {
    fn from(event: MdnsEvent) -> Self {
        Self::Mdns(event)
    }
}