monitor: update screenshots for profile guests (#6)

Author: delan

monitor/src/data.rs

@@ -26,6 +26,12 @@ pub fn get_runner_data_path(id: usize, path: impl AsRef<Path>) -> eyre::Result<P
     Ok(runner_data.join(path))
 }
 
+pub fn get_profile_data_path(key: &str, path: impl AsRef<Path>) -> eyre::Result<PathBuf> {
+    let profile_data = get_data_path("profiles")?.join(key);
+
+    Ok(profile_data.join(path))
+}
+
 #[tracing::instrument]
 pub fn run_migrations() -> eyre::Result<()> {
     let migrations_dir = get_data_path("migrations")?;

monitor/src/libvirt.rs

@@ -1,6 +1,6 @@
 use core::str;
 use std::{
-    fs::rename,
+    fs::{create_dir_all, rename},
     path::Path,
     process::{Command, Stdio},
 };
@@ -36,6 +36,7 @@ pub fn libvirt_prefix() -> String {
 }
 
 pub fn update_screenshot(guest_name: &str, output_dir: &Path) -> Result<(), eyre::Error> {
+    create_dir_all(output_dir)?;
     let new_path = output_dir.join("screenshot.png.new");
     let exit_status = SHELL
         .lock()

monitor/src/main.rs

@@ -468,6 +468,9 @@ fn monitor_thread() -> eyre::Result<()> {
         if let Ok(mut dashboard) = DASHBOARD.write() {
             *dashboard = Some(Dashboard::render(&profile_runner_counts, &runners)?);
             runners.update_screenshots();
+            for (_key, profile) in TOML.profiles() {
+                profile.update_screenshot();
+            }
         }
 
         // Handle one request from the API.

monitor/src/profile.rs

@@ -2,9 +2,11 @@ use std::process::Command;
 
 use jane_eyre::eyre;
 use serde::{Deserialize, Serialize};
-use tracing::info;
+use tracing::{debug, info};
 
 use crate::{
+    data::get_profile_data_path,
+    libvirt::update_screenshot,
     runner::{Runner, Runners, Status},
     DOTENV,
 };
@@ -163,4 +165,21 @@ impl Profile {
             0
         }
     }
+
+    pub fn update_screenshot(&self) {
+        if let Err(error) = self.try_update_screenshot() {
+            debug!(
+                self.base_vm_name,
+                ?error,
+                "Failed to update screenshot for profile guest"
+            );
+        }
+    }
+
+    fn try_update_screenshot(&self) -> eyre::Result<()> {
+        let output_dir = get_profile_data_path(&self.base_vm_name, ".")?;
+        update_screenshot(&self.base_vm_name, &output_dir)?;
+
+        Ok(())
+    }
 }