use crate::types::Bootstrapping;
use crate::types::CurveType;
use crate::types::Padding;
use crate::types::PartnerInfo;
use crate::types::PoolFees;
use crate::types::PoolType;
use borsh::BorshDeserialize;
use borsh::BorshSerialize;
use solana_program::pubkey::Pubkey;
#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, Eq, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct Pool {
pub discriminator: [u8; 8],
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub lp_mint: Pubkey,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub token_a_mint: Pubkey,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub token_b_mint: Pubkey,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub a_vault: Pubkey,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub b_vault: Pubkey,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub a_vault_lp: Pubkey,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub b_vault_lp: Pubkey,
pub a_vault_lp_bump: u8,
pub enabled: bool,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub protocol_token_a_fee: Pubkey,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub protocol_token_b_fee: Pubkey,
pub fee_last_updated_at: u64,
pub padding0: [u8; 24],
pub fees: PoolFees,
pub pool_type: PoolType,
#[cfg_attr(
feature = "serde",
serde(with = "serde_with::As::<serde_with::DisplayFromStr>")
)]
pub stake: Pubkey,
pub total_locked_lp: u64,
pub bootstrapping: Bootstrapping,
pub partner_info: PartnerInfo,
pub padding: Padding,
pub curve_type: CurveType,
}
impl Pool {
#[inline(always)]
pub fn from_bytes(data: &[u8]) -> Result<Self, std::io::Error> {
let mut data = data;
Self::deserialize(&mut data)
}
}
impl<'a> TryFrom<&solana_program::account_info::AccountInfo<'a>> for Pool {
type Error = std::io::Error;
fn try_from(
account_info: &solana_program::account_info::AccountInfo<'a>,
) -> Result<Self, Self::Error> {
let mut data: &[u8] = &(*account_info.data).borrow();
Self::deserialize(&mut data)
}
}
#[cfg(feature = "fetch")]
pub fn fetch_pool(
rpc: &solana_client::rpc_client::RpcClient,
address: &Pubkey,
) -> Result<super::DecodedAccount<Pool>, Error> {
let accounts = fetch_all_pool(rpc, vec![address])?;
Ok(accounts[0].clone())
}
#[cfg(feature = "fetch")]
pub fn fetch_all_pool(
rpc: &solana_client::rpc_client::RpcClient,
addresses: Vec<Pubkey>,
) -> Result<Vec<super::DecodedAccount<Pool>>, Error> {
let accounts = rpc.get_multiple_accounts(&addresses)?;
let mut decoded_accounts: Vec<super::DecodedAccount<Pool>> = Vec::new();
for i in 0..addresses.len() {
let address = addresses[i];
let account = accounts[i]
.as_ref()
.ok_or(format!("Account not found: {}", address))?;
let data = Pool::from_bytes(&account.data)?;
decoded_accounts.push(super::DecodedAccount {
address,
account: account.clone(),
data,
});
}
Ok(decoded_accounts)
}
#[cfg(feature = "fetch")]
pub fn fetch_maybe_pool(
rpc: &solana_client::rpc_client::RpcClient,
address: &Pubkey,
) -> Result<super::MaybeAccount<Pool>, Error> {
let accounts = fetch_all_maybe_pool(rpc, vec![address])?;
Ok(accounts[0].clone())
}
#[cfg(feature = "fetch")]
pub fn fetch_all_maybe_pool(
rpc: &solana_client::rpc_client::RpcClient,
addresses: Vec<Pubkey>,
) -> Result<Vec<super::MaybeAccount<Pool>>, Error> {
let accounts = rpc.get_multiple_accounts(&addresses)?;
let mut decoded_accounts: Vec<super::MaybeAccount<Pool>> = Vec::new();
for i in 0..addresses.len() {
let address = addresses[i];
if let Some(account) = accounts[i].as_ref() {
let data = Pool::from_bytes(&account.data)?;
decoded_accounts.push(super::MaybeAccount::Exists(super::DecodedAccount {
address,
account: account.clone(),
data,
}));
} else {
decoded_accounts.push(super::MaybeAccount::NotFound(address));
}
}
Ok(decoded_accounts)
}
#[cfg(feature = "anchor")]
impl anchor_lang::AccountDeserialize for Pool {
fn try_deserialize_unchecked(buf: &mut &[u8]) -> anchor_lang::Result<Self> {
Ok(Self::deserialize(buf)?)
}
}
#[cfg(feature = "anchor")]
impl anchor_lang::AccountSerialize for Pool {}
#[cfg(feature = "anchor")]
impl anchor_lang::Owner for Pool {
fn owner() -> Pubkey {
crate::AMM_ID
}
}
#[cfg(feature = "anchor-idl-build")]
impl anchor_lang::IdlBuild for Pool {}
#[cfg(feature = "anchor-idl-build")]
impl anchor_lang::Discriminator for Pool {
const DISCRIMINATOR: [u8; 8] = [0; 8];
}