use crate::{
    ethereum_helper::{tc_web3_client, InMemoryWallet},
    htlc_harness::{new_account, timestamp::Timestamp, SECRET_HASH},
    parity_client::ParityClient,
};
use blockchain_contracts::ethereum::herc20::Htlc;
use blockchain_contracts::ethereum::TokenQuantity;
use std::sync::Arc;
use testcontainers::{images::parity_parity::ParityEthereum, Container, Docker};
use web3::{
    transports::EventLoopHandle,
    types::{Address, U256},
};

#[derive(Debug, Clone)]
pub struct Erc20HarnessParams {
    pub alice_initial_wei: U256,
    pub htlc_refund_timestamp: Timestamp,
    pub htlc_secret_hash: [u8; 32],
    pub alice_initial_tokens: U256,
    pub htlc_token_value: U256,
}

impl Default for Erc20HarnessParams {
    fn default() -> Self {
        Self {
            alice_initial_wei: "1000000000000000000".into(),
            htlc_refund_timestamp: Timestamp::now().plus(10),
            htlc_secret_hash: SECRET_HASH,
            alice_initial_tokens: U256::from(1000),
            htlc_token_value: U256::from(400),
        }
    }
}

impl Erc20HarnessParams {
    pub fn with_secret_hash(self, secret_hash: [u8; 32]) -> Self {
        Self {
            htlc_secret_hash: secret_hash,
            ..self
        }
    }
}

pub fn erc20_harness<D: Docker>(
    docker: &D,
    params: Erc20HarnessParams,
) -> (
    Address,
    Address,
    Address,
    Address,
    U256,
    ParityClient,
    EventLoopHandle,
    Container<'_, D, ParityEthereum>,
) {
    let _ = pretty_env_logger::try_init();

    let (alice_secret_key, alice) =
        new_account("63be4b0d638d44b5fee5b050ab0beeeae7b68cde3d829a3321f8009cdd76b992");
    let (_, bob) = new_account("f8218ebf6e2626bd1415c18321496e0c5725f0e1d774c7c2eab69f7650ad6e82");

    let container = docker.run(ParityEthereum::default());

    let (event_loop, web3) = tc_web3_client::new(&container);
    let web3 = Arc::new(web3);

    let alice_client =
        ParityClient::new(Arc::new(InMemoryWallet::new(alice_secret_key, 1)), web3, 0);

    alice_client.give_eth_to(alice, params.alice_initial_wei);

    let token_contract = alice_client.deploy_erc20_token_contract();

    alice_client.mint_tokens(token_contract, params.alice_initial_tokens, alice);

    let erc20_htlc = Htlc::new(
        params.htlc_refund_timestamp.into(),
        blockchain_contracts::ethereum::Address(alice.into()),
        blockchain_contracts::ethereum::Address(bob.into()),
        params.htlc_secret_hash,
        blockchain_contracts::ethereum::Address(token_contract.into()),
        TokenQuantity(params.htlc_token_value.into()),
    );

    let tx_id = alice_client.deploy_htlc(
        erc20_htlc.into(),
        U256::from(0),
        Htlc::deploy_tx_gas_limit().into(),
    );

    let transaction_receipt = alice_client.receipt(tx_id);
    log::debug!("used gas ERC20 deploy {:?}", transaction_receipt.gas_used);

    (
        alice,
        bob,
        alice_client.get_contract_address(tx_id),
        token_contract,
        params.htlc_token_value,
        alice_client,
        event_loop,
        container,
    )
}