canfund 0.8.5

This Rust library provides a simple interface to monitor and add cycles to canisters 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
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
101
102
103
104
105

use crate::types::{WithdrawArgs, WithdrawError};
use async_trait::async_trait;
use candid::Principal;
use ic_cdk::call::CallResult;
use ic_ledger_types::{transfer, TransferArgs, TransferResult};
use icrc_ledger_types::icrc1::transfer::BlockIndex;

#[async_trait]
pub trait LedgerCanister: Send + Sync {
    async fn transfer(&self, args: TransferArgs) -> CallResult<TransferResult>;
}

pub struct IcLedgerCanister {
    canister_id: Principal,
}

impl IcLedgerCanister {
    pub fn new(canister_id: Principal) -> Self {
        Self { canister_id }
    }
}

#[async_trait]
impl LedgerCanister for IcLedgerCanister {
    async fn transfer(&self, args: TransferArgs) -> CallResult<TransferResult> {
        transfer(self.canister_id, &args).await
    }
}

#[async_trait]
pub trait WithdrawableLedgerCanister: Send + Sync {
    async fn withdraw(&self, args: WithdrawArgs) -> CallResult<Result<BlockIndex, WithdrawError>>;
}

pub struct CyclesLedgerCanister {
    canister_id: Principal,
}

impl CyclesLedgerCanister {
    pub fn new(canister_id: Principal) -> Self {
        Self { canister_id }
    }
}

#[async_trait]
impl WithdrawableLedgerCanister for CyclesLedgerCanister {
    async fn withdraw(&self, args: WithdrawArgs) -> CallResult<Result<BlockIndex, WithdrawError>> {
        Ok(
            ic_cdk::call::Call::unbounded_wait(self.canister_id, "withdraw")
                .with_arg(args)
                .await?
                .candid()?,
        )
    }
}

#[cfg(test)]
pub mod test {
    use std::sync::Arc;

    use super::*;
    use async_trait::async_trait;
    use tokio::sync::RwLock;

    #[derive(Default)]
    pub struct TestLedgerCanister {
        pub transfer_called_with: Arc<RwLock<Vec<TransferArgs>>>,
        pub returns_with: Option<CallResult<TransferResult>>,
    }
    #[async_trait]
    impl LedgerCanister for TestLedgerCanister {
        async fn transfer(&self, args: TransferArgs) -> CallResult<TransferResult> {
            let mut locked = self.transfer_called_with.write().await;
            locked.push(args);

            if let Some(value) = &self.returns_with {
                return value.clone();
            }

            Ok(Ok(0))
        }
    }

    #[derive(Default)]
    pub struct TestCyclesLedgerCanister {
        pub transfer_called_with: Arc<RwLock<Vec<WithdrawArgs>>>,
        pub returns_with: Option<CallResult<Result<BlockIndex, WithdrawError>>>,
    }
    #[async_trait]
    impl WithdrawableLedgerCanister for TestCyclesLedgerCanister {
        async fn withdraw(
            &self,
            args: WithdrawArgs,
        ) -> CallResult<Result<BlockIndex, WithdrawError>> {
            let mut locked = self.transfer_called_with.write().await;
            locked.push(args);

            if let Some(value) = &self.returns_with {
                return value.clone();
            }

            Ok(Ok(0_u64.into()))
        }
    }
}