diff --git a/src/lib.rs b/src/lib.rs index 643cfdc..7203f48 100755 --- a/src/lib.rs +++ b/src/lib.rs @@ -294,20 +294,35 @@ impl PumpFun { } #[inline] - pub async fn get_sol_balance(&self, payer: &Pubkey) -> Result { + pub fn get_sol_balance(&self, payer: &Pubkey) -> Result { trade::common::get_sol_balance(&self.rpc, payer) } #[inline] - pub async fn get_token_balance(&self, payer: &Pubkey, mint: &Pubkey) -> Result { + pub fn get_payer_sol_balance(&self) -> Result { + trade::common::get_sol_balance(&self.rpc, &self.payer.pubkey()) + } + + #[inline] + pub fn get_token_balance(&self, payer: &Pubkey, mint: &Pubkey) -> Result { trade::common::get_token_balance(&self.rpc, payer, mint) } #[inline] - pub async fn get_payer_token_balance(&self, mint: &Pubkey) -> Result { + pub fn get_payer_token_balance(&self, mint: &Pubkey) -> Result { trade::common::get_token_balance(&self.rpc, &self.payer.pubkey(), mint) } + #[inline] + pub fn get_payer_pubkey(&self) -> Pubkey { + self.payer.pubkey() + } + + #[inline] + pub fn get_payer(&self) -> &Keypair { + self.payer.as_ref() + } + #[inline] pub fn get_token_price(&self,virtual_sol_reserves: u64, virtual_token_reserves: u64) -> f64 { trade::common::get_token_price(virtual_sol_reserves, virtual_token_reserves)