krusty-kms-client 0.2.2

Starknet RPC client for interacting with TONGO contracts
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
106
107
108
109
110
111
112
113
114
115
116
117

//! Fluent transaction builder for batching multiple calls.

use crate::abi;
use crate::staking::Staking;
use crate::tx::Tx;
use crate::wallet::utils::core_felt_to_rs;
use crate::wallet::WalletExecutor;
use krusty_kms_common::address::Address;
use krusty_kms_common::amount::Amount;
use krusty_kms_common::token::Token;
use krusty_kms_common::Result;
use starknet_rust::core::types::Call;

/// A builder that accumulates `Call`s and sends them as a single multicall.
pub struct TxBuilder<'w> {
    wallet: &'w dyn WalletExecutor,
    calls: Vec<Call>,
}

impl<'w> TxBuilder<'w> {
    pub fn new(wallet: &'w dyn WalletExecutor) -> Self {
        Self {
            wallet,
            calls: Vec::new(),
        }
    }

    /// Add an arbitrary call.
    #[allow(clippy::should_implement_trait)]
    pub fn add(mut self, call: Call) -> Self {
        self.calls.push(call);
        self
    }

    /// Add an ERC-20 approve call.
    pub fn approve(self, token: &Token, spender: &Address, amount: &Amount) -> Self {
        let (low, high) = amount.to_u256();
        let call = Call {
            to: core_felt_to_rs(token.address.as_felt()),
            selector: *abi::erc20::APPROVE,
            calldata: vec![
                core_felt_to_rs(spender.as_felt()),
                core_felt_to_rs(low),
                core_felt_to_rs(high),
            ],
        };
        self.add(call)
    }

    /// Add an ERC-20 transfer call.
    pub fn transfer(self, token: &Token, to: &Address, amount: &Amount) -> Self {
        let (low, high) = amount.to_u256();
        let call = Call {
            to: core_felt_to_rs(token.address.as_felt()),
            selector: *abi::erc20::TRANSFER,
            calldata: vec![
                core_felt_to_rs(to.as_felt()),
                core_felt_to_rs(low),
                core_felt_to_rs(high),
            ],
        };
        self.add(call)
    }

    /// Add calls to enter a delegation pool (approve + enter).
    pub fn enter_pool(self, staking: &Staking, amount: &Amount, reward_address: &Address) -> Self {
        let calls = staking.populate_enter(amount, reward_address);
        let mut s = self;
        for call in calls {
            s = s.add(call);
        }
        s
    }

    /// Add calls to add more stake to a pool (approve + add).
    pub fn add_to_pool(self, staking: &Staking, amount: &Amount) -> Self {
        let calls = staking.populate_add(amount);
        let mut s = self;
        for call in calls {
            s = s.add(call);
        }
        s
    }

    /// Add a claim-rewards call.
    pub fn claim_rewards(self, staking: &Staking, reward_address: &Address) -> Self {
        let call = staking.populate_claim_rewards(reward_address);
        self.add(call)
    }

    /// Add an exit-intent call.
    pub fn exit_intent(self, staking: &Staking, amount: &Amount) -> Self {
        let call = staking.populate_exit_intent(amount);
        self.add(call)
    }

    /// Add an exit-action call.
    pub fn exit_pool(self, staking: &Staking) -> Self {
        let call = staking.populate_exit();
        self.add(call)
    }

    /// Inspect the accumulated calls.
    pub fn calls(&self) -> &[Call] {
        &self.calls
    }

    /// Estimate the fee for all accumulated calls.
    pub async fn estimate_fee(&self) -> Result<starknet_rust::core::types::FeeEstimate> {
        self.wallet.estimate_fee(self.calls.clone()).await
    }

    /// Execute all accumulated calls as a single transaction.
    pub async fn send(self) -> Result<Tx> {
        self.wallet.execute(self.calls).await
    }
}