ysq
b891b2bc27
- Consolidate separate buy/sell modules into unified trading interface - Remove protocol-specific buy/sell files (bonk, pumpfun, pumpswap) - Add new trading constants and utility functions - Simplify API with unified buy/sell methods supporting multiple protocols - Enhance documentation with comprehensive examples and usage guides - Add balance checking and token account management utilities - Improve code organization and maintainability
56 lines
2.0 KiB
Rust
Executable File
56 lines
2.0 KiB
Rust
Executable File
use solana_sdk::pubkey::Pubkey;
|
|
use crate::constants;
|
|
|
|
pub fn get_amount_out(
|
|
amount_in: u64,
|
|
protocol_fee_rate: u128,
|
|
platform_fee_rate: u128,
|
|
share_fee_rate: u128,
|
|
virtual_base: u128,
|
|
virtual_quote: u128,
|
|
real_base_before: u128,
|
|
real_quote_before: u128,
|
|
slippage_basis_points: u128,
|
|
) -> u64 {
|
|
let amount_in_u128 = amount_in as u128;
|
|
let protocol_fee = (amount_in_u128 * protocol_fee_rate / 10000) as u128;
|
|
let platform_fee = (amount_in_u128 * platform_fee_rate / 10000) as u128;
|
|
let share_fee = (amount_in_u128 * share_fee_rate / 10000) as u128;
|
|
let amount_in_net = amount_in_u128
|
|
.checked_sub(protocol_fee)
|
|
.unwrap()
|
|
.checked_sub(platform_fee)
|
|
.unwrap()
|
|
.checked_sub(share_fee)
|
|
.unwrap();
|
|
let input_reserve = virtual_quote.checked_add(real_quote_before).unwrap();
|
|
let output_reserve = virtual_base.checked_sub(real_base_before).unwrap();
|
|
let numerator = amount_in_net.checked_mul(output_reserve).unwrap();
|
|
let denominator = input_reserve.checked_add(amount_in_net).unwrap();
|
|
let mut amount_out = numerator.checked_div(denominator).unwrap();
|
|
|
|
amount_out = amount_out - (amount_out * slippage_basis_points) / 10000;
|
|
amount_out as u64
|
|
}
|
|
|
|
pub fn get_pool_pda(base_mint: &Pubkey, quote_mint: &Pubkey) -> Option<Pubkey> {
|
|
let seeds: &[&[u8]; 3] = &[
|
|
constants::bonk::seeds::POOL_SEED,
|
|
base_mint.as_ref(),
|
|
quote_mint.as_ref(),
|
|
];
|
|
let program_id: &Pubkey = &constants::bonk::accounts::BONK;
|
|
let pda: Option<(Pubkey, u8)> = Pubkey::try_find_program_address(seeds, program_id);
|
|
pda.map(|pubkey| pubkey.0)
|
|
}
|
|
|
|
pub fn get_vault_pda(pool_state: &Pubkey, mint: &Pubkey) -> Option<Pubkey> {
|
|
let seeds: &[&[u8]; 3] = &[
|
|
constants::bonk::seeds::POOL_VAULT_SEED,
|
|
pool_state.as_ref(),
|
|
mint.as_ref(),
|
|
];
|
|
let program_id: &Pubkey = &constants::bonk::accounts::BONK;
|
|
let pda: Option<(Pubkey, u8)> = Pubkey::try_find_program_address(seeds, program_id);
|
|
pda.map(|pubkey| pubkey.0)
|
|
} |