use std::sync::Arc;

use bench_utils::helpers::BenchHelper;
use lite_rpc::DEFAULT_LITE_RPC_ADDR;
use log::info;
use solana_client::{nonblocking::rpc_client::RpcClient, rpc_client::SerializableTransaction};
use solana_sdk::{commitment_config::CommitmentConfig, native_token::LAMPORTS_PER_SOL};

use simplelog::*;

const AMOUNT: usize = 5;

#[tokio::test]
async fn send_and_confirm_txs_get_signature_statuses() {
    TermLogger::init(
        LevelFilter::Info,
        Config::default(),
        TerminalMode::Mixed,
        ColorChoice::Auto,
    )
    .unwrap();

    let rpc_client = Arc::new(RpcClient::new(DEFAULT_LITE_RPC_ADDR.to_string()));
    let bench_helper = BenchHelper::new(rpc_client.clone());

    let funded_payer = bench_helper
        .new_funded_payer(LAMPORTS_PER_SOL * 2)
        .await
        .unwrap();

    let txs = bench_helper
        .generate_txs(AMOUNT, &funded_payer)
        .await
        .unwrap();

    info!("Sending and Confirming {AMOUNT} tx(s)");

    for tx in &txs {
        rpc_client.send_transaction(tx).await.unwrap();
        info!("Tx {}", tx.get_signature());
    }

    for tx in &txs {
        let sig = tx.get_signature();
        info!("Confirming {sig}");
        bench_helper
            .wait_till_signature_status(sig, CommitmentConfig::confirmed())
            .await
            .unwrap();
    }

    info!("Sent and Confirmed {AMOUNT} tx(s)");
}

#[tokio::test]
async fn send_and_confirm_tx_rpc_client() {
    let rpc_client = Arc::new(RpcClient::new(DEFAULT_LITE_RPC_ADDR.to_string()));
    let bench_helper = BenchHelper::new(rpc_client.clone());

    let funded_payer = bench_helper
        .new_funded_payer(LAMPORTS_PER_SOL * 2)
        .await
        .unwrap();

    let tx = &bench_helper.generate_txs(1, &funded_payer).await.unwrap()[0];
    let sig = tx.get_signature();

    bench_helper.send_and_confirm_transaction(tx).await.unwrap();

    info!("Sent and Confirmed {sig}");
}