use crate::{
common::{Gid, Uid},
sys::{Component, Disk, Networks, Process, Processor},
};
use crate::{
DiskType, DiskUsage, LoadAvg, NetworksIter, Pid, ProcessStatus, RefreshKind, Signal, User,
};
use std::collections::HashMap;
use std::ffi::OsStr;
use std::fmt::Debug;
use std::path::Path;
pub trait DiskExt: Debug {
fn type_(&self) -> DiskType;
fn name(&self) -> &OsStr;
fn file_system(&self) -> &[u8];
fn mount_point(&self) -> &Path;
fn total_space(&self) -> u64;
fn available_space(&self) -> u64;
fn is_removable(&self) -> bool;
fn refresh(&mut self) -> bool;
}
pub trait ProcessExt: Debug {
#[doc(hidden)]
fn new(pid: Pid, parent: Option<Pid>, start_time: u64) -> Self;
fn kill(&self, signal: Signal) -> bool;
fn name(&self) -> &str;
fn cmd(&self) -> &[String];
fn exe(&self) -> &Path;
fn pid(&self) -> Pid;
fn environ(&self) -> &[String];
fn cwd(&self) -> &Path;
fn root(&self) -> &Path;
fn memory(&self) -> u64;
fn virtual_memory(&self) -> u64;
fn parent(&self) -> Option<Pid>;
fn status(&self) -> ProcessStatus;
fn start_time(&self) -> u64;
fn cpu_usage(&self) -> f32;
fn disk_usage(&self) -> DiskUsage;
}
pub trait ProcessorExt: Debug {
fn cpu_usage(&self) -> f32;
fn name(&self) -> &str;
fn vendor_id(&self) -> &str;
fn brand(&self) -> &str;
fn frequency(&self) -> u64;
}
pub trait SystemExt: Sized + Debug + Default {
const IS_SUPPORTED: bool;
fn new() -> Self {
Self::new_with_specifics(RefreshKind::new())
}
fn new_all() -> Self {
Self::new_with_specifics(RefreshKind::everything())
}
fn new_with_specifics(refreshes: RefreshKind) -> Self;
fn refresh_specifics(&mut self, refreshes: RefreshKind) {
if refreshes.memory() {
self.refresh_memory();
}
if refreshes.cpu() {
self.refresh_cpu();
}
if refreshes.components_list() {
self.refresh_components_list();
} else if refreshes.components() {
self.refresh_components();
}
if refreshes.networks_list() {
self.refresh_networks_list();
} else if refreshes.networks() {
self.refresh_networks();
}
if refreshes.processes() {
self.refresh_processes();
}
if refreshes.disks_list() {
self.refresh_disks_list();
} else if refreshes.disks() {
self.refresh_disks();
}
if refreshes.users_list() {
self.refresh_users_list();
}
}
fn refresh_system(&mut self) {
self.refresh_memory();
self.refresh_cpu();
self.refresh_components();
}
fn refresh_memory(&mut self);
fn refresh_cpu(&mut self);
fn refresh_components(&mut self) {
for component in self.components_mut() {
component.refresh();
}
}
fn refresh_components_list(&mut self);
fn refresh_processes(&mut self);
fn refresh_process(&mut self, pid: Pid) -> bool;
fn refresh_disks(&mut self) {
for disk in self.disks_mut() {
disk.refresh();
}
}
fn refresh_disks_list(&mut self);
fn refresh_users_list(&mut self);
fn refresh_networks(&mut self) {
self.networks_mut().refresh();
}
fn refresh_networks_list(&mut self) {
self.networks_mut().refresh_networks_list();
}
fn refresh_all(&mut self) {
self.refresh_system();
self.refresh_processes();
self.refresh_disks();
self.refresh_networks();
}
fn processes(&self) -> &HashMap<Pid, Process>;
fn process(&self, pid: Pid) -> Option<&Process>;
fn process_by_name(&self, name: &str) -> Vec<&Process> {
let mut ret = vec![];
for val in self.processes().values() {
if val.name().contains(name) {
ret.push(val);
}
}
ret
}
fn global_processor_info(&self) -> &Processor;
fn processors(&self) -> &[Processor];
fn physical_core_count(&self) -> Option<usize>;
fn total_memory(&self) -> u64;
fn free_memory(&self) -> u64;
fn available_memory(&self) -> u64;
fn used_memory(&self) -> u64;
fn total_swap(&self) -> u64;
fn free_swap(&self) -> u64;
fn used_swap(&self) -> u64;
fn components(&self) -> &[Component];
fn components_mut(&mut self) -> &mut [Component];
fn disks(&self) -> &[Disk];
fn users(&self) -> &[User];
fn disks_mut(&mut self) -> &mut [Disk];
fn networks(&self) -> &Networks;
fn networks_mut(&mut self) -> &mut Networks;
fn uptime(&self) -> u64;
fn boot_time(&self) -> u64;
fn load_average(&self) -> LoadAvg;
fn name(&self) -> Option<String>;
fn kernel_version(&self) -> Option<String>;
fn os_version(&self) -> Option<String>;
fn long_os_version(&self) -> Option<String>;
fn host_name(&self) -> Option<String>;
}
pub trait NetworkExt: Debug {
fn received(&self) -> u64;
fn total_received(&self) -> u64;
fn transmitted(&self) -> u64;
fn total_transmitted(&self) -> u64;
fn packets_received(&self) -> u64;
fn total_packets_received(&self) -> u64;
fn packets_transmitted(&self) -> u64;
fn total_packets_transmitted(&self) -> u64;
fn errors_on_received(&self) -> u64;
fn total_errors_on_received(&self) -> u64;
fn errors_on_transmitted(&self) -> u64;
fn total_errors_on_transmitted(&self) -> u64;
}
pub trait NetworksExt: Debug {
fn iter(&self) -> NetworksIter;
fn refresh_networks_list(&mut self);
fn refresh(&mut self);
}
pub trait ComponentExt: Debug {
fn temperature(&self) -> f32;
fn max(&self) -> f32;
fn critical(&self) -> Option<f32>;
fn label(&self) -> &str;
fn refresh(&mut self);
}
pub trait UserExt: Debug {
fn uid(&self) -> Uid;
fn gid(&self) -> Gid;
fn name(&self) -> &str;
fn groups(&self) -> &[String];
}