use std::error::Error;
use solana_client::{
nonblocking::rpc_client::RpcClient,
rpc_filter::{Memcmp, RpcFilterType},
};
use solana_sdk::pubkey::Pubkey;
use super::fetch_decoded_program_accounts;
use crate::{generated::shared::DecodedAccount, VortexConfig};
pub const VORTEX_CONFIG_DISCRIMINATOR: &[u8] = &[225, 159, 49, 163, 164, 24, 106, 110];
#[derive(Debug, Clone)]
pub enum VortexConfigFilter {
FeeAuthority(Pubkey),
CollectProtocolFeesAuthority(Pubkey),
RewardEmissionsSuperAuthority(Pubkey),
DefaultProtocolFeeRate(u16),
}
impl From<VortexConfigFilter> for RpcFilterType {
fn from(val: VortexConfigFilter) -> Self {
match val {
VortexConfigFilter::FeeAuthority(address) => {
RpcFilterType::Memcmp(Memcmp::new_raw_bytes(8, address.to_bytes().to_vec()))
}
VortexConfigFilter::CollectProtocolFeesAuthority(address) => {
RpcFilterType::Memcmp(Memcmp::new_raw_bytes(40, address.to_bytes().to_vec()))
}
VortexConfigFilter::RewardEmissionsSuperAuthority(address) => {
RpcFilterType::Memcmp(Memcmp::new_raw_bytes(72, address.to_bytes().to_vec()))
}
VortexConfigFilter::DefaultProtocolFeeRate(fee_rate) => {
RpcFilterType::Memcmp(Memcmp::new_raw_bytes(104, fee_rate.to_le_bytes().to_vec()))
}
}
}
}
pub async fn fetch_all_vortex_config_with_filter(
rpc: &RpcClient,
filters: Vec<VortexConfigFilter>,
) -> Result<Vec<DecodedAccount<VortexConfig>>, Box<dyn Error>> {
let mut filters: Vec<RpcFilterType> = filters.into_iter().map(|filter| filter.into()).collect();
filters.push(RpcFilterType::Memcmp(Memcmp::new_raw_bytes(
0,
VORTEX_CONFIG_DISCRIMINATOR.to_vec(),
)));
fetch_decoded_program_accounts(rpc, filters).await
}