astrox_macros 0.2.7

add user management and canister relation management for canister
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

// macro user should implement the on_canister_added method
#[macro_export]
macro_rules! inject_canister_registry {
    () => {
        thread_local! {
          pub static REGISTRY: RefCell<Registry> = RefCell::new(Registry::default());
        }

        use astrox_macros::registry::CanisterTrait;
        use astrox_macros::registry::Registry;
        use std::collections::BTreeMap;

        pub fn canister_add(name: String, canister_id: Principal)  {
            REGISTRY.with(|s| s.borrow_mut().canister_add(name.clone(), canister_id));
            on_canister_added(&name, canister_id);
        }

        pub fn canister_remove(name: String, canister_id: Principal)  {
            REGISTRY.with(|s| s.borrow_mut().canister_remove(name, canister_id));
        }

        pub fn canister_remove_all(name: String)  {
            REGISTRY.with(|s| s.borrow_mut().canister_remove_all(name));
        }

        pub fn canister_list() -> BTreeMap<String, Vec<Principal>> {
            REGISTRY.with(|s| s.borrow().canister_list_all())
        }

        pub fn registry_pre_upgrade() -> Registry {
            REGISTRY.with(|s| s.take().into())
        }

        pub fn registry_post_upgrade(stable_state: Registry) {
            REGISTRY.with(|s| s.replace(stable_state));
        }
    };
}