bottom/src/canvas.rs

use crate::{
	app::{self, data_harvester::processes::ProcessHarvest, WidgetPosition},
	constants::*,
	data_conversion::{ConvertedCpuData, ConvertedProcessData},
	utils::error,
};
use std::cmp::max;
use std::collections::HashMap;
use tui::{
	backend,
	layout::{Alignment, Constraint, Direction, Layout, Rect},
	style::{Color, Style},
	terminal::Frame,
	widgets::{Axis, Block, Borders, Chart, Dataset, Marker, Paragraph, Row, Table, Text, Widget},
	Terminal,
};

mod canvas_colours;
use canvas_colours::*;

mod drawing_utils;
use drawing_utils::*;

// Headers
const CPU_LEGEND_HEADER: [&str; 2] = ["CPU", "Use%"];
const CPU_SELECT_LEGEND_HEADER: [&str; 2] = ["CPU", "Show (Space)"];
const DISK_HEADERS: [&str; 7] = ["Disk", "Mount", "Used", "Free", "Total", "R/s", "W/s"];
const TEMP_HEADERS: [&str; 2] = ["Sensor", "Temp"];
const MEM_HEADERS: [&str; 3] = ["Mem", "Usage", "Use%"];
const NETWORK_HEADERS: [&str; 4] = ["RX", "TX", "Total RX", "Total TX"];
const FORCE_MIN_THRESHOLD: usize = 5;

lazy_static! {
	static ref DEFAULT_TEXT_STYLE: Style = Style::default().fg(Color::Gray);
	static ref DEFAULT_HEADER_STYLE: Style = Style::default().fg(Color::LightBlue);
	static ref DISK_HEADERS_LENS: Vec<usize> = DISK_HEADERS
		.iter()
		.map(|entry| max(FORCE_MIN_THRESHOLD, entry.len()))
		.collect::<Vec<_>>();
	static ref CPU_LEGEND_HEADER_LENS: Vec<usize> = CPU_LEGEND_HEADER
		.iter()
		.map(|entry| max(FORCE_MIN_THRESHOLD, entry.len()))
		.collect::<Vec<_>>();
	static ref CPU_SELECT_LEGEND_HEADER_LENS: Vec<usize> = CPU_SELECT_LEGEND_HEADER
		.iter()
		.map(|entry| max(FORCE_MIN_THRESHOLD, entry.len()))
		.collect::<Vec<_>>();
	static ref TEMP_HEADERS_LENS: Vec<usize> = TEMP_HEADERS
		.iter()
		.map(|entry| max(FORCE_MIN_THRESHOLD, entry.len()))
		.collect::<Vec<_>>();
	static ref MEM_HEADERS_LENS: Vec<usize> = MEM_HEADERS
		.iter()
		.map(|entry| max(FORCE_MIN_THRESHOLD, entry.len()))
		.collect::<Vec<_>>();
	static ref NETWORK_HEADERS_LENS: Vec<usize> = NETWORK_HEADERS
		.iter()
		.map(|entry| max(FORCE_MIN_THRESHOLD, entry.len()))
		.collect::<Vec<_>>();
}

#[derive(Default)]
pub struct DisplayableData {
	pub rx_display: String,
	pub tx_display: String,
	pub total_rx_display: String,
	pub total_tx_display: String,
	pub network_data_rx: Vec<(f64, f64)>,
	pub network_data_tx: Vec<(f64, f64)>,
	pub disk_data: Vec<Vec<String>>,
	pub temp_sensor_data: Vec<Vec<String>>,
	pub process_data: HashMap<u32, ProcessHarvest>, // Not the final value
	pub grouped_process_data: Vec<ConvertedProcessData>, // Not the final value
	pub finalized_process_data: Vec<ConvertedProcessData>, // What's actually displayed
	pub mem_label: String,
	pub swap_label: String,
	pub mem_data: Vec<(f64, f64)>,
	pub swap_data: Vec<(f64, f64)>,
	pub cpu_data: Vec<ConvertedCpuData>,
}

#[allow(dead_code)]
#[derive(Default)]
/// Handles the canvas' state.  TODO: [OPT] implement this.
pub struct Painter {
	height: f64,
	width: f64,
	vertical_dialog_chunk: Vec<Rect>,
	middle_dialog_chunk: Vec<Rect>,
	vertical_chunks: Vec<Rect>,
	middle_chunks: Vec<Rect>,
	middle_divided_chunk_2: Vec<Rect>,
	bottom_chunks: Vec<Rect>,
	cpu_chunk: Vec<Rect>,
	network_chunk: Vec<Rect>,
	pub colours: CanvasColours,
	pub styled_general_help_text: Vec<Text<'static>>,
	pub styled_process_help_text: Vec<Text<'static>>,
	pub styled_search_help_text: Vec<Text<'static>>,
}

impl Painter {
	/// Must be run once before drawing, but after setting colours.
	/// This is to set some remaining styles and text.
	/// This bypasses some logic checks (size > 2, for example) but this
	/// assumes that you, the programmer, are sane and do not do stupid things.
	/// RIGHT?
	pub fn initialize(&mut self) {
		self.styled_general_help_text.push(Text::Styled(
			GENERAL_HELP_TEXT[0].into(),
			self.colours.table_header_style,
		));
		self.styled_general_help_text.extend(
			GENERAL_HELP_TEXT[1..]
				.iter()
				.map(|&text| Text::Styled(text.into(), self.colours.text_style))
				.collect::<Vec<_>>(),
		);

		self.styled_process_help_text.push(Text::Styled(
			PROCESS_HELP_TEXT[0].into(),
			self.colours.table_header_style,
		));
		self.styled_process_help_text.extend(
			PROCESS_HELP_TEXT[1..]
				.iter()
				.map(|&text| Text::Styled(text.into(), self.colours.text_style))
				.collect::<Vec<_>>(),
		);

		self.styled_search_help_text.push(Text::Styled(
			SEARCH_HELP_TEXT[0].into(),
			self.colours.table_header_style,
		));
		self.styled_search_help_text.extend(
			SEARCH_HELP_TEXT[1..]
				.iter()
				.map(|&text| Text::Styled(text.into(), self.colours.text_style))
				.collect::<Vec<_>>(),
		);
	}

	// TODO: [REFACTOR] We should clean this up tbh
	#[allow(clippy::cognitive_complexity)]
	pub fn draw_data<B: backend::Backend>(
		&mut self, terminal: &mut Terminal<B>, app_state: &mut app::App,
	) -> error::Result<()> {
		terminal.autoresize()?;
		terminal.draw(|mut f| {
			if app_state.help_dialog_state.is_showing_help {
				// Only for the help
				let vertical_dialog_chunk = Layout::default()
					.direction(Direction::Vertical)
					.margin(1)
					.constraints(
						[
							Constraint::Percentage(20),
							Constraint::Percentage(60),
							Constraint::Percentage(20),
						]
						.as_ref(),
					)
					.split(f.size());

				let middle_dialog_chunk = Layout::default()
					.direction(Direction::Horizontal)
					.margin(0)
					.constraints(
						[
							Constraint::Percentage(20),
							Constraint::Percentage(60),
							Constraint::Percentage(20),
						]
						.as_ref(),
					)
					.split(vertical_dialog_chunk[1]);

				const HELP_BASE: &str =
					" Help ── 1: General ─── 2: Processes ─── 3: Search ─── Esc to close ";
				let repeat_num = max(
					0,
					middle_dialog_chunk[1].width as i32 - HELP_BASE.chars().count() as i32 - 2,
				);
				let help_title = format!(
					" Help ─{}─ 1: General ─── 2: Processes ─── 3: Search ─── Esc to close ",
					"─".repeat(repeat_num as usize)
				);

				Paragraph::new(
					match app_state.help_dialog_state.current_category {
						app::AppHelpCategory::General => &self.styled_general_help_text,
						app::AppHelpCategory::Process => &self.styled_process_help_text,
						app::AppHelpCategory::Search => &self.styled_search_help_text,
					}
					.iter(),
				)
				.block(
					Block::default()
						.title(&help_title)
						.title_style(self.colours.border_style)
						.style(self.colours.border_style)
						.borders(Borders::ALL)
						.border_style(self.colours.border_style),
				)
				.style(self.colours.text_style)
				.alignment(Alignment::Left)
				.wrap(true)
				.render(&mut f, middle_dialog_chunk[1]);
			} else if app_state.delete_dialog_state.is_showing_dd {
				let vertical_dialog_chunk = Layout::default()
					.direction(Direction::Vertical)
					.margin(1)
					.constraints(
						[
							Constraint::Percentage(35),
							Constraint::Percentage(30),
							Constraint::Percentage(35),
						]
						.as_ref(),
					)
					.split(f.size());

				let middle_dialog_chunk = Layout::default()
					.direction(Direction::Horizontal)
					.margin(0)
					.constraints(
						[
							Constraint::Percentage(25),
							Constraint::Percentage(50),
							Constraint::Percentage(25),
						]
						.as_ref(),
					)
					.split(vertical_dialog_chunk[1]);

				if let Some(dd_err) = &app_state.dd_err {
					let dd_text = [Text::raw(format!(
						"\nFailure to properly kill the process - {}",
						dd_err
					))];

					const ERROR_BASE: &str = " Error ── Esc to close ";
					let repeat_num = max(
						0,
						middle_dialog_chunk[1].width as i32 - ERROR_BASE.chars().count() as i32 - 2,
					);
					let error_title =
						format!(" Error ─{}─ Esc to close ", "─".repeat(repeat_num as usize));

					Paragraph::new(dd_text.iter())
						.block(
							Block::default()
								.title(&error_title)
								.title_style(self.colours.border_style)
								.style(self.colours.border_style)
								.borders(Borders::ALL)
								.border_style(self.colours.border_style),
						)
						.style(self.colours.text_style)
						.alignment(Alignment::Center)
						.wrap(true)
						.render(&mut f, middle_dialog_chunk[1]);
				} else if let Some(to_kill_processes) = app_state.get_to_delete_processes() {
					if let Some(first_pid) = to_kill_processes.1.first() {
						let dd_text = vec![
							if app_state.is_grouped() {
								if to_kill_processes.1.len() != 1 {
									Text::raw(format!(
										"\nAre you sure you want to kill {} processes with the name {}?",
										to_kill_processes.1.len(), to_kill_processes.0
									))
								} else {
									Text::raw(format!(
										"\nAre you sure you want to kill {} process with the name {}?",
										to_kill_processes.1.len(), to_kill_processes.0
									))
								}
							} else {
								Text::raw(format!(
									"\nAre you sure you want to kill process {} with PID {}?",
									to_kill_processes.0, first_pid
								))
							},
							Text::raw("\nNote that if bottom is frozen, it must be unfrozen for changes to be shown.\n\n\n"),
							if app_state.delete_dialog_state.is_on_yes {
								Text::styled("Yes", self.colours.currently_selected_text_style)
							} else {
								Text::raw("Yes")
							},
							Text::raw("                 "),
							if app_state.delete_dialog_state.is_on_yes {
								Text::raw("No")
							} else {
								Text::styled("No", self.colours.currently_selected_text_style)
							},
						];

						const DD_BASE: &str = " Confirm Kill Process ── Esc to close ";
						let repeat_num = max(
							0,
							middle_dialog_chunk[1].width as i32
								- DD_BASE.chars().count() as i32 - 2,
						);
						let dd_title = format!(
							" Confirm Kill Process ─{}─ Esc to close ",
							"─".repeat(repeat_num as usize)
						);

						Paragraph::new(dd_text.iter())
							.block(
								Block::default()
									.title(&dd_title)
									.title_style(self.colours.border_style)
									.style(self.colours.border_style)
									.borders(Borders::ALL)
									.border_style(self.colours.border_style),
							)
							.style(self.colours.text_style)
							.alignment(Alignment::Center)
							.wrap(true)
							.render(&mut f, middle_dialog_chunk[1]);
					} else {
						// This is a bit nasty, but it works well... I guess.
						app_state.delete_dialog_state.is_showing_dd = false;
					}
				} else {
					// This is a bit nasty, but it works well... I guess.
					app_state.delete_dialog_state.is_showing_dd = false;
				}
			} else if app_state.is_expanded {
				// TODO: [REF] we should combine this with normal drawing tbh

				let rect = Layout::default()
					.margin(1)
					.constraints([Constraint::Percentage(100)].as_ref())
					.split(f.size());
				match &app_state.current_widget_selected {
					WidgetPosition::Cpu => {
						let cpu_chunk = Layout::default()
							.direction(Direction::Horizontal)
							.margin(0)
							.constraints(
								if app_state.app_config_fields.left_legend {
									[Constraint::Percentage(15), Constraint::Percentage(85)]
								} else {
									[Constraint::Percentage(85), Constraint::Percentage(15)]
								}
								.as_ref(),
							)
							.split(rect[0]);

						let legend_index = if app_state.app_config_fields.left_legend {
							0
						} else {
							1
						};
						let graph_index = if app_state.app_config_fields.left_legend {
							1
						} else {
							0
						};

						self.draw_cpu_graph(&mut f, &app_state, cpu_chunk[graph_index]);
						self.draw_cpu_legend(&mut f, app_state, cpu_chunk[legend_index]);
					}
					WidgetPosition::Mem => {
						self.draw_memory_graph(&mut f, &app_state, rect[0]);
					}
					WidgetPosition::Disk => {
						self.draw_disk_table(&mut f, app_state, rect[0]);
					}
					WidgetPosition::Temp => {
						self.draw_temp_table(&mut f, app_state, rect[0]);
					}
					WidgetPosition::Network => {
						self.draw_network_graph(&mut f, &app_state, rect[0]);
					}
					WidgetPosition::Process | WidgetPosition::ProcessSearch => {
						if app_state.is_searching() {
							let processes_chunk = Layout::default()
								.direction(Direction::Vertical)
								.margin(0)
								.constraints(
									[Constraint::Percentage(85), Constraint::Percentage(15)]
										.as_ref(),
								)
								.split(rect[0]);

							self.draw_processes_table(&mut f, app_state, processes_chunk[0]);
							self.draw_search_field(&mut f, app_state, processes_chunk[1]);
						} else {
							self.draw_processes_table(&mut f, app_state, rect[0]);
						}
					}
				}
			} else {
				// TODO: [TUI] Change this back to a more even 33/33/34 when TUI releases
				let vertical_chunks = Layout::default()
					.direction(Direction::Vertical)
					.margin(1)
					.constraints(
						[
							Constraint::Percentage(30),
							Constraint::Percentage(37),
							Constraint::Percentage(33),
						]
						.as_ref(),
					)
					.split(f.size());

				let middle_chunks = Layout::default()
					.direction(Direction::Horizontal)
					.margin(0)
					.constraints([Constraint::Percentage(60), Constraint::Percentage(40)].as_ref())
					.split(vertical_chunks[1]);

				let middle_divided_chunk_2 = Layout::default()
					.direction(Direction::Vertical)
					.margin(0)
					.constraints([Constraint::Percentage(50), Constraint::Percentage(50)].as_ref())
					.split(middle_chunks[1]);

				let bottom_chunks = Layout::default()
					.direction(Direction::Horizontal)
					.margin(0)
					.constraints([Constraint::Percentage(50), Constraint::Percentage(50)].as_ref())
					.split(vertical_chunks[2]);

				// Component specific chunks
				let cpu_chunk = Layout::default()
					.direction(Direction::Horizontal)
					.margin(0)
					.constraints(
						if app_state.app_config_fields.left_legend {
							[Constraint::Percentage(15), Constraint::Percentage(85)]
						} else {
							[Constraint::Percentage(85), Constraint::Percentage(15)]
						}
						.as_ref(),
					)
					.split(vertical_chunks[0]);

				let network_chunk = Layout::default()
					.direction(Direction::Vertical)
					.margin(0)
					.constraints(
						if (bottom_chunks[0].height as f64 * 0.25) as u16 >= 4 {
							[Constraint::Percentage(75), Constraint::Percentage(25)]
						} else {
							let required = if bottom_chunks[0].height < 10 {
								bottom_chunks[0].height / 2
							} else {
								5
							};
							let remaining = bottom_chunks[0].height - required;
							[Constraint::Length(remaining), Constraint::Length(required)]
						}
						.as_ref(),
					)
					.split(bottom_chunks[0]);

				// Default chunk index based on left or right legend setting
				let legend_index = if app_state.app_config_fields.left_legend {
					0
				} else {
					1
				};
				let graph_index = if app_state.app_config_fields.left_legend {
					1
				} else {
					0
				};

				// Set up blocks and their components
				// CPU graph + legend
				self.draw_cpu_graph(&mut f, &app_state, cpu_chunk[graph_index]);
				self.draw_cpu_legend(&mut f, app_state, cpu_chunk[legend_index]);

				//Memory usage graph
				self.draw_memory_graph(&mut f, &app_state, middle_chunks[0]);

				// Network graph
				self.draw_network_graph(&mut f, &app_state, network_chunk[0]);
				self.draw_network_labels(&mut f, app_state, network_chunk[1]);

				// Temperature table
				self.draw_temp_table(&mut f, app_state, middle_divided_chunk_2[0]);

				// Disk usage table
				self.draw_disk_table(&mut f, app_state, middle_divided_chunk_2[1]);

				// Processes table
				if app_state.is_searching() {
					let processes_chunk = Layout::default()
						.direction(Direction::Vertical)
						.margin(0)
						.constraints(
							if (bottom_chunks[1].height as f64 * 0.25) as u16 >= 4 {
								[Constraint::Percentage(75), Constraint::Percentage(25)]
							} else {
								let required = if bottom_chunks[1].height < 10 {
									bottom_chunks[1].height / 2
								} else {
									5
								};
								let remaining = bottom_chunks[1].height - required;
								[Constraint::Length(remaining), Constraint::Length(required)]
							}
							.as_ref(),
						)
						.split(bottom_chunks[1]);

					self.draw_processes_table(&mut f, app_state, processes_chunk[0]);
					self.draw_search_field(&mut f, app_state, processes_chunk[1]);
				} else {
					self.draw_processes_table(&mut f, app_state, bottom_chunks[1]);
				}
			}
		})?;

		Ok(())
	}

	fn draw_cpu_graph<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &app::App, draw_loc: Rect,
	) {
		let cpu_data: &[ConvertedCpuData] = &app_state.canvas_data.cpu_data;

		// CPU usage graph
		let x_axis: Axis<String> = Axis::default().bounds([0.0, TIME_STARTS_FROM as f64]);
		let y_axis = Axis::default()
			.style(self.colours.graph_style)
			.labels_style(self.colours.graph_style)
			.bounds([-0.5, 100.5])
			.labels(&["0%", "100%"]);

		let dataset_vector: Vec<Dataset> = cpu_data
			.iter()
			.enumerate()
			.rev()
			.filter_map(|(itx, cpu)| {
				if app_state.cpu_state.core_show_vec[itx] {
					Some(
						Dataset::default()
							.marker(if app_state.app_config_fields.use_dot {
								Marker::Dot
							} else {
								Marker::Braille
							})
							.style(
								self.colours.cpu_colour_styles
									[itx % self.colours.cpu_colour_styles.len()],
							)
							.data(&cpu.cpu_data[..]),
					)
				} else {
					None
				}
			})
			.collect();

		let title = if app_state.is_expanded && !app_state.cpu_state.is_showing_tray {
			const TITLE_BASE: &str = " CPU ── Esc to go back ";
			let repeat_num = max(
				0,
				draw_loc.width as i32 - TITLE_BASE.chars().count() as i32 - 2,
			);
			let result_title =
				format!(" CPU ─{}─ Esc to go back ", "─".repeat(repeat_num as usize));

			result_title
		} else {
			" CPU ".to_string()
		};

		Chart::default()
			.block(
				Block::default()
					.title(&title)
					.title_style(if app_state.is_expanded {
						self.colours.highlighted_border_style
					} else {
						self.colours.widget_title_style
					})
					.borders(Borders::ALL)
					.border_style(match app_state.current_widget_selected {
						app::WidgetPosition::Cpu => self.colours.highlighted_border_style,
						_ => self.colours.border_style,
					}),
			)
			.x_axis(x_axis)
			.y_axis(y_axis)
			.datasets(&dataset_vector)
			.render(f, draw_loc);
	}

	fn draw_cpu_legend<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect,
	) {
		let cpu_data: &[ConvertedCpuData] = &(app_state.canvas_data.cpu_data);

		let num_rows = max(0, i64::from(draw_loc.height) - 5) as u64;
		let start_position = get_start_position(
			num_rows,
			&(app_state.app_scroll_positions.scroll_direction),
			&mut app_state
				.app_scroll_positions
				.cpu_scroll_state
				.previous_scroll_position,
			app_state
				.app_scroll_positions
				.cpu_scroll_state
				.current_scroll_position,
		);

		let sliced_cpu_data = &cpu_data[start_position as usize..];
		let mut stringified_cpu_data: Vec<Vec<String>> = Vec::new();

		for (itx, cpu) in sliced_cpu_data.iter().enumerate() {
			if app_state.cpu_state.is_showing_tray {
				stringified_cpu_data.push(vec![
					cpu.cpu_name.clone(),
					if app_state.cpu_state.core_show_vec[itx + start_position as usize] {
						"[*]".to_string()
					} else {
						"[ ]".to_string()
					},
				]);
			} else if let Some(cpu_data) = cpu.cpu_data.last() {
				if app_state.app_config_fields.show_disabled_data
					|| app_state.cpu_state.core_show_vec[itx]
				{
					stringified_cpu_data.push(vec![
						cpu.cpu_name.clone(),
						format!("{:.0}%", cpu_data.1.round()),
					]);
				}
			}
		}

		let cpu_rows = stringified_cpu_data
			.iter()
			.enumerate()
			.map(|(itx, cpu_string_row)| {
				Row::StyledData(
					cpu_string_row.iter(),
					match app_state.current_widget_selected {
						app::WidgetPosition::Cpu => {
							if itx as u64
								== app_state
									.app_scroll_positions
									.cpu_scroll_state
									.current_scroll_position - start_position
							{
								self.colours.currently_selected_text_style
							} else {
								self.colours.cpu_colour_styles[itx
									+ start_position as usize
										% self.colours.cpu_colour_styles.len()]
							}
						}
						_ => {
							self.colours.cpu_colour_styles[itx
								+ start_position as usize % self.colours.cpu_colour_styles.len()]
						}
					},
				)
			});

		// Calculate widths
		let width = f64::from(draw_loc.width);
		let width_ratios = vec![0.5, 0.5];

		let variable_intrinsic_results = get_variable_intrinsic_widths(
			width as u16,
			&width_ratios,
			if app_state.cpu_state.is_showing_tray {
				&CPU_SELECT_LEGEND_HEADER_LENS
			} else {
				&CPU_LEGEND_HEADER_LENS
			},
		);
		let intrinsic_widths = &(variable_intrinsic_results.0)[0..variable_intrinsic_results.1];

		let title = if app_state.cpu_state.is_showing_tray {
			const TITLE_BASE: &str = " Esc to close ";
			let repeat_num = max(
				0,
				draw_loc.width as i32 - TITLE_BASE.chars().count() as i32 - 2,
			);
			let result_title = format!("{} Esc to close ", "─".repeat(repeat_num as usize));

			result_title
		} else {
			"".to_string()
		};

		// Draw
		Table::new(
			if app_state.cpu_state.is_showing_tray {
				CPU_SELECT_LEGEND_HEADER
			} else {
				CPU_LEGEND_HEADER
			}
			.iter(),
			cpu_rows,
		)
		.block(
			Block::default()
				.title(&title)
				.title_style(if app_state.is_expanded {
					self.colours.highlighted_border_style
				} else {
					match app_state.current_widget_selected {
						app::WidgetPosition::Cpu => self.colours.highlighted_border_style,
						_ => self.colours.border_style,
					}
				})
				.borders(Borders::ALL)
				.border_style(match app_state.current_widget_selected {
					app::WidgetPosition::Cpu => self.colours.highlighted_border_style,
					_ => self.colours.border_style,
				}),
		)
		.header_style(self.colours.table_header_style)
		.widths(
			&(intrinsic_widths
				.iter()
				.map(|calculated_width| Constraint::Length(*calculated_width as u16))
				.collect::<Vec<_>>()),
		)
		.render(f, draw_loc);
	}

	fn draw_memory_graph<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &app::App, draw_loc: Rect,
	) {
		let mem_data: &[(f64, f64)] = &(app_state.canvas_data.mem_data);
		let swap_data: &[(f64, f64)] = &(app_state.canvas_data.swap_data);

		let x_axis: Axis<String> = Axis::default().bounds([0.0, TIME_STARTS_FROM as f64]);

		// Offset as the zero value isn't drawn otherwise...
		let y_axis: Axis<&str> = Axis::default()
			.style(self.colours.graph_style)
			.labels_style(self.colours.graph_style)
			.bounds([-0.5, 100.5])
			.labels(&["0%", "100%"]);

		let mut mem_canvas_vec: Vec<Dataset> = vec![Dataset::default()
			.name(&app_state.canvas_data.mem_label)
			.marker(if app_state.app_config_fields.use_dot {
				Marker::Dot
			} else {
				Marker::Braille
			})
			.style(self.colours.ram_style)
			.data(&mem_data)];

		if !(&swap_data).is_empty() {
			mem_canvas_vec.push(
				Dataset::default()
					.name(&app_state.canvas_data.swap_label)
					.marker(if app_state.app_config_fields.use_dot {
						Marker::Dot
					} else {
						Marker::Braille
					})
					.style(self.colours.swap_style)
					.data(&swap_data),
			);
		}

		let title = if app_state.is_expanded {
			const TITLE_BASE: &str = " Memory ── Esc to go back ";
			let repeat_num = max(
				0,
				draw_loc.width as i32 - TITLE_BASE.chars().count() as i32 - 2,
			);
			let result_title = format!(
				" Memory ─{}─ Esc to go back ",
				"─".repeat(repeat_num as usize)
			);

			result_title
		} else {
			" Memory ".to_string()
		};

		Chart::default()
			.block(
				Block::default()
					.title(&title)
					.title_style(if app_state.is_expanded {
						self.colours.highlighted_border_style
					} else {
						self.colours.widget_title_style
					})
					.borders(Borders::ALL)
					.border_style(match app_state.current_widget_selected {
						app::WidgetPosition::Mem => self.colours.highlighted_border_style,
						_ => self.colours.border_style,
					}),
			)
			.x_axis(x_axis)
			.y_axis(y_axis)
			.datasets(&mem_canvas_vec)
			.render(f, draw_loc);
	}

	fn draw_network_graph<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &app::App, draw_loc: Rect,
	) {
		let network_data_rx: &[(f64, f64)] = &(app_state.canvas_data.network_data_rx);
		let network_data_tx: &[(f64, f64)] = &(app_state.canvas_data.network_data_tx);

		let x_axis: Axis<String> = Axis::default().bounds([0.0, 60_000.0]);
		let y_axis = Axis::default()
			.style(self.colours.graph_style)
			.labels_style(self.colours.graph_style)
			.bounds([-0.5, 30_f64])
			.labels(&["0B", "1KiB", "1MiB", "1GiB"]);

		let title = if app_state.is_expanded {
			const TITLE_BASE: &str = " Network ── Esc to go back ";
			let repeat_num = max(
				0,
				draw_loc.width as i32 - TITLE_BASE.chars().count() as i32 - 2,
			);
			let result_title = format!(
				" Network ─{}─ Esc to go back ",
				"─".repeat(repeat_num as usize)
			);

			result_title
		} else {
			" Network ".to_string()
		};

		Chart::default()
			.block(
				Block::default()
					.title(&title)
					.title_style(if app_state.is_expanded {
						self.colours.highlighted_border_style
					} else {
						self.colours.widget_title_style
					})
					.borders(Borders::ALL)
					.border_style(match app_state.current_widget_selected {
						app::WidgetPosition::Network => self.colours.highlighted_border_style,
						_ => self.colours.border_style,
					}),
			)
			.x_axis(x_axis)
			.y_axis(y_axis)
			.datasets(&[
				Dataset::default()
					.name(&format!("RX: {:7}", app_state.canvas_data.rx_display))
					.marker(if app_state.app_config_fields.use_dot {
						Marker::Dot
					} else {
						Marker::Braille
					})
					.style(self.colours.rx_style)
					.data(&network_data_rx),
				Dataset::default()
					.name(&format!("TX: {:7}", app_state.canvas_data.tx_display))
					.marker(if app_state.app_config_fields.use_dot {
						Marker::Dot
					} else {
						Marker::Braille
					})
					.style(self.colours.tx_style)
					.data(&network_data_tx),
				Dataset::default().name(&format!(
					"Total RX: {:7}",
					app_state.canvas_data.total_rx_display
				)),
				Dataset::default().name(&format!(
					"Total TX: {:7}",
					app_state.canvas_data.total_tx_display
				)),
			])
			.render(f, draw_loc);
	}

	fn draw_network_labels<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect,
	) {
		let rx_display = &app_state.canvas_data.rx_display;
		let tx_display = &app_state.canvas_data.tx_display;
		let total_rx_display = &app_state.canvas_data.total_rx_display;
		let total_tx_display = &app_state.canvas_data.total_tx_display;

		// Gross but I need it to work...
		let total_network = vec![vec![
			rx_display,
			tx_display,
			total_rx_display,
			total_tx_display,
		]];
		let mapped_network = total_network
			.iter()
			.map(|val| Row::StyledData(val.iter(), self.colours.text_style));

		// Calculate widths
		let width_ratios: Vec<f64> = vec![0.25, 0.25, 0.25, 0.25];
		let lens: &Vec<usize> = &NETWORK_HEADERS_LENS;
		let width = f64::from(draw_loc.width);

		let variable_intrinsic_results =
			get_variable_intrinsic_widths(width as u16, &width_ratios, lens);
		let intrinsic_widths = &(variable_intrinsic_results.0)[0..variable_intrinsic_results.1];

		// Draw
		Table::new(NETWORK_HEADERS.iter(), mapped_network)
			.block(Block::default().borders(Borders::ALL).border_style(
				match app_state.current_widget_selected {
					app::WidgetPosition::Network => self.colours.highlighted_border_style,
					_ => self.colours.border_style,
				},
			))
			.header_style(self.colours.table_header_style)
			.style(self.colours.text_style)
			.widths(
				&(intrinsic_widths
					.iter()
					.map(|calculated_width| Constraint::Length(*calculated_width as u16))
					.collect::<Vec<_>>()),
			)
			.render(f, draw_loc);
	}

	fn draw_temp_table<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect,
	) {
		let temp_sensor_data: &[Vec<String>] = &(app_state.canvas_data.temp_sensor_data);

		let num_rows = max(0, i64::from(draw_loc.height) - 5) as u64;
		let start_position = get_start_position(
			num_rows,
			&(app_state.app_scroll_positions.scroll_direction),
			&mut app_state
				.app_scroll_positions
				.temp_scroll_state
				.previous_scroll_position,
			app_state
				.app_scroll_positions
				.temp_scroll_state
				.current_scroll_position,
		);

		let sliced_vec = &(temp_sensor_data[start_position as usize..]);
		let mut temp_row_counter: i64 = 0;

		let temperature_rows = sliced_vec.iter().map(|temp_row| {
			Row::StyledData(
				temp_row.iter(),
				match app_state.current_widget_selected {
					app::WidgetPosition::Temp => {
						if temp_row_counter as u64
							== app_state
								.app_scroll_positions
								.temp_scroll_state
								.current_scroll_position - start_position
						{
							temp_row_counter = -1;
							self.colours.currently_selected_text_style
						} else {
							if temp_row_counter >= 0 {
								temp_row_counter += 1;
							}
							self.colours.text_style
						}
					}
					_ => self.colours.text_style,
				},
			)
		});

		// Calculate widths
		let width = f64::from(draw_loc.width);
		let width_ratios = [0.5, 0.5];
		let variable_intrinsic_results =
			get_variable_intrinsic_widths(width as u16, &width_ratios, &TEMP_HEADERS_LENS);
		let intrinsic_widths = &(variable_intrinsic_results.0)[0..variable_intrinsic_results.1];

		let title = if app_state.is_expanded {
			const TITLE_BASE: &str = " Temperatures ── Esc to go back ";
			let repeat_num = max(
				0,
				draw_loc.width as i32 - TITLE_BASE.chars().count() as i32 - 2,
			);
			let result_title = format!(
				" Temperatures ─{}─ Esc to go back ",
				"─".repeat(repeat_num as usize)
			);

			result_title
		} else {
			" Temperatures ".to_string()
		};

		// Draw
		Table::new(TEMP_HEADERS.iter(), temperature_rows)
			.block(
				Block::default()
					.title(&title)
					.title_style(if app_state.is_expanded {
						self.colours.highlighted_border_style
					} else {
						self.colours.widget_title_style
					})
					.borders(Borders::ALL)
					.border_style(match app_state.current_widget_selected {
						app::WidgetPosition::Temp => self.colours.highlighted_border_style,
						_ => self.colours.border_style,
					}),
			)
			.header_style(self.colours.table_header_style)
			.widths(
				&(intrinsic_widths
					.iter()
					.map(|calculated_width| Constraint::Length(*calculated_width as u16))
					.collect::<Vec<_>>()),
			)
			.render(f, draw_loc);
	}

	fn draw_disk_table<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect,
	) {
		let disk_data: &[Vec<String>] = &(app_state.canvas_data.disk_data);
		let num_rows = max(0, i64::from(draw_loc.height) - 5) as u64;
		let start_position = get_start_position(
			num_rows,
			&(app_state.app_scroll_positions.scroll_direction),
			&mut app_state
				.app_scroll_positions
				.disk_scroll_state
				.previous_scroll_position,
			app_state
				.app_scroll_positions
				.disk_scroll_state
				.current_scroll_position,
		);

		let sliced_vec = &disk_data[start_position as usize..];
		let mut disk_counter: i64 = 0;

		let disk_rows = sliced_vec.iter().map(|disk| {
			Row::StyledData(
				disk.iter(),
				match app_state.current_widget_selected {
					app::WidgetPosition::Disk => {
						if disk_counter as u64
							== app_state
								.app_scroll_positions
								.disk_scroll_state
								.current_scroll_position - start_position
						{
							disk_counter = -1;
							self.colours.currently_selected_text_style
						} else {
							if disk_counter >= 0 {
								disk_counter += 1;
							}
							self.colours.text_style
						}
					}
					_ => self.colours.text_style,
				},
			)
		});

		// Calculate widths
		// TODO: [PRETTY] Ellipsis on strings?
		let width = f64::from(draw_loc.width);
		let width_ratios = [0.2, 0.15, 0.13, 0.13, 0.13, 0.13, 0.13];
		let variable_intrinsic_results =
			get_variable_intrinsic_widths(width as u16, &width_ratios, &DISK_HEADERS_LENS);
		let intrinsic_widths = &variable_intrinsic_results.0[0..variable_intrinsic_results.1];

		let title = if app_state.is_expanded {
			const TITLE_BASE: &str = " Disk ── Esc to go back ";
			let repeat_num = max(
				0,
				draw_loc.width as i32 - TITLE_BASE.chars().count() as i32 - 2,
			);
			let result_title = format!(
				" Disk ─{}─ Esc to go back ",
				"─".repeat(repeat_num as usize)
			);

			result_title
		} else {
			" Disk ".to_string()
		};

		// Draw!
		Table::new(DISK_HEADERS.iter(), disk_rows)
			.block(
				Block::default()
					.title(&title)
					.title_style(if app_state.is_expanded {
						self.colours.highlighted_border_style
					} else {
						self.colours.widget_title_style
					})
					.borders(Borders::ALL)
					.border_style(match app_state.current_widget_selected {
						app::WidgetPosition::Disk => self.colours.highlighted_border_style,
						_ => self.colours.border_style,
					}),
			)
			.header_style(self.colours.table_header_style)
			.widths(
				&(intrinsic_widths
					.iter()
					.map(|calculated_width| Constraint::Length(*calculated_width as u16))
					.collect::<Vec<_>>()),
			)
			.render(f, draw_loc);
	}

	fn draw_search_field<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect,
	) {
		let width = max(0, draw_loc.width as i64 - 34) as u64;
		let query = app_state.get_current_search_query();
		let shrunk_query = if query.len() < width as usize {
			query
		} else {
			&query[(query.len() - width as usize)..]
		};

		let cursor_position = app_state.get_cursor_position();

		let query_with_cursor: Vec<Text> =
			if let app::WidgetPosition::ProcessSearch = app_state.current_widget_selected {
				if cursor_position >= query.len() {
					let mut q = vec![Text::styled(
						shrunk_query.to_string(),
						self.colours.text_style,
					)];

					q.push(Text::styled(
						" ".to_string(),
						self.colours.currently_selected_text_style,
					));

					q
				} else {
					shrunk_query
						.chars()
						.enumerate()
						.map(|(itx, c)| {
							if let app::WidgetPosition::ProcessSearch =
								app_state.current_widget_selected
							{
								if itx == cursor_position {
									return Text::styled(
										c.to_string(),
										self.colours.currently_selected_text_style,
									);
								}
							}
							Text::styled(c.to_string(), self.colours.text_style)
						})
						.collect::<Vec<_>>()
				}
			} else {
				vec![Text::styled(
					shrunk_query.to_string(),
					self.colours.text_style,
				)]
			};

		let mut search_text = vec![if app_state.is_grouped() {
			Text::styled("Search by Name: ", self.colours.table_header_style)
		} else if app_state.process_search_state.is_searching_with_pid {
			Text::styled(
				"Search by PID (Tab for Name): ",
				self.colours.table_header_style,
			)
		} else {
			Text::styled(
				"Search by Name (Tab for PID): ",
				self.colours.table_header_style,
			)
		}];

		// Text options shamelessly stolen from VS Code.
		let mut option_text = vec![];
		for _ in 0..(draw_loc.height - 3) {
			option_text.push(Text::raw("\n"));
		}

		let option_row = vec![
			Text::styled("Match Case (Alt+C)", self.colours.table_header_style),
			if !app_state.process_search_state.is_ignoring_case {
				Text::styled("[*]", self.colours.table_header_style)
			} else {
				Text::styled("[ ]", self.colours.table_header_style)
			},
			Text::styled("     ", self.colours.table_header_style),
			Text::styled("Match Whole Word (Alt+W)", self.colours.table_header_style),
			if app_state.process_search_state.is_searching_whole_word {
				Text::styled("[*]", self.colours.table_header_style)
			} else {
				Text::styled("[ ]", self.colours.table_header_style)
			},
			Text::styled("     ", self.colours.table_header_style),
			Text::styled("Use Regex (Alt+R)", self.colours.table_header_style),
			if app_state.process_search_state.is_searching_with_regex {
				Text::styled("[*]", self.colours.table_header_style)
			} else {
				Text::styled("[ ]", self.colours.table_header_style)
			},
		];
		option_text.extend(option_row);

		search_text.extend(query_with_cursor);
		search_text.extend(option_text);

		const TITLE_BASE: &str = " Esc to close ";
		let repeat_num = max(
			0,
			draw_loc.width as i32 - TITLE_BASE.chars().count() as i32 - 2,
		);
		let title = format!("{} Esc to close ", "─".repeat(repeat_num as usize));

		let current_border_style: Style = if app_state
			.process_search_state
			.search_state
			.is_invalid_search
		{
			Style::default().fg(Color::Rgb(255, 0, 0))
		} else {
			match app_state.current_widget_selected {
				app::WidgetPosition::ProcessSearch => self.colours.highlighted_border_style,
				_ => self.colours.border_style,
			}
		};

		Paragraph::new(search_text.iter())
			.block(
				Block::default()
					.borders(Borders::ALL)
					.title(&title)
					.title_style(current_border_style)
					.border_style(current_border_style),
			)
			.style(self.colours.text_style)
			.alignment(Alignment::Left)
			.wrap(false)
			.render(f, draw_loc);
	}

	fn draw_processes_table<B: backend::Backend>(
		&self, f: &mut Frame<B>, app_state: &mut app::App, draw_loc: Rect,
	) {
		let process_data: &[ConvertedProcessData] = &app_state.canvas_data.finalized_process_data;

		// Admittedly this is kinda a hack... but we need to:
		// * Scroll
		// * Show/hide elements based on scroll position
		//
		// As such, we use a process_counter to know when we've
		// hit the process we've currently scrolled to.
		// We also need to move the list - we can
		// do so by hiding some elements!
		let num_rows = max(0, i64::from(draw_loc.height) - 5) as u64;

		let position = get_start_position(
			num_rows,
			&(app_state.app_scroll_positions.scroll_direction),
			&mut app_state
				.app_scroll_positions
				.process_scroll_state
				.previous_scroll_position,
			app_state
				.app_scroll_positions
				.process_scroll_state
				.current_scroll_position,
		);

		// Sanity check
		let start_position = if position >= process_data.len() as u64 {
			std::cmp::max(0, process_data.len() as i64 - 1) as u64
		} else {
			position
		};

		let sliced_vec = &(process_data[start_position as usize..]);
		let mut process_counter: i64 = 0;

		// Draw!
		let process_rows = sliced_vec.iter().map(|process| {
			let stringified_process_vec: Vec<String> = vec![
				if app_state.is_grouped() {
					process.group_pids.len().to_string()
				} else {
					process.pid.to_string()
				},
				process.name.clone(),
				format!("{:.1}%", process.cpu_usage),
				format!("{:.1}%", process.mem_usage),
			];
			Row::StyledData(
				stringified_process_vec.into_iter(),
				match app_state.current_widget_selected {
					app::WidgetPosition::Process => {
						if process_counter as u64
							== app_state
								.app_scroll_positions
								.process_scroll_state
								.current_scroll_position - start_position
						{
							process_counter = -1;
							self.colours.currently_selected_text_style
						} else {
							if process_counter >= 0 {
								process_counter += 1;
							}
							self.colours.text_style
						}
					}
					_ => self.colours.text_style,
				},
			)
		});

		use app::data_harvester::processes::ProcessSorting;
		let mut pid_or_name = if app_state.is_grouped() {
			"Count"
		} else {
			"PID(p)"
		}
		.to_string();
		let mut name = "Name(n)".to_string();
		let mut cpu = "CPU%(c)".to_string();
		let mut mem = "Mem%(m)".to_string();

		let direction_val = if app_state.process_sorting_reverse {
			"⯆".to_string()
		} else {
			"⯅".to_string()
		};

		match app_state.process_sorting_type {
			ProcessSorting::CPU => cpu += &direction_val,
			ProcessSorting::MEM => mem += &direction_val,
			ProcessSorting::PID => pid_or_name += &direction_val,
			ProcessSorting::NAME => name += &direction_val,
		};

		let process_headers = [pid_or_name, name, cpu, mem];
		let process_headers_lens: Vec<usize> = process_headers
			.iter()
			.map(|entry| entry.len())
			.collect::<Vec<_>>();

		// Calculate widths
		let width = f64::from(draw_loc.width);
		let width_ratios = [0.2, 0.4, 0.2, 0.2];
		let variable_intrinsic_results =
			get_variable_intrinsic_widths(width as u16, &width_ratios, &process_headers_lens);
		let intrinsic_widths = &(variable_intrinsic_results.0)[0..variable_intrinsic_results.1];

		let title =
			if app_state.is_expanded && !app_state.process_search_state.search_state.is_enabled {
				const TITLE_BASE: &str = " Processes ── Esc to go back ";
				let repeat_num = max(
					0,
					draw_loc.width as i32 - TITLE_BASE.chars().count() as i32 - 2,
				);
				let result_title = format!(
					" Processes ─{}─ Esc to go back ",
					"─".repeat(repeat_num as usize)
				);

				result_title
			} else {
				" Processes ".to_string()
			};

		Table::new(process_headers.iter(), process_rows)
			.block(
				Block::default()
					.title(&title)
					.title_style(if app_state.is_expanded {
						match app_state.current_widget_selected {
							app::WidgetPosition::Process => self.colours.highlighted_border_style,
							_ => self.colours.border_style,
						}
					} else {
						self.colours.widget_title_style
					})
					.borders(Borders::ALL)
					.border_style(match app_state.current_widget_selected {
						app::WidgetPosition::Process => self.colours.highlighted_border_style,
						_ => self.colours.border_style,
					}),
			)
			.header_style(self.colours.table_header_style)
			.widths(
				&(intrinsic_widths
					.iter()
					.map(|calculated_width| Constraint::Length(*calculated_width as u16))
					.collect::<Vec<_>>()),
			)
			.render(f, draw_loc);
	}
}