bwt 0.2.4

use std::collections::hash_map::{Entry, HashMap};
use std::time::Duration;
use std::{sync::mpsc, thread};

use serde_json::Value;

use bitcoin::Txid;

#[macro_use]
mod macros;

pub mod auth;
pub mod banner;
pub mod bitcoincore_ext;
pub mod descriptor;
pub mod progress;
pub mod whitepaper;
pub mod xpub;

pub use bitcoincore_ext::RpcApiExt;

const VSIZE_BIN_WIDTH: u32 = 50_000; // vbytes

// Make the fee histogram out of a list of `getrawmempool true` entries
pub fn make_fee_histogram(mempool_entries: HashMap<Txid, Value>) -> Vec<(f32, u32)> {
    let mut entries: Vec<_> = mempool_entries
        .into_iter()
        .map(|(_, entry)| {
            let vsize = entry["vsize"]
                .as_u64()
                .or_else(|| entry["size"].as_u64())
                .unwrap(); // bitcoind is borked if this fails
            let fee = entry["fee"].as_f64().unwrap();
            let feerate = fee as f32 / vsize as f32 * 100_000_000f32;
            (vsize as u32, feerate)
        })
        .collect();

    // XXX should take unconfirmed parents feerates into account

    entries.sort_unstable_by(|a, b| a.1.partial_cmp(&b.1).unwrap());

    let mut histogram = vec![];
    let mut bin_size = 0;
    let mut last_feerate = 0.0;

    for (vsize, feerate) in entries.into_iter().rev() {
        if bin_size > VSIZE_BIN_WIDTH && (last_feerate - feerate).abs() > f32::EPSILON {
            // vsize of transactions paying >= last_feerate
            histogram.push((last_feerate, bin_size));
            bin_size = 0;
        }
        bin_size += vsize;
        last_feerate = feerate;
    }

    if bin_size > 0 {
        histogram.push((last_feerate, bin_size));
    }

    histogram
}

pub fn remove_if<K, V>(hm: &mut HashMap<K, V>, key: K, predicate: impl Fn(&mut V) -> bool) -> bool
where
    K: Eq + std::hash::Hash,
{
    if let Entry::Occupied(mut entry) = hm.entry(key) {
        if predicate(entry.get_mut()) {
            entry.remove_entry();
        }
        true
    } else {
        false
    }
}

pub fn fmt_date(unix: u64) -> String {
    let dt = chrono::NaiveDateTime::from_timestamp(unix as i64, 0);
    dt.format("%Y-%m-%d").to_string()
}

pub fn fmt_duration(dur: &Duration) -> String {
    const DAY: u64 = 86400;
    const HOUR: u64 = 3600;
    const MINUTE: u64 = 60;

    let secs = dur.as_secs();
    if secs > 90 * DAY {
        format!("{} months", secs / DAY / 30)
    } else if secs > 21 * DAY {
        format!("{} weeks", secs / DAY / 7)
    } else if secs > 3 * DAY {
        format!("{} days", secs / DAY)
    } else if secs > 3 * HOUR {
        format!("{} hours", secs / HOUR)
    } else if secs > 3 * MINUTE {
        return format!("{} mins", secs / MINUTE);
    } else {
        format!("{} secs", secs)
    }
}

// throttle a Sender to emit events at most once every `duration` seconds
pub fn throttle_sender(forward_tx: mpsc::Sender<()>, duration: u64) -> mpsc::Sender<()> {
    let duration = Duration::from_secs(duration);
    let (debounce_tx, debounce_rx) = mpsc::channel();

    thread::spawn(move || {
        loop {
            // wait to receive, then forward
            if debounce_rx.recv().is_err() {
                break;
            }
            debug!(target: "bwt::real-time", "triggering real-time index sync");
            if forward_tx.send(()).is_err() {
                break;
            }

            // drain
            while debounce_rx.try_recv().is_ok() {}

            // ignore events for `duration`
            thread::sleep(duration);
        }
        trace!(target: "bwt::real-time", "debounce sync thread shutting down");
    });

    debounce_tx
}

/// Wait for the future to resolve, blocking the current thread until it does
#[cfg(feature = "tokio")]
pub fn block_on_future<F: std::future::Future>(future: F) -> F::Output {
    let rt = tokio::runtime::Builder::new_current_thread()
        .enable_all()
        .build()
        .unwrap();
    rt.block_on(future)
}

/// Wait for the oneshot receiver `source_tx` to either receive or a message or to disconnect,
/// then trigger `f`. Returns a new receiver that mimics the source one.
pub fn on_oneshot_done<T: Send + 'static, F: Fn() + Send + 'static>(
    source_rx: mpsc::Receiver<T>,
    f: F,
) -> mpsc::Receiver<T> {
    let (tx, rx) = mpsc::sync_channel(1);
    thread::spawn(move || {
        if let Ok(v) = source_rx.recv() {
            tx.send(v).ok();
        }
        f();
    });
    rx
}

pub trait BoolThen {
    // Similar to https://doc.rust-lang.org/std/primitive.bool.html#method.then (nightly only)
    fn do_then<T>(self, f: impl FnOnce() -> T) -> Option<T>;

    // Alternative version where the closure returns an Option<T>
    fn and_then<T>(self, f: impl FnOnce() -> Option<T>) -> Option<T>;
}

impl BoolThen for bool {
    fn do_then<T>(self, f: impl FnOnce() -> T) -> Option<T> {
        if self {
            Some(f())
        } else {
            None
        }
    }
    fn and_then<T>(self, f: impl FnOnce() -> Option<T>) -> Option<T> {
        if self {
            f()
        } else {
            None
        }
    }
}
/// Daemon readiness notification via file descriptors
/// Primarly for use with s6, but can useful for any program that wants to be notified when the dameon is ready
/// See https://skarnet.org/software/s6/notifywhenup.html
#[cfg(all(unix, feature = "cli"))]
pub fn fd_readiness_notification() {
    use std::os::unix::io::FromRawFd;
    use std::{env, fs::File, io::Write};
    if let Ok(fd_number) = env::var("NOTIFY_FD") {
        if let Ok(fd_number) = fd_number.parse::<u32>() {
            let mut f = unsafe { File::from_raw_fd(fd_number as i32) };
            write!(f, "\n").ok();
        }
    }
}
#[cfg(not(all(unix, feature = "cli")))]
pub fn fd_readiness_notification() {}