122 lines
3.9 KiB
Rust
122 lines
3.9 KiB
Rust
use mango_v4::accounts_zerocopy::*;
|
|
use mango_v4::state::{Bank, MangoAccountValue, MintInfo, PerpMarket, TokenIndex};
|
|
|
|
use anyhow::Context;
|
|
use fixed::types::I80F48;
|
|
|
|
use solana_sdk::account::AccountSharedData;
|
|
use solana_sdk::pubkey::Pubkey;
|
|
|
|
pub use mango_v4_client::snapshot_source::is_mango_account;
|
|
use mango_v4_client::{chain_data, MangoClient};
|
|
|
|
pub fn is_mango_bank<'a>(account: &'a AccountSharedData, group_id: &Pubkey) -> Option<&'a Bank> {
|
|
let bank = account.load::<Bank>().ok()?;
|
|
if bank.group != *group_id {
|
|
return None;
|
|
}
|
|
Some(bank)
|
|
}
|
|
|
|
pub fn is_mint_info<'a>(account: &'a AccountSharedData, group_id: &Pubkey) -> Option<&'a MintInfo> {
|
|
let mint_info = account.load::<MintInfo>().ok()?;
|
|
if mint_info.group != *group_id {
|
|
return None;
|
|
}
|
|
Some(mint_info)
|
|
}
|
|
|
|
pub fn is_perp_market<'a>(
|
|
account: &'a AccountSharedData,
|
|
group_id: &Pubkey,
|
|
) -> Option<&'a PerpMarket> {
|
|
let perp_market = account.load::<PerpMarket>().ok()?;
|
|
if perp_market.group != *group_id {
|
|
return None;
|
|
}
|
|
Some(perp_market)
|
|
}
|
|
|
|
/// Convenience wrapper for getting max swap amounts for a token pair
|
|
///
|
|
/// This applies net borrow and deposit limits, which is useful for true swaps.
|
|
pub fn max_swap_source_with_limits(
|
|
client: &MangoClient,
|
|
account_fetcher: &chain_data::AccountFetcher,
|
|
account: &MangoAccountValue,
|
|
source: TokenIndex,
|
|
target: TokenIndex,
|
|
price: I80F48,
|
|
min_health_ratio: I80F48,
|
|
) -> anyhow::Result<I80F48> {
|
|
let mut account = account.clone();
|
|
|
|
// Ensure the tokens are activated, so they appear in the health cache and
|
|
// max_swap_source() will work.
|
|
account.ensure_token_position(source)?;
|
|
account.ensure_token_position(target)?;
|
|
|
|
let health_cache =
|
|
mango_v4_client::health_cache::new_sync(&client.context, account_fetcher, &account)
|
|
.context("health cache")?;
|
|
|
|
let source_bank: Bank = account_fetcher.fetch(&client.context.token(source).first_bank())?;
|
|
let target_bank: Bank = account_fetcher.fetch(&client.context.token(target).first_bank())?;
|
|
|
|
let source_price = health_cache.token_info(source).unwrap().prices.oracle;
|
|
|
|
let amount = health_cache
|
|
.max_swap_source_for_health_ratio_with_limits(
|
|
&account,
|
|
&source_bank,
|
|
source_price,
|
|
&target_bank,
|
|
price,
|
|
min_health_ratio,
|
|
)
|
|
.context("getting max_swap_source")?;
|
|
Ok(amount)
|
|
}
|
|
|
|
/// Convenience wrapper for getting max swap amounts for a token pair
|
|
///
|
|
/// This is useful for liquidations, which don't increase deposits or net borrows.
|
|
/// Tcs execution can also increase deposits/net borrows.
|
|
pub fn max_swap_source_ignoring_limits(
|
|
client: &MangoClient,
|
|
account_fetcher: &chain_data::AccountFetcher,
|
|
account: &MangoAccountValue,
|
|
source: TokenIndex,
|
|
target: TokenIndex,
|
|
price: I80F48,
|
|
min_health_ratio: I80F48,
|
|
) -> anyhow::Result<I80F48> {
|
|
let mut account = account.clone();
|
|
|
|
// Ensure the tokens are activated, so they appear in the health cache and
|
|
// max_swap_source() will work.
|
|
account.ensure_token_position(source)?;
|
|
account.ensure_token_position(target)?;
|
|
|
|
let health_cache =
|
|
mango_v4_client::health_cache::new_sync(&client.context, account_fetcher, &account)
|
|
.context("health cache")?;
|
|
|
|
let source_bank: Bank = account_fetcher.fetch(&client.context.token(source).first_bank())?;
|
|
let target_bank: Bank = account_fetcher.fetch(&client.context.token(target).first_bank())?;
|
|
|
|
let source_price = health_cache.token_info(source).unwrap().prices.oracle;
|
|
|
|
let amount = health_cache
|
|
.max_swap_source_for_health_ratio_ignoring_limits(
|
|
&account,
|
|
&source_bank,
|
|
source_price,
|
|
&target_bank,
|
|
price,
|
|
min_health_ratio,
|
|
)
|
|
.context("getting max_swap_source")?;
|
|
Ok(amount)
|
|
}
|