poldercast 0.13.2

use crate::nodes::Entry;
use crate::{
    Address, DefaultPolicy, Gossips, GossipsBuilder, Layer, Node, NodeProfile, Nodes, Policy,
    PolicyReport, Selection, ViewBuilder,
};

pub struct Topology {
    /// The local node identity
    profile: NodeProfile,

    nodes: Nodes,

    layers: Vec<Box<dyn Layer + Send + Sync>>,

    policy: Box<dyn Policy + Send + Sync>,
}

impl Topology {
    /// create a new topology for the given [`NodeProfile`].
    ///
    /// the topology will manage a set of peers that is capped by the given
    /// `cap` default value of `1_024`.
    ///
    /// [`NodeProfile`]: ./struct.NodeProfile.html
    pub fn new(profile: NodeProfile) -> Self {
        Self::new_with(1_024, profile)
    }

    /// create a new topology for the given [`NodeProfile`] and capacity
    ///
    /// The capacity will be the max number of entries in the topology. Removing
    /// the items that are the least recently used.
    ///
    /// [`NodeProfile`]: ./struct.NodeProfile.html
    pub fn new_with(cap: usize, profile: NodeProfile) -> Self {
        Self {
            profile,
            nodes: Nodes::new(cap),
            layers: Vec::default(),
            policy: Box::new(DefaultPolicy::default()),
        }
    }

    pub fn profile(&self) -> &NodeProfile {
        &self.profile
    }

    pub fn add_layer<L>(&mut self, layer: L)
    where
        L: Layer + Send + Sync + 'static,
    {
        self.layers.push(Box::new(layer));
    }

    pub fn set_policy<P>(&mut self, policy: P)
    where
        P: Policy + Send + Sync + 'static,
    {
        self.policy = Box::new(policy);
    }

    pub fn view(&mut self, from: Option<Address>, selection: Selection) -> Vec<Address> {
        let mut view_builder = ViewBuilder::new(selection);

        if let Some(from) = from {
            view_builder.with_origin(from);
        }

        for layer in self.layers.iter_mut() {
            layer.view(&mut view_builder, &mut self.nodes)
        }

        view_builder.build(&mut self.nodes)
    }

    fn update_known_nodes(&mut self, from: Address, gossips: Gossips) {
        for gossip in gossips.into_iter() {
            if gossip.address() == self.profile.address() {
                // ignore ourselves
                continue;
            }

            // can only happen once by the remote
            let address = gossip.address().cloned().unwrap_or_else(|| from.clone());

            match self.nodes.entry(address.clone()) {
                Entry::Occupied(mut occupied) => {
                    occupied.modify(&mut self.policy, |node| node.update_gossip(gossip));
                }
                Entry::Vacant(mut vacant) => {
                    vacant.insert(Node::new(address, gossip));
                }
            }
        }
    }

    pub fn initiate_gossips(&mut self, with: Address) -> Gossips {
        if let Some(with) = self.nodes.get_mut(&with) {
            with.logs_mut().gossiping();
        }
        let mut gossips_builder = GossipsBuilder::new(with);

        for layer in self.layers.iter_mut() {
            layer.gossips(&self.profile, &mut gossips_builder, &self.nodes)
        }

        gossips_builder.build(self.profile.clone(), &self.nodes)
    }

    /// reset the layers, allowing an update of the internal state
    ///
    pub fn force_reset_layers(&mut self) {
        self.nodes.reset(&mut self.policy);
        self.reset_layers()
    }

    fn reset_layers(&mut self) {
        for layer in self.layers.iter_mut() {
            layer.reset();
            layer.populate(&self.profile, &self.nodes);
        }
    }

    pub fn accept_gossips(&mut self, from: Address, gossips: Gossips) {
        if let Some(from) = self.nodes.get_mut(&from) {
            from.logs_mut().gossiping();
        }

        self.update_known_nodes(from, gossips);

        self.reset_layers();
    }

    pub fn exchange_gossips(&mut self, with: Address, gossips: Gossips) -> Gossips {
        if let Some(with) = self.nodes.get_mut(&with) {
            with.logs_mut().gossiping();
        }

        self.update_known_nodes(with.clone(), gossips);

        let mut gossips_builder = GossipsBuilder::new(with);

        for layer in self.layers.iter_mut() {
            layer.reset();
            layer.populate(&self.profile, &self.nodes);
            layer.gossips(&self.profile, &mut gossips_builder, &self.nodes)
        }

        gossips_builder.build(self.profile.clone(), &self.nodes)
    }

    pub fn update_node<F>(&mut self, id: Address, update: F) -> Option<PolicyReport>
    where
        F: FnOnce(&mut Node),
    {
        self.nodes.entry(id).and_modify(&mut self.policy, update)
    }

    /// function to access the nodes data structure. From there it is possible
    /// to query the available nodes, the non-publicly-reachable nodes and the
    /// quarantined nodes.
    ///
    pub fn nodes(&self) -> &Nodes {
        &self.nodes
    }
}