lamport transfer helper

thewuhxyz · Oct 16, 2024 · 193dd5f · 193dd5f
1 parent 4796ca6
commit 193dd5f
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/lib/src/loaders.rs b/lib/src/loaders.rs
@@ -3,7 +3,7 @@ use bytemuck::Pod;
 use solana_program::program_pack::Pack;
 use solana_program::{account_info::AccountInfo, program_error::ProgramError, pubkey::Pubkey};
 
-use crate::{AccountDeserialize, AccountInfoValidation, AsAccount, Discriminator};
+use crate::{AccountDeserialize, AccountInfoValidation, AsAccount, Discriminator, LamportTransfer};
 #[cfg(feature = "spl")]
 use crate::{AccountValidation, AsSplToken};
 
@@ -104,6 +104,14 @@ impl AsAccount for AccountInfo<'_> {
     }
 }
 
+impl LamportTransfer for AccountInfo<'_> {
+    fn transfer(&self, amount: u64, to: AccountInfo) -> Result<(), ProgramError> {
+        **self.lamports.borrow_mut() -= amount;
+        **to.lamports.borrow_mut() += amount;
+        Ok(())
+    }
+}
+
 #[cfg(feature = "spl")]
 impl AsSplToken for AccountInfo<'_> {
     fn as_mint(&self) -> Result<spl_token::state::Mint, ProgramError> {

diff --git a/lib/src/traits.rs b/lib/src/traits.rs
@@ -1,5 +1,5 @@
 use bytemuck::Pod;
-use solana_program::{program_error::ProgramError, pubkey::Pubkey};
+use solana_program::{account_info::AccountInfo, program_error::ProgramError, pubkey::Pubkey};
 
 pub trait AccountDeserialize {
     fn try_from_bytes(data: &[u8]) -> Result<&Self, ProgramError>;
@@ -144,6 +144,10 @@ pub trait AsSplToken {
     ) -> Result<spl_token::state::Account, ProgramError>;
 }
 
+pub trait LamportTransfer {
+    fn transfer(&self, amount: u64, to: AccountInfo) -> Result<(), ProgramError>;
+}
+
 pub trait ProgramOwner {
     fn owner() -> Pubkey;
 }