astrox_macros 0.2.2

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
39
40
41
42

// 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;

        #[update(name = "canister_add", guard = "op_guard")]
        #[candid_method(update, rename = "canister_add")]
        pub fn canister_add(name: String, canister_id: Principal) -> Result<(), String> {
            REGISTRY.with(|s| s.borrow_mut().canister_add(name.clone(), canister_id));
            on_canister_added(&name, canister_id);
            Ok(())
        }

        #[update(name = "canister_remove", guard = "op_guard")]
        #[candid_method(update, rename = "canister_remove")]
        pub fn canister_remove(name: String, canister_id: Principal) -> Result<(), String> {
            REGISTRY.with(|s| s.borrow_mut().canister_remove(name, canister_id));
            Ok(())
        }

        #[update(name = "canister_list", guard = "op_guard")]
        #[candid_method(update, rename = "canister_list")]
        pub fn canister_list() -> Result<BTreeMap<String, Vec<Principal>>, String> {
            REGISTRY.with(|s| Ok(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));
        }
    };
}