Files
grin-explorer/src/main.rs
2024-10-01 14:33:05 +03:00

787 lines
22 KiB
Rust

#[macro_use] extern crate rocket;
use chrono::Utc;
use either::Either;
use num_format::{Locale, ToFormattedString};
use rocket_dyn_templates::{Template, context};
use rocket::fs::FileServer;
use rocket::{State, tokio};
use rocket::response::Redirect;
use rocket::serde::json::json;
use std::sync::{Arc, Mutex};
use std::time::Duration;
use serde_json::Value;
use crate::data::{Block, Dashboard, Kernel, Output, Statistics, Transactions, OUTPUT_SIZE, KERNEL_SIZE};
use crate::requests::CONFIG;
mod data;
mod requests;
mod worker;
// Rendering main (Dashboard) page.
#[get("/")]
fn index(dashboard: &State<Arc<Mutex<Dashboard>>>) -> Template {
let data = dashboard.lock().unwrap();
Template::render("index", context! {
route: "index",
node_ver: &data.node_ver,
proto_ver: &data.proto_ver,
cg_api: CONFIG.coingecko_api.clone(),
})
}
// Rendering block list (Blocks) page.
#[get("/block_list")]
fn block_list() -> Template {
Template::render("block_list", context! {
route: "block_list",
})
}
// Rendering block list starting with a specified height.
// [<--] and [-->] buttons at the bottom of the block list (Blocks) page.
#[get("/block_list/<input_height>")]
async fn block_list_by_height(input_height: &str) -> Template {
let mut blocks = Vec::<Block>::new();
// Store current latest height
let mut height = 0;
let _ = requests::get_block_list_by_height(&input_height, &mut blocks, &mut height).await;
// Check if user's input doesn't overflow current height
if blocks.is_empty() == false && blocks[0].height.is_empty() == false {
let index = blocks[0].height.parse::<u64>().unwrap();
if index >= height {
Template::render("block_list", context! {
route: "block_list",
})
} else {
Template::render("block_list", context! {
route: "block_list_by_height",
index,
blocks,
height,
})
}
} else {
Template::render("block_list", context! {
route: "block_list",
})
}
}
// Rendering page for a specified block (by height).
#[get("/block/<height>")]
async fn block_details_by_height(height: &str) -> Template {
let mut block = Block::new();
if height.is_empty() == false && height.chars().all(char::is_numeric) == true {
let _ = requests::get_block_data(&height, &mut block).await;
if block.height.is_empty() == false {
return Template::render("block_details", context! {
route: "block_details",
block,
});
}
}
Template::render("error", context! {
route: "error",
})
}
// Rendering page for a specified block (by hash).
#[get("/hash/<hash>")]
async fn block_header_by_hash(hash: &str) -> Either<Template, Redirect> {
let mut height = String::new();
let _ = requests::get_block_header(&hash, &mut height).await;
if hash.is_empty() == false {
if height.is_empty() == false {
return Either::Right(Redirect::to(uri!(block_details_by_height(height.as_str()))));
}
}
return Either::Left(Template::render("error", context! {
route: "error",
}))
}
// Rendering page for a specified kernel.
#[get("/kernel/<excess>")]
async fn kernel(excess: &str) -> Template {
let mut kernel = Kernel::new();
let _ = requests::get_kernel(&excess, &mut kernel).await;
if kernel.excess.is_empty() == false {
return Template::render("kernel", context! {
route: "kernel",
kernel,
})
}
return Template::render("error", context! {
route: "error",
})
}
// Rendering page for a specified output.
#[get("/output/<commit>")]
async fn output(commit: &str) -> Template {
let mut output = Output::new();
let _ = requests::get_output(&commit, &mut output).await;
if output.commit.is_empty() == false {
return Template::render("output", context! {
route: "output",
output,
})
}
return Template::render("error", context! {
route: "error",
})
}
// Handling search request.
// Using Option<&str> to match '/search' query without input params.
// https://github.com/rwf2/Rocket/issues/608
#[get("/search?<input>")]
pub async fn search(input: Option<&str>) -> Either<Template, Redirect> {
// Unwrap Option and forward to Search page if no parameters
let input = match input {
Some(value) => value,
None => return Either::Left(Template::render("search", context! {
route: "search",
})),
};
// Check for valid chars
if input.chars().all(|x| (x >= 'a' && x <= 'f') || (x >= '0' && x <= '9')) == true {
// Block number
if input.chars().all(char::is_numeric) == true {
return Either::Right(Redirect::to(uri!(block_details_by_height(input))));
// Block hash
} else if input.len() == 64 {
return Either::Right(Redirect::to(uri!(block_header_by_hash(input))));
// Kernel or Unspent Output
} else if input.len() == 66 {
// First search for Kernel.
// If found, redirect to Kernel page, otherwise search for Unspent Output.
// As we can't distinguish between Kernel and Output, this will produce a redundant
// get_kernel call, but will allow for better UI (no need to ask user to input the type
// of the search request).
let mut kernel = Kernel::new();
let _ = requests::get_kernel(&input, &mut kernel).await;
if kernel.excess.is_empty() == false {
// Here we are redirecting to kernel page and call get_kernel again there.
// Kernel page is a separate route and we want it to be accessed directly and
// via search functionality.
return Either::Right(Redirect::to(uri!(kernel(input))));
} else {
// If Kernel not found, then search for Unspent Output
return Either::Right(Redirect::to(uri!(output(input))));
}
}
}
Either::Left(Template::render("error", context! {
route: "error",
}))
}
// Rendering Statistics page.
#[get("/stats")]
fn stats(statistics: &State<Arc<Mutex<Statistics>>>) -> Template {
let data = statistics.lock().unwrap();
Template::render("stats", context! {
route: "stats",
date: data.date.clone(),
user_agent: data.user_agent.clone(),
count: data.count.clone(),
total: data.total,
hashrate: data.hashrate.clone(),
txns: data.txns.clone(),
fees: data.fees.clone(),
utxo_count: data.utxo_count.clone(),
kernels: data.kernels.clone(),
output_size: OUTPUT_SIZE,
kernel_size: KERNEL_SIZE,
})
}
// Owner API.
// Whitelisted methods: get_connected_peers, get_peers, get_status.
#[post("/v2/owner", data="<data>")]
async fn api_owner(data: &str) -> Value {
if CONFIG.public_api == "enabled" {
let result = serde_json::from_str(data);
let v: Value = match result {
Ok(value) => value,
Err(_err) => return json!({"error":"bad syntax"}),
};
let method = match v["method"].as_str() {
Some(value) => value,
_ => return json!({"error":"bad syntax"}),
};
if method == "get_connected_peers" || method == "get_peers" || method == "get_status" {
let resp = requests::call(method, v["params"].to_string().as_str(), v["id"].to_string().as_str(), "owner").await;
let result = match resp {
Ok(value) => value,
Err(_err) => return json!({"error":"rpc call failed"}),
};
return json!(result);
}
json!({"error":"not allowed"})
} else {
json!({"error":"not allowed"})
}
}
// Foreign API.
// All methods are whitelisted.
#[post("/v2/foreign", data="<data>")]
async fn api_foreign(data: &str) -> Value {
if CONFIG.public_api == "enabled" {
let result = serde_json::from_str(data);
let v: Value = match result {
Ok(value) => value,
Err(_err) => return json!({"error":"bad syntax"}),
};
let method = match v["method"].as_str() {
Some(value) => value,
_ => return json!({"error":"bad syntax"}),
};
let resp = requests::call(method, v["params"].to_string().as_str(), v["id"].to_string().as_str(), "foreign").await;
let result = match resp {
Ok(value) => value,
Err(_err) => return json!({"error":"rpc call failed"}),
};
return json!(result);
} else {
json!({"error":"not allowed"})
}
}
// Start of HTMX routes.
#[get("/rpc/peers/inbound")]
fn peers_inbound(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
data.inbound.to_string()
}
#[get("/rpc/peers/outbound")]
fn peers_outbound(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
data.outbound.to_string()
}
#[get("/rpc/sync/status")]
fn sync_status(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
if data.sync == "no_sync" {
"Synced".to_string()
} else {
format!("Syncing
<div class='spinner-grow spinner-grow-sm' role='status'>
<span class='visually-hidden'>Syncing...</span></div>")
}
}
#[get("/rpc/market/supply")]
fn market_supply(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("{}", data.supply)
}
#[get("/rpc/market/soft_supply")]
fn soft_supply(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
if data.supply.is_empty() == false {
// 9 digits plus 2 commas, e.g. 168,038,400
if data.supply.len() == 11 {
return format!("{} % ({}M/3150M)", data.soft_supply, &data.supply[..3]);
// 10 digits plus 2 commas
} else if data.supply.len() == 12 {
return format!("{} % ({}M/3150M)", data.soft_supply, &data.supply[..4]);
}
}
"3150M".to_string()
}
#[get("/rpc/inflation/rate")]
fn inflation_rate(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("{} %", data.inflation)
}
#[get("/rpc/market/volume_usd")]
fn volume_usd(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("$ {}", data.volume_usd)
}
#[get("/rpc/market/volume_btc")]
fn volume_btc(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("{}", data.volume_btc)
}
#[get("/rpc/price/usd")]
fn price_usd(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("$ {}", data.price_usd)
}
#[get("/rpc/price/btc")]
fn price_btc(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
let trim: &[_] = &['0', '.'];
format!("{} sats", data.price_btc.trim_start_matches(trim))
}
#[get("/rpc/market/cap_usd")]
fn mcap_usd(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("$ {}", data.cap_usd)
}
#[get("/rpc/market/cap_btc")]
fn mcap_btc(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("{}", data.cap_btc)
}
#[get("/rpc/block/latest")]
fn latest_height(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
data.height.clone()
}
#[get("/rpc/block/time_since_last")]
fn last_block_age(blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false {
return data[0].time.clone();
}
"".to_string()
}
#[get("/rpc/disk/usage")]
fn disk_usage(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
if data.disk_usage.is_empty() == false {
return format!("{} GB", data.disk_usage);
} else {
return format!("<i class=\"bi bi-x-lg\"></i>");
}
}
#[get("/rpc/network/hashrate")]
fn network_hashrate(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
data.hashrate.clone()
}
#[get("/rpc/mining/production_cost")]
fn production_cost(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("$ {}", data.production_cost)
}
#[get("/rpc/mining/reward_ratio")]
fn reward_ratio(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
if data.reward_ratio.is_empty() == false {
let ratio = data.reward_ratio.parse::<f64>().unwrap();
if ratio <= 1.0 {
return format!("x{} <i class='bi bi-hand-thumbs-down'></i>", data.reward_ratio);
} else if ratio < 2.0 {
return format!("x{} <i class='bi bi-hand-thumbs-up'></i>", data.reward_ratio);
} else if ratio < 3.0 {
return format!("x{} <i class='bi bi-emoji-sunglasses'></i>", data.reward_ratio);
} else if ratio >= 3.0 {
return format!("x{} <i class='bi bi-rocket-takeoff'></i>", data.reward_ratio);
}
}
data.reward_ratio.clone()
}
#[get("/rpc/mining/breakeven_cost")]
fn breakeven_cost(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
format!("$ {} (kW/h)", data.breakeven_cost)
}
#[get("/rpc/network/difficulty")]
fn network_difficulty(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
data.difficulty.to_string()
}
#[get("/rpc/mempool/txns")]
fn mempool_txns(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
data.txns.to_string()
}
#[get("/rpc/mempool/stem")]
fn mempool_stem(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
data.stem.to_string()
}
#[get("/rpc/txns/count_1h")]
fn txns_count_1h(transactions: &State<Arc<Mutex<Transactions>>>) -> String {
let data = transactions.lock().unwrap();
format!("{}, ツ {}", data.period_1h, data.fees_1h)
}
#[get("/rpc/txns/count_24h")]
fn txns_count_24h(transactions: &State<Arc<Mutex<Transactions>>>) -> String {
let data = transactions.lock().unwrap();
format!("{}, ツ {}", data.period_24h, data.fees_24h)
}
#[get("/rpc/block/link?<count>")]
fn block_link(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return format!("<a href=/block/{} class='text-decoration-none'>{}</a>",
data[count].height, data[count].height);
}
"".to_string()
}
#[get("/rpc/block/link_color?<count>")]
fn block_link_color(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return format!("<a href=/block/{} class='text-decoration-none darkorange-text'>{}</a>",
data[count].height, data[count].height);
}
"".to_string()
}
#[get("/rpc/block/time?<count>")]
fn block_time(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return data[count].time.clone();
}
"".to_string()
}
#[get("/rpc/block/kernels?<count>")]
fn block_txns(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return data[count].ker_len.to_string();
}
"".to_string()
}
#[get("/rpc/block/inputs?<count>")]
fn block_inputs(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return data[count].in_len.to_string();
}
"".to_string()
}
#[get("/rpc/block/outputs?<count>")]
fn block_outputs(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return data[count].out_len.to_string();
}
"".to_string()
}
#[get("/rpc/block/fees?<count>")]
fn block_fees(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return format!("{}", data[count].fees / 1000000000.0);
}
"".to_string()
}
#[get("/rpc/block/size?<count>")]
fn block_size(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return data[count].size.clone();
}
"".to_string()
}
#[get("/rpc/block/weight?<count>")]
fn block_weight(count: usize, blocks: &State<Arc<Mutex<Vec<Block>>>>) -> String {
let data = blocks.lock().unwrap();
if data.is_empty() == false && count < 10 {
return format!("{} %", data[count].weight);
}
"".to_string()
}
#[get("/rpc/block_list/index")]
fn block_list_index(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
if data.height.is_empty() == false && data.height.parse::<u64>().unwrap() > 10 {
return format!("<a class='text-decoration-none' href='/block_list/{}'>
<div class='col-sm'><h2><i class='bi bi-arrow-right-square'></i></h2></div>
</a>", data.height.parse::<u64>().unwrap() - 10);
}
"".to_string()
}
#[get("/rpc/blockchain/unspent_outputs")]
fn unspent_outputs(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
if data.utxo_count.is_empty() == false {
let utxo_count = data.utxo_count.parse::<u64>().unwrap();
let mut utxo_size = utxo_count as f64 * OUTPUT_SIZE as f64 / 1000.0 / 1000.0;
let mut unit = "MB";
if utxo_size > 1000.0 {
unit = "GB";
utxo_size = utxo_size / 1000.0;
}
return format!("{} ({:.2} {})", utxo_count.to_formatted_string(&Locale::en), utxo_size, unit);
}
"".to_string()
}
#[get("/rpc/blockchain/kernels")]
fn kernels(dashboard: &State<Arc<Mutex<Dashboard>>>) -> String {
let data = dashboard.lock().unwrap();
if data.kernel_mmr_size.is_empty() == false {
let kernel_count = data.kernel_mmr_size.parse::<u64>().unwrap() / 2;
let mut kernel_size = kernel_count as f64 * KERNEL_SIZE as f64 / 1000.0 / 1000.0;
let mut unit = "MB";
if kernel_size > 1000.0 {
unit = "GB";
kernel_size = kernel_size / 1000.0;
}
return format!("{} ({:.2} {})", kernel_count.to_formatted_string(&Locale::en), kernel_size, unit);
}
"".to_string()
}
// End of HTMX routes.
// Main
#[rocket::main]
async fn main() {
env_logger::init();
info!("starting up.");
let dash = Arc::new(Mutex::new(Dashboard::new()));
let dash_clone = dash.clone();
let blocks = Arc::new(Mutex::new(Vec::<Block>::new()));
let blocks_clone = blocks.clone();
let txns = Arc::new(Mutex::new(Transactions::new()));
let txns_clone = txns.clone();
let stats = Arc::new(Mutex::new(Statistics::new()));
let stats_clone = stats.clone();
let mut ready_data = false;
let mut ready_stats = false;
let mut date = "".to_string();
// Collecting main data
tokio::spawn(async move {
loop {
let result = worker::data(dash_clone.clone(), blocks_clone.clone(),
txns_clone.clone(), stats_clone.clone()).await;
match result {
Ok(_v) => {
if ready_data == false {
ready_data = true;
info!("worker::data ready.");
}
},
Err(e) => {
ready_data = false;
error!("{}", e);
},
}
let date_now = format!("\"{}\"", Utc::now().format("%d-%m-%Y"));
if date.is_empty() || date != date_now {
date = date_now;
let result = worker::stats(dash_clone.clone(), txns_clone.clone(),
stats_clone.clone()).await;
match result {
Ok(_v) => {
if ready_stats == false {
ready_stats = true;
info!("worker::stats ready.");
}
},
Err(e) => {
ready_stats = false;
error!("{}", e);
},
}
}
tokio::time::sleep(Duration::from_secs(15)).await;
}
});
// Starting Rocket engine.
let _ = rocket::build()
.manage(dash)
.manage(blocks)
.manage(txns)
.manage(stats)
.mount("/", routes![index, peers_inbound, peers_outbound, sync_status, market_supply,
inflation_rate, volume_usd, volume_btc, price_usd, price_btc,
mcap_usd, mcap_btc,latest_height, disk_usage, network_hashrate,
network_difficulty, mempool_txns, mempool_stem, txns_count_1h,
txns_count_24h, block_list, block_link, block_link_color,
block_time, block_txns, block_inputs, block_outputs, block_fees,
block_size, block_weight, block_details_by_height, block_header_by_hash,
soft_supply, production_cost, reward_ratio, breakeven_cost,
last_block_age, block_list_by_height, block_list_index, search, kernel,
output, api_owner, api_foreign, stats, unspent_outputs, kernels])
.mount("/static", FileServer::from("static"))
.attach(Template::fairing())
.launch()
.await;
}