use contract_wrapper::eth_network::EthNetwork;
use dotenv::dotenv;
use serde::Deserialize;
use std::env;
use std::io::Read;
use std::path::PathBuf;

#[derive(Deserialize, Debug, Clone)]
pub struct ConfigForTests {
    pub beacon_endpoint: String,
    pub eth1_endpoint: String,
    pub path_to_current_sync_committee: String,
    pub path_to_next_sync_committee: String,
    pub path_to_execution_blocks_headers: String,
    pub path_to_light_client_updates: String,
    pub path_to_attested_state: String,
    pub network_name: EthNetwork,
    pub first_slot: u64,
    pub slot_without_block: u64,
    pub wasm_filepath: String,
    pub right_bound_in_slot_search: u64,
    pub left_empty_slot: u64,
    pub right_empty_slot: u64,
    pub finalized_slot_before_new_period: u64,
    pub slot_without_block_2: u64,
}

impl ConfigForTests {
    pub fn load_from_toml(path: PathBuf) -> Self {
        let mut config = std::fs::File::open(path).unwrap();
        let mut content = String::new();
        config.read_to_string(&mut content).unwrap();
        let mut config: Self = toml::from_str(content.as_str()).unwrap();

        dotenv().ok();

        let api_key_string = env::var("ETH1_INFURA_API_KEY").unwrap();
        config.eth1_endpoint = config
            .eth1_endpoint
            .replace("ETH1_INFURA_API_KEY", &api_key_string);

        config
    }
}