feat: Add PumpSwap trading functionality module

2025-06-06 22:29:30 +08:00
parent 63109713f4
commit 94c2a22ab7
19 changed files with 2334 additions and 21 deletions
@@ -0,0 +1,292 @@
+use crate::error::ClientResult;
+use crate::common::pumpswap::{
+    logs_data::{
+        PumpSwapInstruction,
+        BuyEvent, SellEvent, CreatePoolEvent, DepositEvent, WithdrawEvent,
+        DisableEvent, UpdateAdminEvent, UpdateFeeConfigEvent, discriminators
+    },
+    logs_events::PumpSwapEvent
+};
+
+use solana_sdk::pubkey::Pubkey;
+use solana_sdk::instruction::CompiledInstruction;
+use std::time::{SystemTime, UNIX_EPOCH};
+
+/// 处理PumpSwap日志并调用回调函数
+pub async fn process_pumpswap_logs<F>(
+    signature: &str,
+    logs: Vec<String>,
+    slot: Option<u64>,
+    callback: F,
+) -> ClientResult<()>
+where
+    F: Fn(&str, PumpSwapEvent) + Send + Sync,
+{
+    let events = PumpSwapEvent::parse_logs(&logs);
+    for mut event in events {
+        // 设置签名和slot
+        match &mut event {
+            PumpSwapEvent::Buy(e) => {
+                e.signature = signature.to_string();
+                if let Some(s) = slot {
+                    e.slot = s;
+                }
+            },
+            PumpSwapEvent::Sell(e) => {
+                e.signature = signature.to_string();
+                if let Some(s) = slot {
+                    e.slot = s;
+                }
+            },
+            PumpSwapEvent::CreatePool(e) => {
+                e.signature = signature.to_string();
+                if let Some(s) = slot {
+                    e.slot = s;
+                }
+            },
+            PumpSwapEvent::Deposit(e) => {
+                e.signature = signature.to_string();
+                if let Some(s) = slot {
+                    e.slot = s;
+                }
+            },
+            PumpSwapEvent::Withdraw(e) => {
+                e.signature = signature.to_string();
+                if let Some(s) = slot {
+                    e.slot = s;
+                }
+            },
+            PumpSwapEvent::Disable(e) => {
+                e.signature = signature.to_string();
+                if let Some(s) = slot {
+                    e.slot = s;
+                }
+            },
+            PumpSwapEvent::UpdateAdmin(e) => {
+                e.signature = signature.to_string();
+                if let Some(s) = slot {
+                    e.slot = s;
+                }
+            },
+            PumpSwapEvent::UpdateFeeConfig(e) => {
+                e.signature = signature.to_string();
+                if let Some(s) = slot {
+                    e.slot = s;
+                }
+            },
+            _ => {}
+        }
+        callback(signature, event);
+    }
+    Ok(())
+}
+
+/// 获取当前时间戳
+fn current_timestamp() -> i64 {
+    SystemTime::now()
+        .duration_since(UNIX_EPOCH)
+        .expect("Time went backwards")
+        .as_secs() as i64
+}
+
+/// 从指令中解析PumpSwap指令
+pub fn parse_pumpswap_instruction(instruction: &CompiledInstruction, accounts: &[Pubkey]) -> Option<PumpSwapInstruction> {
+    if instruction.data.len() < 8 {
+        return None;
+    }
+
+    let discriminator = &instruction.data[..8];
+    let data = &instruction.data[8..];
+
+    match discriminator {
+        d if d == discriminators::BUY_IX => {
+            // buy指令参数: base_amount_out: u64, max_quote_amount_in: u64
+            // 账户顺序：pool, user, global_config, base_mint, quote_mint, user_base_token_account, 
+            // user_quote_token_account, pool_base_token_account, pool_quote_token_account, 
+            // protocol_fee_recipient, protocol_fee_recipient_token_account, ...
+            if data.len() < 16 || accounts.len() < 11 {
+                return None;
+            }
+            let base_amount_out = u64::from_le_bytes(data[0..8].try_into().ok()?);
+            let max_quote_amount_in = u64::from_le_bytes(data[8..16].try_into().ok()?);
+            
+            Some(PumpSwapInstruction::Buy(BuyEvent {
+                base_amount_out,
+                max_quote_amount_in,
+                pool: accounts[0],
+                user: accounts[1],
+                user_base_token_account: accounts[5],
+                user_quote_token_account: accounts[6],
+                protocol_fee_recipient: accounts[9],
+                protocol_fee_recipient_token_account: accounts[10],
+                timestamp: current_timestamp(),
+                ..Default::default()
+            }))
+        },
+        d if d == discriminators::SELL_IX => {
+            // sell指令参数: base_amount_in: u64, min_quote_amount_out: u64
+            // 账户顺序：pool, user, global_config, base_mint, quote_mint, user_base_token_account,
+            // user_quote_token_account, pool_base_token_account, pool_quote_token_account,
+            // protocol_fee_recipient, protocol_fee_recipient_token_account, ...
+            if data.len() < 16 || accounts.len() < 11 {
+                return None;
+            }
+            let base_amount_in = u64::from_le_bytes(data[0..8].try_into().ok()?);
+            let min_quote_amount_out = u64::from_le_bytes(data[8..16].try_into().ok()?);
+            
+            Some(PumpSwapInstruction::Sell(SellEvent {
+                base_amount_in,
+                min_quote_amount_out,
+                pool: accounts[0],
+                user: accounts[1],
+                user_base_token_account: accounts[5],
+                user_quote_token_account: accounts[6],
+                protocol_fee_recipient: accounts[9],
+                protocol_fee_recipient_token_account: accounts[10],
+                timestamp: current_timestamp(),
+                ..Default::default()
+            }))
+        },
+        d if d == discriminators::CREATE_POOL_IX => {
+            // create_pool指令参数: index: u16, base_amount_in: u64, quote_amount_in: u64
+            // 账户顺序：pool, global_config, creator, base_mint, quote_mint, lp_mint,
+            // user_base_token_account, user_quote_token_account, user_pool_token_account,
+            // pool_base_token_account, pool_quote_token_account, ...
+            if data.len() < 18 || accounts.len() < 11 {
+                return None;
+            }
+            let index = u16::from_le_bytes(data[0..2].try_into().ok()?);
+            let base_amount_in = u64::from_le_bytes(data[2..10].try_into().ok()?);
+            let quote_amount_in = u64::from_le_bytes(data[10..18].try_into().ok()?);
+            
+            Some(PumpSwapInstruction::CreatePool(CreatePoolEvent {
+                index,
+                base_amount_in,
+                quote_amount_in,
+                pool: accounts[0],
+                creator: accounts[2],
+                base_mint: accounts[3],
+                quote_mint: accounts[4],
+                lp_mint: accounts[5],
+                user_base_token_account: accounts[6],
+                user_quote_token_account: accounts[7],
+                timestamp: current_timestamp(),
+                ..Default::default()
+            }))
+        },
+        d if d == discriminators::DEPOSIT_IX => {
+            // deposit指令参数: lp_token_amount_out: u64, max_base_amount_in: u64, max_quote_amount_in: u64
+            // 账户顺序：pool, global_config, user, base_mint, quote_mint, lp_mint,
+            // user_base_token_account, user_quote_token_account, user_pool_token_account,
+            // pool_base_token_account, pool_quote_token_account, ...
+            if data.len() < 24 || accounts.len() < 11 {
+                return None;
+            }
+            let lp_token_amount_out = u64::from_le_bytes(data[0..8].try_into().ok()?);
+            let max_base_amount_in = u64::from_le_bytes(data[8..16].try_into().ok()?);
+            let max_quote_amount_in = u64::from_le_bytes(data[16..24].try_into().ok()?);
+            
+            Some(PumpSwapInstruction::Deposit(DepositEvent {
+                lp_token_amount_out,
+                max_base_amount_in,
+                max_quote_amount_in,
+                pool: accounts[0],
+                user: accounts[2],
+                user_base_token_account: accounts[6],
+                user_quote_token_account: accounts[7],
+                user_pool_token_account: accounts[8],
+                timestamp: current_timestamp(),
+                ..Default::default()
+            }))
+        },
+        d if d == discriminators::WITHDRAW_IX => {
+            // withdraw指令参数: lp_token_amount_in: u64, min_base_amount_out: u64, min_quote_amount_out: u64
+            // 账户顺序：pool, global_config, user, base_mint, quote_mint, lp_mint,
+            // user_base_token_account, user_quote_token_account, user_pool_token_account,
+            // pool_base_token_account, pool_quote_token_account, ...
+            if data.len() < 24 || accounts.len() < 11 {
+                return None;
+            }
+            let lp_token_amount_in = u64::from_le_bytes(data[0..8].try_into().ok()?);
+            let min_base_amount_out = u64::from_le_bytes(data[8..16].try_into().ok()?);
+            let min_quote_amount_out = u64::from_le_bytes(data[16..24].try_into().ok()?);
+            
+            Some(PumpSwapInstruction::Withdraw(WithdrawEvent {
+                lp_token_amount_in,
+                min_base_amount_out,
+                min_quote_amount_out,
+                pool: accounts[0],
+                user: accounts[2],
+                user_base_token_account: accounts[6],
+                user_quote_token_account: accounts[7],
+                user_pool_token_account: accounts[8],
+                timestamp: current_timestamp(),
+                ..Default::default()
+            }))
+        },
+        d if d == discriminators::DISABLE_IX => {
+            // disable指令参数: disable_create_pool: bool, disable_deposit: bool, disable_withdraw: bool, disable_buy: bool, disable_sell: bool
+            // 账户顺序：admin, global_config, event_authority, program
+            if data.len() < 5 || accounts.len() < 2 {
+                return None;
+            }
+            let disable_create_pool = data[0] != 0;
+            let disable_deposit = data[1] != 0;
+            let disable_withdraw = data[2] != 0;
+            let disable_buy = data[3] != 0;
+            let disable_sell = data[4] != 0;
+            
+            Some(PumpSwapInstruction::Disable(DisableEvent {
+                disable_create_pool,
+                disable_deposit,
+                disable_withdraw,
+                disable_buy,
+                disable_sell,
+                admin: accounts[0],
+                timestamp: current_timestamp(),
+                ..Default::default()
+            }))
+        },
+        d if d == discriminators::UPDATE_ADMIN_IX => {
+            // update_admin指令参数: 无
+            // 账户顺序：admin, global_config, new_admin, event_authority, program
+            if accounts.len() < 3 {
+                return None;
+            }
+            Some(PumpSwapInstruction::UpdateAdmin(UpdateAdminEvent {
+                old_admin: accounts[0],
+                new_admin: accounts[2],
+                timestamp: current_timestamp(),
+                ..Default::default()
+            }))
+        },
+        d if d == discriminators::UPDATE_FEE_CONFIG_IX => {
+            // update_fee_config指令参数: lp_fee_basis_points: u64, protocol_fee_basis_points: u64, protocol_fee_recipients: [pubkey; 8]
+            // 账户顺序：admin, global_config, event_authority, program
+            if data.len() < 272 || accounts.len() < 2 { // 8 + 8 + 32*8 = 272 bytes
+                return None;
+            }
+            let lp_fee_basis_points = u64::from_le_bytes(data[0..8].try_into().ok()?);
+            let protocol_fee_basis_points = u64::from_le_bytes(data[8..16].try_into().ok()?);
+            
+            let mut protocol_fee_recipients = [Pubkey::default(); 8];
+            for i in 0..8 {
+                let start = 16 + i * 32;
+                let end = start + 32;
+                if let Ok(pubkey_bytes) = data[start..end].try_into() {
+                    protocol_fee_recipients[i] = Pubkey::new_from_array(pubkey_bytes);
+                }
+            }
+            
+            Some(PumpSwapInstruction::UpdateFeeConfig(UpdateFeeConfigEvent {
+                admin: accounts[0],
+                new_lp_fee_basis_points: lp_fee_basis_points,
+                new_protocol_fee_basis_points: protocol_fee_basis_points,
+                new_protocol_fee_recipients: protocol_fee_recipients,
+                timestamp: current_timestamp(),
+                ..Default::default()
+            }))
+        },
+        _ => Some(PumpSwapInstruction::Other),
+    }
+}