mod common;

use anyhow::Result;
use claim::assert_gt;
use common::TestContext;
use test_context::test_context;
use zilliqa_rs::{
    core::parse_zil,
    core::CreateTransactionResponse,
    middlewares::Middleware,
    providers::{Http, Provider},
    signers::LocalWallet,
    transaction::{TransactionBuilder, Version},
    Error,
};

#[test_context(TestContext)]
#[tokio::test]
async fn send_transaction(ctx: &TestContext) -> Result<()> {
    let provider = ctx.provider();

    let sender_balance = provider.get_balance(&ctx.wallet.address).await?;

    assert_gt!(sender_balance.balance, parse_zil("0.1")?);

    let receiver = LocalWallet::create_random()?;
    let tx = TransactionBuilder::default()
        .to_address(receiver.address.clone())
        .amount(parse_zil("0.1")?)
        .gas_price(parse_zil("0.002")?)
        .gas_limit(50u64)
        .build();

    let tx = provider.send_transaction(tx).await?;
    tx.confirm().await?;

    let res = provider.get_balance(&receiver.address).await?;

    assert_eq!(res.balance, parse_zil("0.1")?);

    Ok(())
}

#[tokio::test]
async fn if_version_is_not_set_create_transaction_should_return_error() -> Result<()> {
    const END_POINT: &str = "http://localhost:5555";

    let wallet = "d96e9eb5b782a80ea153c937fa83e5948485fbfc8b7e7c069d7b914dbc350aba".parse::<LocalWallet>()?;

    // Chain ID is not set, so version
    let provider = Provider::<Http>::try_from(END_POINT)?.with_signer(wallet.clone());

    let receiver = LocalWallet::create_random()?;
    let tx = TransactionBuilder::default()
        .to_address(receiver.address)
        .amount(parse_zil("2")?)
        .gas_price(parse_zil("0.002")?)
        .gas_limit(50u64)
        .build();

    let res: Result<CreateTransactionResponse, _> = provider.send_transaction_without_confirm(tx).await;
    assert!(matches!(
        res,
        Err(Error::InvalidVersionIsSetForTransaction(v)
        ) if v == Version::new(0)
    ));

    Ok(())
}

#[test_context(TestContext)]
#[tokio::test]
async fn send_zil_using_pay_function(ctx: &TestContext) -> Result<()> {
    let provider = ctx.provider();
    let sender_balance = provider.get_balance(&ctx.wallet.address).await?;

    assert_gt!(sender_balance.balance, parse_zil("0.1")?);

    let receiver = LocalWallet::create_random()?;

    let tx = TransactionBuilder::default()
        .pay(parse_zil("0.1")?, receiver.address.clone())
        .build();

    provider
        .send_transaction_without_confirm::<CreateTransactionResponse>(tx)
        .await?;

    let res = provider.get_balance(&receiver.address).await?;

    assert_eq!(res.balance, parse_zil("0.1")?);

    Ok(())
}

#[ignore]
#[tokio::test]
async fn get_transaction_receipt() -> Result<()> {
    const END_POINT: &str = "https://api.devnet.zilliqa.com";

    let wallet = "0x8ce73c46c1b8d09171319cf1498e538bbd151a4b65d6688cccdee1473d626c49".parse::<LocalWallet>()?;
    let provider = Provider::<Http>::try_from(END_POINT)?
        .with_chain_id(617)
        .with_signer(wallet.clone());

    let sender_balance = provider.get_balance(&wallet.address).await?;
    println!("{sender_balance:?}");

    let receiver = LocalWallet::create_random()?;
    let amount = parse_zil("0.2")?;

    let tx = TransactionBuilder::default().pay(amount, receiver.address.clone()).build();
    let tx = provider.send_transaction(tx).await?;

    let res = tx.confirm().await?;
    println!("{:?}", res.receipt);

    let sender_balance = provider.get_balance(&receiver.address).await?;
    assert_eq!(sender_balance.balance, parse_zil("0.2")?);

    Ok(())
}

#[tokio::test]
async fn send_zil_with_empty_data_code() -> Result<()> {
    const END_POINT: &str = "https://zilliqa-isolated-server.zilliqa.com/";

    let wallet = "d96e9eb5b782a80ea153c937fa83e5948485fbfc8b7e7c069d7b914dbc350aba".parse::<LocalWallet>()?;

    let provider = Provider::<Http>::try_from(END_POINT)?
        .with_chain_id(222)
        .with_signer(wallet.clone());

    let receiver = LocalWallet::create_random()?;

    let tx = TransactionBuilder::default()
        .data("".to_string())
        .code("".to_string())
        .pay(parse_zil("0.1")?, receiver.address.clone())
        .build();

    provider
        .send_transaction_without_confirm::<CreateTransactionResponse>(tx)
        .await?;

    let res = provider.get_balance(&receiver.address).await?;

    assert_eq!(res.balance, parse_zil("0.1")?);
    Ok(())
}