use crate::common::SolanaRpcClient;
use solana_sdk::pubkey::Pubkey;
#[derive(Clone)]
pub struct RaydiumCpmmParams {
pub pool_state: Pubkey,
pub amm_config: Pubkey,
pub base_mint: Pubkey,
pub quote_mint: Pubkey,
pub base_reserve: u64,
pub quote_reserve: u64,
pub base_vault: Pubkey,
pub quote_vault: Pubkey,
pub base_token_program: Pubkey,
pub quote_token_program: Pubkey,
pub observation_state: Pubkey,
}
impl RaydiumCpmmParams {
pub fn from_trade(
pool_state: Pubkey,
amm_config: Pubkey,
input_token_mint: Pubkey,
output_token_mint: Pubkey,
input_vault: Pubkey,
output_vault: Pubkey,
input_token_program: Pubkey,
output_token_program: Pubkey,
observation_state: Pubkey,
base_reserve: u64,
quote_reserve: u64,
) -> Self {
Self {
pool_state: pool_state,
amm_config: amm_config,
base_mint: input_token_mint,
quote_mint: output_token_mint,
base_reserve: base_reserve,
quote_reserve: quote_reserve,
base_vault: input_vault,
quote_vault: output_vault,
base_token_program: input_token_program,
quote_token_program: output_token_program,
observation_state: observation_state,
}
}
pub async fn from_pool_address_by_rpc(
rpc: &SolanaRpcClient,
pool_address: &Pubkey,
) -> Result<Self, anyhow::Error> {
let pool =
crate::instruction::utils::raydium_cpmm::fetch_pool_state(rpc, pool_address).await?;
let (token0_balance, token1_balance) =
crate::instruction::utils::raydium_cpmm::get_pool_token_balances(
rpc,
pool_address,
&pool.token0_mint,
&pool.token1_mint,
)
.await?;
Ok(Self {
pool_state: *pool_address,
amm_config: pool.amm_config,
base_mint: pool.token0_mint,
quote_mint: pool.token1_mint,
base_reserve: token0_balance,
quote_reserve: token1_balance,
base_vault: pool.token0_vault,
quote_vault: pool.token1_vault,
base_token_program: pool.token0_program,
quote_token_program: pool.token1_program,
observation_state: pool.observation_key,
})
}
}