use crate::bitcoin::{CancelTimelock, PunishTimelock};
use std::time::Duration;

#[derive(Debug, Copy, Clone)]
pub struct ExecutionParams {
    pub bob_time_to_act: Duration,
    pub bitcoin_finality_confirmations: u32,
    pub bitcoin_avg_block_time: Duration,
    pub monero_finality_confirmations: u32,
    pub bitcoin_cancel_timelock: CancelTimelock,
    pub bitcoin_punish_timelock: PunishTimelock,
}

pub trait GetExecutionParams {
    fn get_execution_params() -> ExecutionParams;
}

#[derive(Clone, Copy)]
pub struct Mainnet;

#[derive(Clone, Copy)]
pub struct Testnet;

#[derive(Clone, Copy)]
pub struct Regtest;

impl GetExecutionParams for Mainnet {
    fn get_execution_params() -> ExecutionParams {
        ExecutionParams {
            bob_time_to_act: Duration::from_secs(10 * 60),
            bitcoin_finality_confirmations: 3,
            bitcoin_avg_block_time: Duration::from_secs(10 * 60),
            monero_finality_confirmations: 15,
            bitcoin_cancel_timelock: CancelTimelock::new(72),
            bitcoin_punish_timelock: PunishTimelock::new(72),
        }
    }
}

impl GetExecutionParams for Testnet {
    fn get_execution_params() -> ExecutionParams {
        ExecutionParams {
            bob_time_to_act: Duration::from_secs(60 * 60),
            bitcoin_finality_confirmations: 1,
            bitcoin_avg_block_time: Duration::from_secs(5 * 60),
            monero_finality_confirmations: 10,
            bitcoin_cancel_timelock: CancelTimelock::new(12),
            bitcoin_punish_timelock: PunishTimelock::new(6),
        }
    }
}

impl GetExecutionParams for Regtest {
    fn get_execution_params() -> ExecutionParams {
        ExecutionParams {
            bob_time_to_act: Duration::from_secs(30),
            bitcoin_finality_confirmations: 1,
            bitcoin_avg_block_time: Duration::from_secs(5),
            monero_finality_confirmations: 10,
            bitcoin_cancel_timelock: CancelTimelock::new(100),
            bitcoin_punish_timelock: PunishTimelock::new(50),
        }
    }
}