klever-sc-modules 0.45.1

Klever WebAssembly standard smart contract modules
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

use klever_sc::imports::*;

#[klever_sc::module]
pub trait OnlyAdminModule {
    #[view(isAdmin)]
    fn is_admin(&self, address: ManagedAddress) -> bool {
        self.admins().contains(&address)
    }

    #[only_owner]
    #[endpoint(addAdmin)]
    fn add_admin(&self, address: ManagedAddress) {
        let caller = self.blockchain().get_caller();
        self.admins().insert(address.clone());

        self.admin_added_event(&caller, &address);
    }

    #[only_owner]
    #[endpoint(removeAdmin)]
    fn remove_admin(&self, address: ManagedAddress) {
        let caller = self.blockchain().get_caller();
        self.admins().swap_remove(&address);

        self.admin_removed_event(&caller, &address);
    }

    #[view(getAdmins)]
    #[storage_mapper("only_admin_module:admins")]
    fn admins(&self) -> UnorderedSetMapper<ManagedAddress>;

    fn require_caller_is_admin(&self) {
        require!(
            self.is_admin(self.blockchain().get_caller()),
            "Endpoint can only be called by admins"
        );
    }

    #[event("admin_added")]
    fn admin_added_event(
        &self,
        #[indexed] caller: &ManagedAddress,
        #[indexed] new_admin: &ManagedAddress,
    );

    #[event("admin_removed")]
    fn admin_removed_event(
        &self,
        #[indexed] caller: &ManagedAddress,
        #[indexed] removed_admin: &ManagedAddress,
    );
}