Skip to main content

defituna_client/gpa/
referral.rs

1use crate::accounts::{Referral, REFERRAL_DISCRIMINATOR};
2use crate::gpa::utils::fetch_decoded_program_accounts;
3use crate::DecodedAccount;
4use solana_client::rpc_client::RpcClient;
5use solana_client::rpc_filter::{Memcmp, RpcFilterType};
6use solana_pubkey::Pubkey;
7use std::error::Error;
8
9#[derive(Debug, Clone)]
10pub enum ReferralFilter {
11    Authority(Pubkey),
12    ReferralId(u32),
13}
14
15impl From<ReferralFilter> for RpcFilterType {
16    fn from(val: ReferralFilter) -> Self {
17        match val {
18            ReferralFilter::Authority(address) => RpcFilterType::Memcmp(Memcmp::new_base58_encoded(8, &address.to_bytes())),
19            ReferralFilter::ReferralId(referral_id) => RpcFilterType::Memcmp(Memcmp::new_base58_encoded(48, &referral_id.to_le_bytes())),
20        }
21    }
22}
23
24pub fn fetch_all_referral_with_filter(rpc: &RpcClient, filters: Vec<ReferralFilter>) -> Result<Vec<DecodedAccount<Referral>>, Box<dyn Error>> {
25    let mut filters: Vec<RpcFilterType> = filters.into_iter().map(|filter| filter.into()).collect();
26    filters.push(RpcFilterType::Memcmp(Memcmp::new_base58_encoded(0, &REFERRAL_DISCRIMINATOR)));
27    fetch_decoded_program_accounts(rpc, filters)
28}