typhoon-utility-traits 0.1.0

TODO
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

use {
    pinocchio::program_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 mut payer_lamports = self.mut_lamports()?;
        let mut recipient_lamports = to.mut_lamports()?;

        *payer_lamports = payer_lamports
            .checked_sub(amount)
            .ok_or(ProgramError::ArithmeticOverflow)?;
        *recipient_lamports = recipient_lamports
            .checked_add(amount)
            .ok_or(ProgramError::ArithmeticOverflow)?;

        Ok(())
    }

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

        *payer_lamports = payer_lamports
            .checked_sub(amount)
            .ok_or(ProgramError::ArithmeticOverflow)?;
        *recipient_lamports = recipient_lamports
            .checked_add(amount)
            .ok_or(ProgramError::ArithmeticOverflow)?;

        Ok(())
    }
}

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