typhoon-utility-traits 0.3.0

Trait extensions for account operations
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

use {
    pinocchio::error::ProgramError,
    typhoon_accounts::{
        Mut, Signer, SignerAccount, SignerCheck, SystemAccount, UncheckedAccount, WritableAccount,
    },
    typhoon_errors::Error,
};

pub trait LamportsChecked: WritableAccount + SignerAccount {
    #[inline(always)]
    fn send(&self, to: &impl WritableAccount, amount: u64) -> Result<(), Error> {
        let payer_lamports = self.lamports();
        let recipient_lamports = to.lamports();

        self.set_lamports(
            payer_lamports
                .checked_sub(amount)
                .ok_or(ProgramError::ArithmeticOverflow)?,
        );
        to.set_lamports(recipient_lamports.wrapping_add(amount));

        Ok(())
    }

    #[inline(always)]
    fn send_all(&self, to: &impl WritableAccount) -> Result<(), Error> {
        let amount = self.lamports();
        let recipient_lamports = to.lamports();

        self.set_lamports(0);
        to.set_lamports(recipient_lamports.wrapping_add(amount));

        Ok(())
    }
}

impl<C: SignerCheck> LamportsChecked for Mut<Signer<'_, SystemAccount<'_>, C>> {}
impl<C: SignerCheck> LamportsChecked for Mut<Signer<'_, UncheckedAccount<'_>, C>> {}