X-Git-Url: http://git.euphorik.ch/index.cgi?a=blobdiff_plain;f=backend%2Fsrc%2Fvalheim_controller.rs;h=8c04e238f21e022a84810ee11640013ca724907c;hb=150411e0cffb5ad0127d822c39a38c2e92b2bd99;hp=3fd4970a37886413cc8ecbd7cf8550be4465e0c4;hpb=aa84eb0853e2207014062877d539fca150a6d9c7;p=valheim_web.git

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