namada_ibc 0.150.2

Namada IBC
Documentation 
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
47
48
49
50
51
52
53
54
55
56

//! Functions to handle IBC modules

use std::rc::Rc;

use ibc::core::host::types::identifiers::PortId;
use ibc::core::router::module::Module;
use ibc::core::router::router::Router;
use ibc::core::router::types::module::ModuleId;
use namada_core::collections::HashMap;

use super::super::ModuleWrapper;

/// IBC router
#[derive(Debug, Default)]
pub struct IbcRouter<'a> {
    modules: HashMap<ModuleId, Rc<dyn ModuleWrapper + 'a>>,
    ports: HashMap<PortId, ModuleId>,
}

impl<'a> IbcRouter<'a> {
    /// Make new Router
    pub fn new() -> Self {
        Self {
            modules: HashMap::new(),
            ports: HashMap::new(),
        }
    }

    /// Add TokenTransfer route
    pub fn add_transfer_module(&mut self, module: impl ModuleWrapper + 'a) {
        let module_id = module.module_id();
        let port_id = module.port_id();
        self.modules.insert(module_id.clone(), Rc::new(module));
        self.ports.insert(port_id, module_id);
    }
}

impl Router for IbcRouter<'_> {
    fn get_route(&self, module_id: &ModuleId) -> Option<&dyn Module> {
        self.modules.get(module_id).map(|b| b.as_module())
    }

    fn get_route_mut(
        &mut self,
        module_id: &ModuleId,
    ) -> Option<&mut dyn Module> {
        self.modules
            .get_mut(module_id)
            .and_then(Rc::get_mut)
            .map(|b| b.as_module_mut())
    }

    fn lookup_module(&self, port_id: &PortId) -> Option<ModuleId> {
        self.ports.get(port_id).cloned()
    }
}