backend/src/minecraft_controller.rs

   1 use sysinfo::{ ProcessExt, SystemExt };
   2
   3 use std::{ fs, time::SystemTime };
   4
   5 use chrono::{ DateTime, offset::Local };
   6
   7 #[derive(Clone, Debug)]
   8 pub struct MinecraftExe {
   9     memory: u64, // [kB].
  10     load_average_5min: f64, // [%].
  11     uptime: u64, // [s].
  12     world_size: u64, // [B].
  13     active_players: Vec<String>,
  14     last_backup: Option<SystemTime>,
  15 }
  16
  17 impl MinecraftExe {
  18     pub fn format_memory(&self) -> String {
  19         format_byte_size(self.memory * 1024, 2)
  20     }
  21
  22     pub fn format_load_average(&self) -> String {
  23         format!("{:.2} %", self.load_average_5min)
  24     }
  25
  26     pub fn format_uptime(&self) -> String {
  27         let mins = self.uptime / 60;
  28         let hours = mins / 60;
  29         let days = hours / 24;
  30         format!("{}d {}h {}min", days, hours - 24 * days, mins - 60 * hours)
  31     }
  32
  33     pub fn format_world_size(&self) -> String {
  34         format_byte_size(self.world_size, 2)
  35     }
  36
  37     pub fn format_active_players(&self) -> String {
  38         if self.active_players.len() == 0 {
  39             String::from("<none>")
  40         } else {
  41             self.active_players.join(", ")
  42         }
  43     }
  44
  45     pub fn format_last_backup(&self) -> String {
  46         match self.last_backup {
  47             Some(t) => {
  48                 let datetime: DateTime<Local> = t.into();
  49                 datetime.format("%d/%m/%Y %T").to_string()
  50             },
  51             None => String::from("?")
  52         }
  53     }
  54 }
  55
  56 const BINARY_PREFIXES: [&str; 8] = ["B", "KiB", "MiB", "GiB", "TiB", "PiB", "EiB", "ZiB"];
  57
  58 fn format_byte_size(bytes: u64, precision: usize) -> String {
  59     for i in 0 .. 8 {
  60         let mut size: u64 = 1;
  61         size *= 1024u64.pow(i as u32);
  62
  63         if bytes < 1024 {
  64             return format!("{} {}", std::cmp::max(0u64, bytes), BINARY_PREFIXES[i]);
  65         }
  66         else if bytes < 1024 * size {
  67             return format!("{:.prec$} {}", bytes as f64 / size as f64, BINARY_PREFIXES[i], prec = precision);
  68         }
  69     }
  70
  71     String::from("")
  72 }
  73
  74 const MINECRAFT_PROCESS_NAME: &str = "java";
  75
  76 struct StatusFromRcon {
  77     players: Vec<String>,
  78 }
  79
  80 fn get_status_from_rcon(rcon_password: &str) -> StatusFromRcon {
  81     let mut client = minecraft_client_rs::Client::new("127.0.0.1:25575".to_string()).unwrap();
  82
  83     let status =
  84         match client.authenticate(rcon_password.to_string()) {
  85             Ok(_) => {
  86                 StatusFromRcon {
  87                     players:
  88                         match client.send_command("list".to_string()) {
  89                             Ok(resp) =>
  90                                 match resp.body.find(':') {
  91                                     Some(i) if i < resp.body.len() -1 =>
  92                                         resp.body[i + 1..resp.body.len()]
  93                                             .split(',')
  94                                             .map(|nick| nick.trim().to_string())
  95                                             .filter(|nick| !nick.is_empty())
  96                                             .collect(),
  97                                     _ => Vec::new()
  98                                 },
  99                             Err(_e) => {
 100                                 println!("Error from 'list' command");
 101                                 Vec::new()
 102                             },
 103                         }
 104                 }
 105             },
 106             Err(_e) => {
 107                 println!("Authentication error");
 108                 StatusFromRcon { players: Vec::new() }
 109             },
 110         };
 111
 112     client.close().unwrap();
 113
 114     status
 115 }
 116
 117 fn get_last_backup_datetime(backup_path: &str) -> Option<SystemTime> {
 118     let mut times =
 119         fs::read_dir(backup_path).ok()?.filter_map(
 120             |e| {
 121                 let dir = e.ok()?;
 122                 if dir.path().is_file() { Some(dir.metadata().ok()?.modified().ok()?) } else { None }
 123             }
 124         )
 125         .collect::<Vec<SystemTime>>();
 126
 127     times.sort();
 128
 129     Some(times.last()?.clone())
 130 }
 131
 132 pub fn get_minecraft_executable_information(world_path: &str, backup_path: &str, rcon_password: &str) -> Option<MinecraftExe> {
 133     let mut system = sysinfo::System::new_all();
 134     system.refresh_system();
 135     let mut processes = system.processes_by_name(MINECRAFT_PROCESS_NAME);
 136
 137     // TODO: find the correct process by checking the correct jar name in parameters.
 138     if let Some(process) = processes.next() {
 139         let world_size = match fs_extra::dir::get_size(world_path) { Ok(l) => l, Err(_) => 0u64 };
 140
 141         let status_from_rcon = get_status_from_rcon(rcon_password);
 142
 143         Some(
 144             MinecraftExe {
 145                 memory: process.memory(),
 146                 load_average_5min: system.load_average().five / system.processors().len() as f64 * 100.,
 147                 uptime: std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH).unwrap().as_secs() - process.start_time(),
 148                 world_size,
 149                 active_players: status_from_rcon.players,
 150                 last_backup: get_last_backup_datetime(backup_path)
 151             }
 152         )
 153     } else {
 154         None
 155     }
 156 }