From f38fb91076761ac6282cd15429d90a4536bde371 Mon Sep 17 00:00:00 2001
From: wei <1415121722@qq.com>
Date: Sun, 15 Jun 2025 20:50:51 +0800
Subject: [PATCH] Add a method to obtain pumpswap token balance

---
 src/lib.rs | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/lib.rs b/src/lib.rs
index 0ee710e..03f21ae 100755
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -674,4 +674,13 @@ impl PumpFun {
         
         Ok(quote_amount)
     }
+
+    #[inline]
+    pub async fn get_payer_token_balance_with_pumpswap(&self, pool_address: &Pubkey) -> Result<u64, anyhow::Error> {
+        let pool = pumpswap::pool::Pool::fetch(&self.rpc, pool_address).await?;
+        
+        let (base_amount, _) = pool.get_token_balances(&self.rpc).await?;
+        
+        Ok(base_amount)
+    }
 }