use std::time::Duration;
use chrono::{DateTime, Utc};
use color_eyre::Report;
use zebra_chain::{
block::Height,
chain_tip::{ChainTip, NetworkChainTipHeightEstimator},
parameters::Network,
};
use crate::application::release_version;
pub const ESTIMATED_RELEASE_HEIGHT: u32 = 3_526_000;
pub const EOS_PANIC_AFTER: u32 = 56;
pub const EOS_WARN_AFTER: u32 = EOS_PANIC_AFTER - 14;
pub const EOS_PANIC_MESSAGE_HEADER: &str = "Zebra refuses to run";
pub const EOS_WARN_MESSAGE_HEADER: &str = "Your Zebra release is too old and it will stop running";
const CHECK_INTERVAL: Duration = Duration::from_secs(60 * 60);
const INITIAL_WAIT: Duration = Duration::from_secs(10);
pub async fn start(
network: Network,
latest_chain_tip: impl ChainTip + std::fmt::Debug,
) -> Result<(), Report> {
info!("Starting end of support task");
tokio::time::sleep(INITIAL_WAIT).await;
loop {
if network == Network::Mainnet {
if let Some(tip_height) = latest_chain_tip.best_tip_height() {
check(tip_height, &network);
}
} else {
info!("Release always valid in Testnet");
}
tokio::time::sleep(CHECK_INTERVAL).await;
}
}
pub fn end_of_support_height(network: &Network) -> Option<Height> {
(network == &Network::Mainnet).then(|| estimated_height_after_release(network, EOS_PANIC_AFTER))
}
pub fn estimated_height_after_release(network: &Network, days: u32) -> Height {
let release_time = DateTime::<Utc>::UNIX_EPOCH;
NetworkChainTipHeightEstimator::new(release_time, Height(ESTIMATED_RELEASE_HEIGHT), network)
.estimate_height_at(release_time + chrono::Duration::days(days.into()))
}
pub fn check(tip_height: Height, network: &Network) {
info!("Checking if Zebra release is inside support range ...");
let panic_height = estimated_height_after_release(network, EOS_PANIC_AFTER);
let warn_height = estimated_height_after_release(network, EOS_WARN_AFTER);
if tip_height > panic_height {
panic!(
"{EOS_PANIC_MESSAGE_HEADER} if the release date is older than {EOS_PANIC_AFTER} days. \
\nRelease name: {}, Estimated release height: {ESTIMATED_RELEASE_HEIGHT} \
\nHint: Download and install the latest Zebra release from: https://github.com/ZcashFoundation/zebra/releases/latest",
release_version()
);
} else if tip_height > warn_height {
warn!(
"{EOS_WARN_MESSAGE_HEADER} at block {}. \
\nRelease name: {}, Estimated release height: {ESTIMATED_RELEASE_HEIGHT} \
\nHint: Download and install the latest Zebra release from: https://github.com/ZcashFoundation/zebra/releases/latest", panic_height.0, release_version()
);
} else {
info!("Zebra release is supported until block {}, please report bugs at https://github.com/ZcashFoundation/zebra/issues", panic_height.0);
}
}