vdash 0.20.0

Autonomi node Dashboard
use std::collections::HashMap;

use chrono::{DateTime, Duration, Utc};
use ratatui::style::Color;
use serde::{Deserialize, Serialize};

use super::app::OPT;
use super::timelines::{Buckets, Timeline};
use std::sync::LazyLock;

pub static TIMESCALES: LazyLock<std::vec::Vec<(&'static str, Duration)>> = LazyLock::new(|| {
	vec![
		("1 second columns", Duration::seconds(1)),
		("1 minute columns", Duration::minutes(1)),
		("1 hour columns", Duration::hours(1)),
		("1 day columns", Duration::days(1)),
		("1 week columns", Duration::days(7)),
		("1 year columns", Duration::days(365)),
	]
});

pub const EARNINGS_UNITS_TEXT: &str = "attos";
pub const STORAGE_COST_UNITS_TEXT: &str = "attos/MB";

/// keys (used to access timelines)
pub const EARNINGS_TIMELINE_KEY: &str = "earnings";
pub const STORAGE_COST_TIMELINE_KEY: &str = "storage";
pub const PUTS_TIMELINE_KEY: &str = "puts";
pub const GETS_TIMELINE_KEY: &str = "gets";
pub const CONNECTIONS_TIMELINE_KEY: &str = "connections";
pub const RAM_TIMELINE_KEY: &str = "ram";
pub const ERRORS_TIMELINE_KEY: &str = "errors";

/// Defines the Timelines available for display
pub const APP_TIMELINES: [(&str, &str, &str, bool, bool, Color); 7] = [
	//  (key, UI name, units_text, is_mmm, is_cumulative, colour)
	(
		EARNINGS_TIMELINE_KEY,
		"Earnings",
		EARNINGS_UNITS_TEXT,
		false,
		true,
		Color::LightCyan,
	),
	(
		STORAGE_COST_TIMELINE_KEY,
		"Storage Cost",
		STORAGE_COST_UNITS_TEXT,
		true,
		false,
		Color::LightBlue,
	),
	(PUTS_TIMELINE_KEY, "PUTS", "", false, true, Color::Yellow),
	(GETS_TIMELINE_KEY, "GETS", "", false, true, Color::Green),
	(
		CONNECTIONS_TIMELINE_KEY,
		"Connections",
		"",
		true,
		false,
		Color::Blue,
	),
	(RAM_TIMELINE_KEY, "RAM", "MB", true, false, Color::Magenta),
	(ERRORS_TIMELINE_KEY, "ERRORS", "", false, true, Color::Red),
];

/// Holds the Timeline structs for a node, as used by this app
#[derive(Clone, Default, Debug, Serialize, Deserialize)]
pub struct AppTimelines {
	timelines: HashMap<String, Timeline>,
}

impl AppTimelines {
	pub fn new() -> AppTimelines {
		let opt_timeline_steps = {
			let opt = OPT.lock().unwrap();
			opt.timeline_steps
		};

		let mut app_timelines = AppTimelines {
			timelines: HashMap::<String, Timeline>::new(),
		};

		for (key, name, units_text, is_mmm, is_cumulative, colour) in APP_TIMELINES {
			app_timelines.timelines.insert(
				key.to_string(),
				Timeline::new(
					name.to_string(),
					units_text.to_string(),
					is_mmm,
					is_cumulative,
					colour,
				),
			);
		}

		for (_, timeline) in app_timelines.timelines.iter_mut() {
			for i in 0..TIMESCALES.len() {
				if let Some(spec) = TIMESCALES.get(i) {
					timeline.add_bucket_set(spec.0, spec.1, opt_timeline_steps);
				}
			}
		}

		return app_timelines;
	}

	pub fn update_timelines(&mut self, now: &DateTime<Utc>) {
		for (_, timeline) in self.timelines.iter_mut() {
			timeline.update_current_time(&now);
		}
	}

	pub fn get_timeline_by_key(&mut self, key: &str) -> Option<&mut Timeline> {
		return self.timelines.get_mut(key);
	}

	pub fn get_timeline_by_index(&self, index: usize) -> Option<&Timeline> {
		let (key, _, _, _, _, _) = APP_TIMELINES[index];
		return self.timelines.get(key);
	}

	// Gets the set of buckets for the index'th Timeline, selecting with Min, Mean, Max if appropriate
	pub fn get_timeline_buckets(&mut self, index: usize, timescale_name: &str) -> Option<&Buckets> {
		let (key, _, _, _, _, _) = APP_TIMELINES[index];
		if let Some(timeline) = self.timelines.get_mut(key) {
			return timeline.get_bucket_set(timescale_name);
		}
		return None;
	}

	pub fn get_num_timelines(self: &AppTimelines) -> usize {
		return APP_TIMELINES.len();
	}
}