use super::util::cluster::{cluster_api_url, Cluster};
use solana_sdk::pubkey::Pubkey;
use std::str::FromStr;
use web3_rs::core::{Provider, RcpClient};
use web3_rs::solana::rpc::*;

pub async fn get_token_accounts_by_owner(
    cluster: &Cluster,
    wallet_pubkey_str: &str,
    mint_pubkey_str: &str,
) -> Vec<TokenAccountsByOwnerResponseValue> {
    let wallet_pubkey = Pubkey::from_str(wallet_pubkey_str).unwrap();
    let mint_pubkey = Pubkey::from_str(mint_pubkey_str).unwrap();

    let api_url = cluster_api_url(cluster);
    let provider = Provider::new_http_provider(api_url.into());
    let client = RcpClient::new(provider);
    let request = GetTokenAccountsByOwnerRequest::new_mint(wallet_pubkey, mint_pubkey).into();

    match client.send(request).await {
        Ok(response) => {
            let result = GetTokenAccountsByOwnerResponse::from(response);
            log::info!("result:{:?}", result);
            let accounts = result.value.unwrap();
            log::info!("accounts:{:?}", accounts);

            accounts
        }
        Err(error) => {
            log::error!("{:?}", error);

            vec![]
        }
    }
}

#[cfg(test)]
#[tokio::test]
async fn test_get_token_accounts_by_owner() {
    let wallet_pubkey_str = "9K9RDUPvRfcVmHnoThUGkdR2bfQwa9oH1bs8RsmR2fjc";
    let mint_pubkey_str = "8YLbPb3TTEUL36gUyUiSx3ZffzaeqPoA1BkccyxTdn6i";

    let accounts: Vec<TokenAccountsByOwnerResponseValue> =
        get_token_accounts_by_owner(&Cluster::Devnet, wallet_pubkey_str, mint_pubkey_str).await;

    println!("accounts: {:?}", accounts);
    assert_ne!(Some(accounts).is_some(), true);
}