pub use crate::miner::Connection;
use async_std::sync::{Arc, RwLock};
use log::info;
use std::collections::HashMap;
use std::net::SocketAddr;
use stratum_types::traits::{StratumManager, StratumParams};
use stratum_types::Result;

#[derive(Default)]
pub struct MinerList<SM: StratumManager> {
    pub miners: RwLock<HashMap<SocketAddr, Arc<Connection<SM>>>>,
}

impl<SM> MinerList<SM>
where
    SM: StratumManager,
{
    pub fn new() -> Self {
        MinerList {
            miners: RwLock::new(HashMap::new()),
        }
    }

    pub async fn add_miner(&self, addr: SocketAddr, miner: Arc<Connection<SM>>) -> Result<()> {
        self.miners.write().await.insert(addr, miner);
        Ok(())
    }

    pub async fn remove_miner(&self, addr: SocketAddr) -> Result<()> {
        self.miners.write().await.remove(&addr);
        Ok(())
    }

    pub async fn broadcast_new_job(
        &self,
        _job: <SM::StratumParams as StratumParams>::Notify,
    ) -> Result<()> {
        let miners = self.miners.read().await;

        info!("Broadcasting new work to miners.");

        for miner in miners.values() {
            // miner.send().await;
            //@todo might be more efficient to pass it down to the miners specifically rather than
            //put strain on the data_provider, but let's test this.
            miner.send_work().await?;
        }

        Ok(())
    }
}