ic-atp 0.1.3

ATP(Account Transfer Protocol) implementation for 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
55
56
57
58

//! Abstraction layer for Internet Computer API calls
//! This allows for dependency injection and easier testing

use candid::Principal;
use std::cell::RefCell;
use std::rc::Rc;

/// A trait that abstracts the IC API calls needed by the application
pub trait IcApi {
    /// Get the caller's principal
    fn caller(&self) -> Principal;

    /// Get the canister's own principal
    fn id(&self) -> Principal;

    /// Get the current IC time in nanoseconds
    fn time(&self) -> u64;

    /// Print a debug message to the IC console
    fn println(&self, message: &str);
}

/// Default implementation that uses the actual ic_cdk::api
pub struct DefaultIcApi;

impl IcApi for DefaultIcApi {
    fn caller(&self) -> Principal {
        ic_cdk::api::caller()
    }

    fn id(&self) -> Principal {
        ic_cdk::api::id()
    }

    fn time(&self) -> u64 {
        ic_cdk::api::time()
    }

    fn println(&self, message: &str) {
        ic_cdk::println!("{}", message);
    }
}

thread_local! {
    static CURRENT_IC_API: RefCell<Rc<dyn IcApi>> = RefCell::new(Rc::new(DefaultIcApi));
}

/// Set a custom IC API implementation (primarily for testing)
pub fn set_ic_api(api: Rc<dyn IcApi>) {
    CURRENT_IC_API.with(|current| {
        *current.borrow_mut() = api;
    });
}

/// Get a reference to the current IC API implementation
pub fn get_ic_api() -> Rc<dyn IcApi> {
    CURRENT_IC_API.with(|current| current.borrow().clone())
}