diff --git a/manager/src/main.rs b/manager/src/main.rs index 5404830..06486f0 100644 --- a/manager/src/main.rs +++ b/manager/src/main.rs @@ -33,8 +33,11 @@ pub struct ChainInfo { #[derive(Clone, Debug, serde::Deserialize)] pub struct NetworkInfo { + #[allow(dead_code)] connections: usize, + #[allow(dead_code)] connections_in: usize, + #[allow(dead_code)] connections_out: usize, } @@ -112,6 +115,50 @@ pub struct Stat { fn sidecar(config: &Mapping, addr: &str) -> Result<(), Box> { let mut stats = LinearMap::new(); + // Node uptime with years included + let uptime_info = std::process::Command::new("bitcoin-cli") + .arg("-conf=/root/.bitcoin/bitcoin.conf") + .arg("uptime") + .output()?; + + if uptime_info.status.success() { + let uptime_seconds = String::from_utf8_lossy(&uptime_info.stdout) + .trim() + .parse::() + .unwrap_or(0); + + let years = uptime_seconds / 31_536_000; + let days = (uptime_seconds % 31_536_000) / 86400; + let hours = (uptime_seconds % 86400) / 3600; + let minutes = (uptime_seconds % 3600) / 60; + + let uptime_formatted = if years > 0 { + format!("{} years, {} days, {} hours, {} minutes", years, days, hours, minutes) + } else if days > 0 { + format!("{} days, {} hours, {} minutes", days, hours, minutes) + } else if hours > 0 { + format!("{} hours, {} minutes", hours, minutes) + } else { + format!("{} minutes", minutes) + }; + + stats.insert( + Cow::from("Node Uptime"), + Stat { + value_type: "string", + value: uptime_formatted, + description: Some(Cow::from("Total time the Bitcoin node has been running")), + copyable: false, + qr: false, + masked: false, + }, + ); + } else { + eprintln!( + "Error retrieving uptime info: {}", + std::str::from_utf8(&uptime_info.stderr).unwrap_or("UNKNOWN ERROR") + ); + } if let (Some(user), Some(pass)) = ( config .get(&Value::String("rpc".to_owned())) @@ -222,65 +269,45 @@ fn sidecar(config: &Mapping, addr: &str) -> Result<(), Box> { ); } - // New section to fetch mempool statistics - let mempool_info = std::process::Command::new("bitcoin-cli") - .arg("-conf=/root/.bitcoin/bitcoin.conf") - .arg("getmempoolinfo") - .output()?; - - if mempool_info.status.success() { - let mempool_data: serde_json::Value = serde_json::from_slice(&mempool_info.stdout)?; - - let max_mempool = mempool_data["maxmempool"].as_u64().unwrap_or(0) as f64 / 1024_f64.powf(2.0); // Convert bytes to MB - let mempool_usage = mempool_data["usage"].as_u64().unwrap_or(0) as f64 / 1024_f64.powf(2.0); // Convert bytes to MB - let mempool_percent = if max_mempool > 0.0 { - (mempool_usage / max_mempool) * 100.0 - } else { - 0.0 - }; - let tx_count = mempool_data["size"].as_u64().unwrap_or(0); // Number of transactions - - stats.insert( - Cow::from("Max Mempool Size"), - Stat { - value_type: "string", - value: format!("{:.2} MB", max_mempool), - description: Some(Cow::from("Maximum memory pool size")), - copyable: false, - qr: false, - masked: false, - }, - ); +// New section to fetch and display a simplified mempool summary +let mempool_info = std::process::Command::new("bitcoin-cli") + .arg("-conf=/root/.bitcoin/bitcoin.conf") + .arg("getmempoolinfo") + .output()?; - stats.insert( - Cow::from("Current Mempool Usage"), - Stat { - value_type: "string", - value: format!("{:.2} MB ({:.2}%)", mempool_usage, mempool_percent), - description: Some(Cow::from("Current memory pool usage as a percentage of max size")), - copyable: false, - qr: false, - masked: false, - }, - ); +if mempool_info.status.success() { + let mempool_data: serde_json::Value = serde_json::from_slice(&mempool_info.stdout)?; - stats.insert( - Cow::from("Mempool Transaction Count"), - Stat { - value_type: "string", - value: format!("{}", tx_count), - description: Some(Cow::from("Current number of transactions in the mempool")), - copyable: false, - qr: false, - masked: false, - }, - ); + let max_mempool = mempool_data["maxmempool"].as_u64().unwrap_or(0) as f64 / 1024_f64.powf(2.0); // Convert to MB + let mempool_usage = mempool_data["usage"].as_u64().unwrap_or(0) as f64 / 1024_f64.powf(2.0); // Convert to MB + let mempool_percent = if max_mempool > 0.0 { + (mempool_usage / max_mempool) * 100.0 } else { - eprintln!( - "Error retrieving mempool info: {}", - std::str::from_utf8(&mempool_info.stderr).unwrap_or("UNKNOWN ERROR") - ); - } + 0.0 + }; + let tx_count = mempool_data["size"].as_u64().unwrap_or(0); // Number of transactions + + // Insert the simplified mempool summary + stats.insert( + Cow::from("Mempool Summary"), + Stat { + value_type: "string", + value: format!( + "{:.2}/{:.2} MB ({:.2}%), {} Transactions", + mempool_usage, max_mempool, mempool_percent, tx_count + ), + description: Some(Cow::from("Summary of current mempool usage and transaction count")), + copyable: false, + qr: false, + masked: false, + }, + ); +} else { + eprintln!( + "Error retrieving mempool info: {}", + std::str::from_utf8(&mempool_info.stderr).unwrap_or("UNKNOWN ERROR") + ); +} // Existing code for blockchain and network info retrieval continues here... let info_res = std::process::Command::new("bitcoin-cli") @@ -289,45 +316,31 @@ fn sidecar(config: &Mapping, addr: &str) -> Result<(), Box> { .output()?; if info_res.status.success() { let info: ChainInfo = serde_json::from_slice(&info_res.stdout)?; + // Consolidated block height and sync progress information + // Extract values for current synced blocks, total headers, and sync progress percentage + let current_blocks = info.blocks; + let total_headers = info.headers; + let sync_progress = if current_blocks < total_headers { + 100.0 * info.verificationprogress + } else { + 100.0 + }; + + // Insert the simplified blockchain sync summary stats.insert( - Cow::from("Block Height"), - Stat { - value_type: "string", - value: format!("{}", info.headers), - description: Some(Cow::from("The current block height for the network")), - copyable: false, - qr: false, - masked: false, - }, - ); - stats.insert( - Cow::from("Synced Block Height"), - Stat { - value_type: "string", - value: format!("{}", info.blocks), - description: Some(Cow::from("The number of blocks the node has verified")), - copyable: false, - qr: false, - masked: false, - }, - ); - stats.insert( - Cow::from("Sync Progress"), + Cow::from("Blockchain Sync Summary"), Stat { value_type: "string", - value: if info.blocks < info.headers { - format!("{:.2}%", 100.0 * info.verificationprogress) - } else { - "100%".to_owned() - }, - description: Some(Cow::from( - "The percentage of the blockchain that has been verified", - )), + value: format!( + "{}/{} ({:.2}%)", + current_blocks, total_headers, sync_progress + ), + description: Some(Cow::from("Current synced block height out of total headers and sync progress")), copyable: false, qr: false, masked: false, }, - ); + ); for (sf_name, sf_data) in info.softforks { let sf_name_pretty = sf_name.to_title_case(); let status_desc = Some(Cow::from(format!( @@ -494,31 +507,76 @@ fn sidecar(config: &Mapping, addr: &str) -> Result<(), Box> { std::str::from_utf8(&info_res.stderr).unwrap_or("UNKNOWN ERROR") ); } - let info_res = std::process::Command::new("bitcoin-cli") + // Detailed peer info by network type + let peer_info = std::process::Command::new("bitcoin-cli") .arg("-conf=/root/.bitcoin/bitcoin.conf") - .arg("getnetworkinfo") + .arg("getpeerinfo") .output()?; - if info_res.status.success() { - let info: NetworkInfo = serde_json::from_slice(&info_res.stdout)?; + + if peer_info.status.success() { + let peers: Vec = serde_json::from_slice(&peer_info.stdout)?; + let mut clearnet_in = 0; + let mut clearnet_out = 0; + let mut tor_in = 0; + let mut tor_out = 0; + let mut i2p_in = 0; + let mut i2p_out = 0; + + for peer in peers { + let network = peer["network"].as_str().unwrap_or(""); + let inbound = peer["inbound"].as_bool().unwrap_or(false); + + match network { + "ipv4" | "ipv6" => { + if inbound { + clearnet_in += 1; + } else { + clearnet_out += 1; + } + } + "onion" => { + if inbound { + tor_in += 1; + } else { + tor_out += 1; + } + } + "i2p" => { + if inbound { + i2p_in += 1; + } else { + i2p_out += 1; + } + } + _ => {} + } + } + + let total_in = clearnet_in + tor_in + i2p_in; + let total_out = clearnet_out + tor_out + i2p_out; + let total_connections = total_in + total_out; + stats.insert( - Cow::from("Connections"), + Cow::from("Peer Connection Summary"), Stat { value_type: "string", - value: format!("{} ({} in / {} out)", info.connections, info.connections_in, info.connections_out), - description: Some(Cow::from("The number of peers connected (inbound and outbound)")), + value: format!( + "Total: {} | Clearnet: In-{}, Out-{} | Tor: In-{}, Out-{} | I2P: In-{}, Out-{}", + total_connections, clearnet_in, clearnet_out, tor_in, tor_out, i2p_in, i2p_out + ), + description: Some(Cow::from("Summary of peer connections by type: Clearnet, Tor, I2P")), copyable: false, qr: false, masked: false, }, ); - } else if info_res.status.code() == Some(28) { - return Ok(()); } else { eprintln!( - "Error updating network info: {}", - std::str::from_utf8(&info_res.stderr).unwrap_or("UNKNOWN ERROR") + "Error retrieving peer info: {}", + std::str::from_utf8(&peer_info.stderr).unwrap_or("UNKNOWN ERROR") ); } + serde_yaml::to_writer( std::fs::File::create("/root/.bitcoin/start9/.stats.yaml.tmp")?, &Stats { @@ -532,6 +590,7 @@ fn sidecar(config: &Mapping, addr: &str) -> Result<(), Box> { )?; Ok(()) } + fn inner_main(reindex: bool) -> Result<(), Box> { while !Path::new("/root/.bitcoin/start9/config.yaml").exists() {