bottom/src/app.rs

pub mod data_harvester;
use data_harvester::{processes, temperature};
use std::time::Instant;

pub mod data_farmer;
use data_farmer::*;

use crate::{canvas, constants, data_conversion::ConvertedProcessHarvest, utils::error::Result};

mod process_killer;

#[derive(Clone, Copy)]
pub enum ApplicationPosition {
	Cpu,
	Mem,
	Disk,
	Temp,
	Network,
	Process,
	ProcessSearch,
}

#[derive(Debug)]
pub enum ScrollDirection {
	// UP means scrolling up --- this usually DECREMENTS
	UP,
	// DOWN means scrolling down --- this usually INCREMENTS
	DOWN,
}

lazy_static! {
	static ref BASE_REGEX: std::result::Result<regex::Regex, regex::Error> =
		regex::Regex::new(".*");
}

/// AppConfigFields is meant to cover basic fields that would normally be set
/// by config files or launch options.  Don't need to be mutable (set and forget).
pub struct AppConfigFields {
	pub update_rate_in_milliseconds: u64,
	pub temperature_type: temperature::TemperatureType,
	pub use_dot: bool,
}

/// AppScrollWidgetState deals with fields for a scrollable app's current state.
pub struct AppScrollWidgetState {
	pub widget_scroll_position: i64,
}

/// AppSearchState only deals with the search's state.
pub struct AppSearchState {}

// TODO: [OPT] Group like fields together... this is kinda gross to step through
pub struct App {
	// Sorting
	pub process_sorting_type: processes::ProcessSorting,
	pub process_sorting_reverse: bool,
	pub update_process_gui: bool,
	// Positioning
	pub scroll_direction: ScrollDirection,
	pub currently_selected_process_position: i64,
	pub currently_selected_disk_position: i64,
	pub currently_selected_temperature_position: i64,
	pub currently_selected_cpu_table_position: i64,
	pub previous_disk_position: i64,
	pub previous_temp_position: i64,
	pub previous_process_position: i64,
	pub previous_cpu_table_position: i64,
	pub temperature_type: temperature::TemperatureType,
	pub update_rate_in_milliseconds: u64,
	pub show_average_cpu: bool,
	pub current_application_position: ApplicationPosition,
	pub data: data_harvester::Data,
	awaiting_second_char: bool,
	second_char: char,
	pub use_dot: bool,
	pub show_help: bool,
	pub show_dd: bool,
	pub dd_err: Option<String>,
	to_delete_process_list: Option<Vec<ConvertedProcessHarvest>>,
	pub is_frozen: bool,
	pub left_legend: bool,
	pub use_current_cpu_total: bool,
	last_key_press: Instant,
	pub canvas_data: canvas::CanvasData,
	enable_grouping: bool,
	enable_searching: bool,
	current_search_query: String,
	searching_pid: bool,
	pub use_simple: bool,
	current_regex: std::result::Result<regex::Regex, regex::Error>,
	current_cursor_position: usize,
	pub data_collection: DataCollection,
}

impl App {
	pub fn new(
		show_average_cpu: bool, temperature_type: temperature::TemperatureType,
		update_rate_in_milliseconds: u64, use_dot: bool, left_legend: bool,
		use_current_cpu_total: bool,
	) -> App {
		App {
			process_sorting_type: processes::ProcessSorting::CPU,
			process_sorting_reverse: true,
			update_process_gui: false,
			temperature_type,
			update_rate_in_milliseconds,
			show_average_cpu,
			current_application_position: ApplicationPosition::Process,
			scroll_direction: ScrollDirection::DOWN,
			currently_selected_process_position: 0,
			currently_selected_disk_position: 0,
			currently_selected_temperature_position: 0,
			currently_selected_cpu_table_position: 0,
			previous_process_position: 0,
			previous_disk_position: 0,
			previous_temp_position: 0,
			previous_cpu_table_position: 0,
			data: data_harvester::Data::default(),
			awaiting_second_char: false,
			second_char: ' ',
			use_dot,
			show_help: false,
			show_dd: false,
			dd_err: None,
			to_delete_process_list: None,
			is_frozen: false,
			left_legend,
			use_current_cpu_total,
			last_key_press: Instant::now(),
			canvas_data: canvas::CanvasData::default(),
			enable_grouping: false,
			enable_searching: false,
			current_search_query: String::default(),
			searching_pid: false,
			use_simple: false,
			current_regex: BASE_REGEX.clone(), //TODO: [OPT] seems like a thing we can switch to lifetimes to avoid cloning
			current_cursor_position: 0,
			data_collection: DataCollection::default(),
		}
	}

	pub fn reset(&mut self) {
		self.reset_multi_tap_keys();
		self.show_help = false;
		self.show_dd = false;
		if self.enable_searching {
			self.current_application_position = ApplicationPosition::Process;
			self.enable_searching = false;
		}
		self.current_search_query = String::new();
		self.searching_pid = false;
		self.to_delete_process_list = None;
		self.dd_err = None;
	}

	pub fn on_esc(&mut self) {
		self.reset_multi_tap_keys();
		if self.is_in_dialog() {
			self.show_help = false;
			self.show_dd = false;
			self.to_delete_process_list = None;
			self.dd_err = None;
		} else if self.enable_searching {
			self.current_application_position = ApplicationPosition::Process;
			self.enable_searching = false;
		}
	}

	fn reset_multi_tap_keys(&mut self) {
		self.awaiting_second_char = false;
		self.second_char = ' ';
	}

	fn is_in_dialog(&self) -> bool {
		self.show_help || self.show_dd
	}

	pub fn toggle_grouping(&mut self) {
		// Disallow usage whilst in a dialog and only in processes
		if !self.is_in_dialog() {
			if let ApplicationPosition::Process = self.current_application_position {
				self.enable_grouping = !(self.enable_grouping);
			}
		}
	}

	pub fn on_tab(&mut self) {
		match self.current_application_position {
			ApplicationPosition::Process => self.toggle_grouping(),
			ApplicationPosition::Disk => {}
			_ => {}
		}
	}

	pub fn is_grouped(&self) -> bool {
		self.enable_grouping
	}

	pub fn enable_searching(&mut self) {
		if !self.is_in_dialog() {
			match self.current_application_position {
				ApplicationPosition::Process | ApplicationPosition::ProcessSearch => {
					// Toggle on
					self.enable_searching = true;
					self.current_application_position = ApplicationPosition::ProcessSearch;
				}
				_ => {}
			}
		}
	}

	pub fn is_searching(&self) -> bool {
		self.enable_searching
	}

	pub fn is_in_search_widget(&self) -> bool {
		if let ApplicationPosition::ProcessSearch = self.current_application_position {
			true
		} else {
			false
		}
	}

	pub fn search_with_pid(&mut self) {
		if !self.is_in_dialog() && self.is_searching() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
				self.searching_pid = true;
			}
		}
	}

	pub fn search_with_name(&mut self) {
		if !self.is_in_dialog() && self.is_searching() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
				self.searching_pid = false;
			}
		}
	}

	pub fn is_searching_with_pid(&self) -> bool {
		self.searching_pid
	}

	pub fn get_current_search_query(&self) -> &String {
		&self.current_search_query
	}

	pub fn toggle_simple_search(&mut self) {
		if !self.is_in_dialog() && self.is_searching() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
				self.use_simple = !self.use_simple;

				// Update to latest (when simple is on this is not updated)
				if !self.use_simple {
					self.current_regex = if self.current_search_query.is_empty() {
						BASE_REGEX.clone()
					} else {
						regex::Regex::new(&(self.current_search_query))
					};
				}

				// Force update to process display in GUI
				self.update_process_gui = true;
			}
		}
	}

	pub fn get_cursor_position(&self) -> usize {
		self.current_cursor_position
	}

	/// One of two functions allowed to run while in a dialog...
	pub fn on_enter(&mut self) {
		if self.show_dd {
			// If within dd...
			if self.dd_err.is_none() {
				// Also ensure that we didn't just fail a dd...
				let dd_result = self.kill_highlighted_process();
				if let Err(dd_err) = dd_result {
					// There was an issue... inform the user...
					self.dd_err = Some(dd_err.to_string());
				} else {
					self.show_dd = false;
				}
			}
		}
	}

	pub fn on_backspace(&mut self) {
		if let ApplicationPosition::ProcessSearch = self.current_application_position {
			if self.current_cursor_position > 0 {
				self.current_cursor_position -= 1;
				self.current_search_query
					.remove(self.current_cursor_position);

				if !self.use_simple {
					self.current_regex = if self.current_search_query.is_empty() {
						BASE_REGEX.clone()
					} else {
						regex::Regex::new(&(self.current_search_query))
					};
				}
				self.update_process_gui = true;
			}
		}
	}

	pub fn get_current_regex_matcher(&self) -> &std::result::Result<regex::Regex, regex::Error> {
		&self.current_regex
	}

	pub fn on_up_key(&mut self) {
		if !self.is_in_dialog() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
			} else {
				self.decrement_position_count();
			}
		}
	}

	pub fn on_down_key(&mut self) {
		if !self.is_in_dialog() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
			} else {
				self.increment_position_count();
			}
		}
	}

	pub fn on_left_key(&mut self) {
		if !self.is_in_dialog() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
				if self.current_cursor_position > 0 {
					self.current_cursor_position -= 1;
				}
			}
		}
	}

	pub fn on_right_key(&mut self) {
		if !self.is_in_dialog() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
				if self.current_cursor_position < self.current_search_query.len() {
					self.current_cursor_position += 1;
				}
			}
		}
	}

	pub fn skip_cursor_beginning(&mut self) {
		if !self.is_in_dialog() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
				self.current_cursor_position = 0;
			}
		}
	}

	pub fn skip_cursor_end(&mut self) {
		if !self.is_in_dialog() {
			if let ApplicationPosition::ProcessSearch = self.current_application_position {
				self.current_cursor_position = self.current_search_query.len();
			}
		}
	}

	pub fn on_char_key(&mut self, caught_char: char) {
		// Forbid any char key presses when showing a dialog box...
		if !self.is_in_dialog() {
			let current_key_press_inst = Instant::now();
			if current_key_press_inst
				.duration_since(self.last_key_press)
				.as_millis() > constants::MAX_KEY_TIMEOUT_IN_MILLISECONDS
			{
				self.reset_multi_tap_keys();
			}
			self.last_key_press = current_key_press_inst;

			if let ApplicationPosition::ProcessSearch = self.current_application_position {
				self.current_search_query
					.insert(self.current_cursor_position, caught_char);
				self.current_cursor_position += 1;

				if !self.use_simple {
					self.current_regex = if self.current_search_query.is_empty() {
						BASE_REGEX.clone()
					} else {
						regex::Regex::new(&(self.current_search_query))
					};
				}
				self.update_process_gui = true;
			} else {
				match caught_char {
					'/' => {
						self.enable_searching();
					}
					'd' => {
						if let ApplicationPosition::Process = self.current_application_position {
							if self.awaiting_second_char && self.second_char == 'd' {
								self.awaiting_second_char = false;
								self.second_char = ' ';
								let current_process = if self.is_grouped() {
									let mut res: Vec<ConvertedProcessHarvest> = Vec::new();
									for pid in &self.canvas_data.grouped_process_data
										[self.currently_selected_process_position as usize]
										.group
									{
										let result = self
											.canvas_data
											.process_data
											.iter()
											.find(|p| p.pid == *pid);
										if let Some(process) = result {
											res.push((*process).clone());
										}
									}
									res
								} else {
									vec![self.canvas_data.process_data
										[self.currently_selected_process_position as usize]
										.clone()]
								};
								self.to_delete_process_list = Some(current_process);
								self.show_dd = true;
								self.reset_multi_tap_keys();
							} else {
								self.awaiting_second_char = true;
								self.second_char = 'd';
							}
						}
					}
					'g' => {
						if self.awaiting_second_char && self.second_char == 'g' {
							self.awaiting_second_char = false;
							self.second_char = ' ';
							self.skip_to_first();
						} else {
							self.awaiting_second_char = true;
							self.second_char = 'g';
						}
					}
					'G' => self.skip_to_last(),
					'k' => self.decrement_position_count(),
					'j' => self.increment_position_count(),
					'f' => {
						self.is_frozen = !self.is_frozen;
					}
					'c' => {
						match self.process_sorting_type {
							processes::ProcessSorting::CPU => {
								self.process_sorting_reverse = !self.process_sorting_reverse
							}
							_ => {
								self.process_sorting_type = processes::ProcessSorting::CPU;
								self.process_sorting_reverse = true;
							}
						}
						self.update_process_gui = true;
						self.currently_selected_process_position = 0;
					}
					'm' => {
						match self.process_sorting_type {
							processes::ProcessSorting::MEM => {
								self.process_sorting_reverse = !self.process_sorting_reverse
							}
							_ => {
								self.process_sorting_type = processes::ProcessSorting::MEM;
								self.process_sorting_reverse = true;
							}
						}
						self.update_process_gui = true;
						self.currently_selected_process_position = 0;
					}
					'p' => {
						// Disable if grouping
						if !self.enable_grouping {
							match self.process_sorting_type {
								processes::ProcessSorting::PID => {
									self.process_sorting_reverse = !self.process_sorting_reverse
								}
								_ => {
									self.process_sorting_type = processes::ProcessSorting::PID;
									self.process_sorting_reverse = false;
								}
							}
							self.update_process_gui = true;
							self.currently_selected_process_position = 0;
						}
					}
					'n' => {
						match self.process_sorting_type {
							processes::ProcessSorting::NAME => {
								self.process_sorting_reverse = !self.process_sorting_reverse
							}
							_ => {
								self.process_sorting_type = processes::ProcessSorting::NAME;
								self.process_sorting_reverse = false;
							}
						}
						self.update_process_gui = true;
						self.currently_selected_process_position = 0;
					}
					'?' => {
						self.show_help = true;
					}
					_ => {}
				}
				if self.awaiting_second_char && caught_char != self.second_char {
					self.awaiting_second_char = false;
				}
			}
		}
	}

	pub fn kill_highlighted_process(&mut self) -> Result<()> {
		// Technically unnecessary but this is a good check...
		if let ApplicationPosition::Process = self.current_application_position {
			if let Some(current_selected_processes) = &(self.to_delete_process_list) {
				for current_selected_process in current_selected_processes {
					process_killer::kill_process_given_pid(current_selected_process.pid)?;
				}
			}
			self.to_delete_process_list = None;
		}
		Ok(())
	}

	pub fn get_current_highlighted_process_list(&self) -> Option<Vec<ConvertedProcessHarvest>> {
		self.to_delete_process_list.clone()
	}

	// For now, these are hard coded --- in the future, they shouldn't be!
	//
	// General idea for now:
	// CPU -(down)> MEM
	// MEM -(down)> Network, -(right)> TEMP
	// TEMP -(down)> Disk, -(left)> MEM, -(up)> CPU
	// Disk -(down)> Processes OR PROC_SEARCH, -(left)> MEM, -(up)> TEMP
	// Network -(up)> MEM, -(right)> PROC
	// PROC -(up)> Disk OR PROC_SEARCH if enabled, -(left)> Network
	// PROC_SEARCH -(up)> Disk, -(down)> PROC, -(left)> Network
	pub fn move_left(&mut self) {
		if !self.is_in_dialog() {
			self.current_application_position = match self.current_application_position {
				ApplicationPosition::Process => ApplicationPosition::Network,
				ApplicationPosition::ProcessSearch => ApplicationPosition::Network,
				ApplicationPosition::Disk => ApplicationPosition::Mem,
				ApplicationPosition::Temp => ApplicationPosition::Mem,
				_ => self.current_application_position,
			};
			self.reset_multi_tap_keys();
		}
	}

	pub fn move_right(&mut self) {
		if !self.is_in_dialog() {
			self.current_application_position = match self.current_application_position {
				ApplicationPosition::Mem => ApplicationPosition::Temp,
				ApplicationPosition::Network => ApplicationPosition::Process,
				_ => self.current_application_position,
			};
			self.reset_multi_tap_keys();
		}
	}

	pub fn move_up(&mut self) {
		if !self.is_in_dialog() {
			self.current_application_position = match self.current_application_position {
				ApplicationPosition::Mem => ApplicationPosition::Cpu,
				ApplicationPosition::Network => ApplicationPosition::Mem,
				ApplicationPosition::Process => {
					if self.is_searching() {
						ApplicationPosition::ProcessSearch
					} else {
						ApplicationPosition::Disk
					}
				}
				ApplicationPosition::ProcessSearch => ApplicationPosition::Disk,
				ApplicationPosition::Temp => ApplicationPosition::Cpu,
				ApplicationPosition::Disk => ApplicationPosition::Temp,
				_ => self.current_application_position,
			};
			self.reset_multi_tap_keys();
		}
	}

	pub fn move_down(&mut self) {
		if !self.is_in_dialog() {
			self.current_application_position = match self.current_application_position {
				ApplicationPosition::Cpu => ApplicationPosition::Mem,
				ApplicationPosition::Mem => ApplicationPosition::Network,
				ApplicationPosition::Temp => ApplicationPosition::Disk,
				ApplicationPosition::Disk => {
					if self.is_searching() {
						ApplicationPosition::ProcessSearch
					} else {
						ApplicationPosition::Process
					}
				}
				ApplicationPosition::ProcessSearch => ApplicationPosition::Process,
				_ => self.current_application_position,
			};
			self.reset_multi_tap_keys();
		}
	}

	pub fn skip_to_first(&mut self) {
		if !self.is_in_dialog() {
			match self.current_application_position {
				ApplicationPosition::Process => self.currently_selected_process_position = 0,
				ApplicationPosition::Temp => self.currently_selected_temperature_position = 0,
				ApplicationPosition::Disk => self.currently_selected_disk_position = 0,
				ApplicationPosition::Cpu => self.currently_selected_cpu_table_position = 0,

				_ => {}
			}
			self.scroll_direction = ScrollDirection::UP;
			self.reset_multi_tap_keys();
		}
	}

	pub fn skip_to_last(&mut self) {
		if !self.is_in_dialog() {
			match self.current_application_position {
				ApplicationPosition::Process => {
					self.currently_selected_process_position =
						self.data.list_of_processes.len() as i64 - 1
				}
				ApplicationPosition::Temp => {
					self.currently_selected_temperature_position =
						self.data.temperature_sensors.len() as i64 - 1
				}
				ApplicationPosition::Disk => {
					self.currently_selected_disk_position = self.data.disks.len() as i64 - 1
				}
				ApplicationPosition::Cpu => {
					self.currently_selected_cpu_table_position =
						self.canvas_data.cpu_data.len() as i64 - 1;
				}
				_ => {}
			}
			self.scroll_direction = ScrollDirection::DOWN;
			self.reset_multi_tap_keys();
		}
	}

	pub fn decrement_position_count(&mut self) {
		if !self.is_in_dialog() {
			match self.current_application_position {
				ApplicationPosition::Process => self.change_process_position(-1),
				ApplicationPosition::Temp => self.change_temp_position(-1),
				ApplicationPosition::Disk => self.change_disk_position(-1),
				ApplicationPosition::Cpu => self.change_cpu_table_position(-1), // TODO: Temporary, may change if we add scaling
				_ => {}
			}
			self.scroll_direction = ScrollDirection::UP;
			self.reset_multi_tap_keys();
		}
	}

	pub fn increment_position_count(&mut self) {
		if !self.is_in_dialog() {
			match self.current_application_position {
				ApplicationPosition::Process => self.change_process_position(1),
				ApplicationPosition::Temp => self.change_temp_position(1),
				ApplicationPosition::Disk => self.change_disk_position(1),
				ApplicationPosition::Cpu => self.change_cpu_table_position(1), // TODO: Temporary, may change if we add scaling
				_ => {}
			}
			self.scroll_direction = ScrollDirection::DOWN;
			self.reset_multi_tap_keys();
		}
	}

	fn change_cpu_table_position(&mut self, num_to_change_by: i64) {
		if self.currently_selected_cpu_table_position + num_to_change_by >= 0
			&& self.currently_selected_cpu_table_position + num_to_change_by
				< self.canvas_data.cpu_data.len() as i64
		{
			self.currently_selected_cpu_table_position += num_to_change_by;
		}
	}

	fn change_process_position(&mut self, num_to_change_by: i64) {
		if self.currently_selected_process_position + num_to_change_by >= 0
			&& self.currently_selected_process_position + num_to_change_by
				< self.data.list_of_processes.len() as i64
		{
			self.currently_selected_process_position += num_to_change_by;
		}
	}

	fn change_temp_position(&mut self, num_to_change_by: i64) {
		if self.currently_selected_temperature_position + num_to_change_by >= 0
			&& self.currently_selected_temperature_position + num_to_change_by
				< self.data.temperature_sensors.len() as i64
		{
			self.currently_selected_temperature_position += num_to_change_by;
		}
	}

	fn change_disk_position(&mut self, num_to_change_by: i64) {
		if self.currently_selected_disk_position + num_to_change_by >= 0
			&& self.currently_selected_disk_position + num_to_change_by
				< self.data.disks.len() as i64
		{
			self.currently_selected_disk_position += num_to_change_by;
		}
	}
}