'unix' -> 'linux'

[valheim_web.git] / backend / src / valheim_controller.rs

diff --git a/backend/src/valheim_controller.rs b/backend/src/valheim_controller.rs
index c09f080..79b92b3 100644 (file)
--- a/backend/src/valheim_controller.rs
+++ b/backend/src/valheim_controller.rs
@@ -1,29 +1,55 @@
-use sysinfo::{ProcessExt, SystemExt};\r
+use sysinfo::{ ProcessExt, SystemExt };\r
 \r
-#[derive(Debug)]\r
+use std::{ fs, time::SystemTime };\r
+\r
+use chrono::{ DateTime, offset::Local };\r
+\r
+#[cfg(target_os = "unix")]\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
+    nb_of_players: u32,\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 get_nb_of_player(&self) -> u32 {\r
+        self.nb_of_players\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
@@ -45,9 +71,56 @@ fn format_byte_size(bytes: u64, precision: usize) -> String {
 }\r
 \r
 const VALHEIM_PROCESS_NAME: &str = "valheim_server";\r
-const VALHEIM_WORLD_PATH: &str = "/home/greg/ValheimWorld/pouet.db"; // TODO: Put in conf.\r
 \r
-pub fn get_valheim_executable_information() -> Option<ValheimExe> {\r
+#[cfg(target_os = "linux")]\r
+fn get_number_of_players() -> u32 {\r
+    let mut journal =\r
+        journal::OpenOptions::default().current_user(true).open().unwrap();\r
+\r
+    journal.seek_tail().unwrap();\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("Connections") {\r
+                            let nb_of_connections_str = mess.get(pos+12..).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
+                                    return n;\r
+                                }\r
+                            }\r
+                        }\r
+                    }\r
+                }\r
+            },\r
+            _ => return 0\r
+        }\r
+    }\r
+}\r
+\r
+#[cfg(target_os = "windows")]\r
+fn get_number_of_players() -> u32 {\r
+    0\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()?.created().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_system();\r
     let processes = system.get_process_by_name(VALHEIM_PROCESS_NAME);\r
@@ -55,14 +128,16 @@ pub fn get_valheim_executable_information() -> Option<ValheimExe> {
     if processes.len() >= 1 {\r
         let process = processes.first().unwrap();\r
 \r
-        let world_size = match std::fs::metadata(VALHEIM_WORLD_PATH) { Ok(f) => f.len(), Err(_) => 0u64 };\r
+        let world_size = match std::fs::metadata(world_path) { Ok(f) => f.len(), Err(_) => 0u64 };\r
 \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
+                world_size,\r
+                nb_of_players: get_number_of_players(),\r
+                last_backup: get_last_backup_datetime(backup_path)\r
             }\r
         )\r
     } else {\r