use sysinfo::{ ProcessExt, SystemExt };\r
\r
-pub fn is_valheim_running() -> bool {\r
+use std::{ fs, time::SystemTime };\r
+\r
+use chrono::{ DateTime, offset::Local };\r
+\r
+#[cfg(target_os = "linux")]\r
+use systemd::journal;\r
+\r
+#[derive(Clone, Debug)]\r
+pub struct ValheimExe {\r
+ memory: u64, // [kB].\r
+ load_average_5min: f64, // [%].\r
+ uptime: u64, // [s].\r
+ world_size: u64, // [B].\r
+ active_players: Vec<String>,\r
+ last_backup: Option<SystemTime>,\r
+}\r
+\r
+impl ValheimExe {\r
+ pub fn format_memory(&self) -> String {\r
+ format_byte_size(self.memory * 1024, 2)\r
+ }\r
+\r
+ pub fn format_load_average(&self) -> String {\r
+ format!("{:.2} %", self.load_average_5min)\r
+ }\r
+\r
+ pub fn format_uptime(&self) -> String {\r
+ let mins = self.uptime / 60;\r
+ let hours = mins / 60;\r
+ let days = hours / 24;\r
+ format!("{}d {}h {}min", days, hours - 24 * days, mins - 60 * hours)\r
+ }\r
+\r
+ pub fn format_world_size(&self) -> String {\r
+ format_byte_size(self.world_size, 2)\r
+ }\r
+\r
+ pub fn format_active_players(&self) -> String {\r
+ if self.active_players.len() == 0 {\r
+ String::from("<none>")\r
+ } else {\r
+ self.active_players.join(", ")\r
+ }\r
+ }\r
+\r
+ pub fn format_last_backup(&self) -> String {\r
+ match self.last_backup {\r
+ Some(t) => {\r
+ let datetime: DateTime<Local> = t.into();\r
+ datetime.format("%d/%m/%Y %T").to_string()\r
+ },\r
+ None => String::from("?")\r
+ }\r
+ }\r
+}\r
+\r
+const BINARY_PREFIXES: [&str; 8] = ["B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB"];\r
+\r
+fn format_byte_size(bytes: u64, precision: usize) -> String {\r
+ for i in 0 .. 8 {\r
+ let mut size: u64 = 1;\r
+ size *= 1024u64.pow(i as u32);\r
+\r
+ if bytes < 1024 {\r
+ return format!("{} {}", std::cmp::max(0u64, bytes), BINARY_PREFIXES[i]);\r
+ }\r
+ else if bytes < 1024 * size {\r
+ return format!("{:.prec$} {}", bytes as f64 / size as f64, BINARY_PREFIXES[i], prec = precision);\r
+ }\r
+ }\r
+\r
+ String::from("")\r
+}\r
+\r
+const VALHEIM_PROCESS_NAME: &str = "valheim_server";\r
+const STRING_BEFORE_CHARACTER_NAME: &str = "Got character ZDOID from";\r
+const STRING_BEFORE_NB_OF_CONNECTIONS: &str = "Connections";\r
+\r
+#[cfg(target_os = "linux")]\r
+fn get_active_players() -> Vec<String> {\r
+ let mut journal =\r
+ journal::OpenOptions::default().current_user(true).open().unwrap();\r
+\r
+ journal.seek_tail().unwrap();\r
+\r
+ let mut number_of_connections = 0;\r
+ let mut players : Vec<String> = Vec::new();\r
+\r
+ loop {\r
+ match journal.previous_entry() {\r
+ Ok(Some(entry)) => {\r
+ if let (Some(unit), Some(mess)) = (entry.get("_SYSTEMD_UNIT"), entry.get("MESSAGE")) {\r
+ if unit == "valheim.service" {\r
+ if let Some(pos) = mess.find(STRING_BEFORE_CHARACTER_NAME) {\r
+ let character_str = mess.get(pos+STRING_BEFORE_CHARACTER_NAME.len()+1..).unwrap();\r
+ if let Some(pos_end) = character_str.find(" : ") {\r
+ let player_name = String::from(character_str.get(0..pos_end).unwrap());\r
+ if !players.contains(&player_name) {\r
+ players.push(player_name);\r
+ if players.len() == number_of_connections {\r
+ return players;\r
+ }\r
+ }\r
+ }\r
+ }\r
+ else if let Some(pos) = mess.find(STRING_BEFORE_NB_OF_CONNECTIONS) {\r
+ let nb_of_connections_str = mess.get(pos+STRING_BEFORE_NB_OF_CONNECTIONS.len()+1..).unwrap();\r
+ if let Some(pos_end) = nb_of_connections_str.find(' ') {\r
+ if let Ok(n) = nb_of_connections_str.get(0..pos_end).unwrap().parse() {\r
+ number_of_connections = n;\r
+ if players.len() >= number_of_connections {\r
+ return players;\r
+ }\r
+ }\r
+ }\r
+ }\r
+ }\r
+ }\r
+ },\r
+ _ => return players\r
+ }\r
+ }\r
+}\r
+\r
+#[cfg(target_os = "windows")]\r
+fn get_active_players() -> Vec<String> {\r
+ Vec::new()\r
+}\r
+\r
+fn get_last_backup_datetime(backup_path: &str) -> Option<SystemTime> {\r
+ let mut times =\r
+ fs::read_dir(backup_path).ok()?.filter_map(\r
+ |e| {\r
+ let dir = e.ok()?;\r
+ if dir.path().is_file() { Some(dir.metadata().ok()?.modified().ok()?) } else { None }\r
+ }\r
+ )\r
+ .collect::<Vec<SystemTime>>();\r
+\r
+ times.sort();\r
+\r
+ Some(times.last()?.clone())\r
+}\r
+\r
+pub fn get_valheim_executable_information(world_path: &str, backup_path: &str) -> Option<ValheimExe> {\r
let mut system = sysinfo::System::new_all();\r
- system.refresh_all();\r
- let process = system.get_process_by_name("valheim_server");\r
+ system.refresh_system();\r
+ let processes = system.get_process_by_name(VALHEIM_PROCESS_NAME);\r
+\r
+ if processes.len() >= 1 {\r
+ let process = processes.first().unwrap();\r
+\r
+ let world_size = match std::fs::metadata(world_path) { Ok(f) => f.len(), Err(_) => 0u64 };\r
\r
- if process.len() >= 1 {\r
- println!("Valheim running");\r
- true\r
+ Some(\r
+ ValheimExe {\r
+ memory: process.memory(),\r
+ load_average_5min: system.get_load_average().five / system.get_processors().len() as f64 * 100.,\r
+ uptime: std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH).unwrap().as_secs() - process.start_time(),\r
+ world_size,\r
+ active_players: get_active_players(),\r
+ last_backup: get_last_backup_datetime(backup_path)\r
+ }\r
+ )\r
} else {\r
- println!("Valheim not running");\r
- false\r
+ None\r
}\r
-}
\ No newline at end of file
+}\r