b3_utils 0.13.1

Utility functions for building on the Internet Computer
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

use std::cell::RefCell;

use candid::Principal;

use crate::{
    memory::{error::StableMemoryError, types::DefaultStableCell, with_stable_mem_mut},
    principal::StoredPrincipal,
};

fn init_owner(
    name: &str,
    id: u8,
) -> Result<RefCell<DefaultStableCell<StoredPrincipal>>, StableMemoryError> {
    let memory = with_stable_mem_mut(|pm| pm.create(name, id))?;

    let owner_id = StoredPrincipal::from(ic_cdk::caller());

    let cell = DefaultStableCell::init(memory, owner_id)
        .map_err(|e| StableMemoryError::UnableToCreateMemory(e.to_string()))?;

    Ok(RefCell::new(cell))
}

thread_local! {
    static OWNER: RefCell<DefaultStableCell<StoredPrincipal>> = init_owner("owner", 251).unwrap();
}

pub fn get_owner() -> Principal {
    OWNER.with(|states| {
        let state = states.borrow();

        state.get().into()
    })
}

pub fn set_owner(new_owner: Principal) -> Result<Principal, String> {
    OWNER.with(|states| {
        let mut state = states.borrow_mut();
        let old_owner = state.set(new_owner.into()).expect("Unable to set owner");

        Ok(old_owner.into())
    })
}

pub fn caller_is_owner() -> Result<(), String> {
    let caller_id = ic_cdk::caller();
    let owner_id = get_owner();

    if caller_id == owner_id {
        Ok(())
    } else {
        Err(format!("Caller is not the owner. Caller: {}", caller_id))
    }
}