astrox_macros 0.1.0

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

use candid::{CandidType, Principal};
use ic_cdk::export::candid::Deserialize;
use ic_cdk::id;
use serde::Serialize;

/// OwnerTrait, keep it simple
pub trait OwnerTrait {
    /// set owners to canister
    fn role_owner_set(&mut self, owners: Vec<Principal>);

    /// add owner to canister
    fn role_owner_add(&mut self, user_id: Principal);

    /// remove owner from canister
    fn role_owner_remove(&mut self, user_id: Principal);

    /// check whether `who` is the owner of canister, here we have 3 status
    ///     NotSet,Same,Different
    fn check_owner(&self, who: Principal) -> bool;
}

pub trait UserTrait {
    /// set owners to canister
    fn role_user_set(&mut self, users: Vec<Principal>);

    /// add owner to canister
    fn role_user_add(&mut self, user_id: Principal);

    /// remove owner from canister
    fn role_user_remove(&mut self, user_id: Principal);

    /// check whether `who` is the owner of canister, here we have 3 status
    ///     NotSet,Same,Different
    fn check_user(&self, who: Principal) -> bool;
}

#[derive(CandidType, Serialize, Deserialize, Clone, Debug)]
pub struct User {
    owners: Vec<Principal>,
    users: Vec<Principal>,
}

impl Default for User {
    fn default() -> Self {
        User {
            owners: vec![],
            users: vec![],
        }
    }
}

impl OwnerTrait for User {
    fn role_owner_set(&mut self, owners: Vec<Principal>) {
        ic_cdk::println!("{} set owner to: {:?}", id(), owners);
        self.owners = owners;
    }

    fn role_owner_add(&mut self, user_id: Principal) {
        ic_cdk::println!("{} add owner: {}", id(), user_id);
        if !self.owners.contains(&user_id) {
            self.owners.push(user_id);
        }
    }

    fn role_owner_remove(&mut self, user_id: Principal) {
        ic_cdk::println!("{} remove owner: {}", id(), user_id);
        if self.owners.contains(&user_id) {
            self.owners.retain(|p| *p != user_id);
        }
    }

    fn check_owner(&self, who: Principal) -> bool {
        self.owners.contains(&who)
    }
}

impl UserTrait for User {
    fn role_user_set(&mut self, users: Vec<Principal>) {
        ic_cdk::println!("{} set user to: {:?}", id(), users);
        self.users = users;
    }

    fn role_user_add(&mut self, user_id: Principal) {
        ic_cdk::println!("{} add user: {}", id(), user_id);
        if !self.users.contains(&user_id) {
            self.users.push(user_id);
        }
    }

    fn role_user_remove(&mut self, user_id: Principal) {
        ic_cdk::println!("{} remove user: {}", id(), user_id);
        if self.users.contains(&user_id) {
            self.users.retain(|p| *p != user_id);
        }
    }

    fn check_user(&self, who: Principal) -> bool {
        self.users.contains(&who) || self.owners.contains(&who)
    }
}