Show resource usage

Author: ranfdev

src/container.rs

@@ -1,4 +1,5 @@
 use crate::{
+    container_stats::Usage,
     distrobox::{ContainerInfo, CreateArgName, CreateArgs, ExportableApp, Status},
     distrobox_task::DistroboxTask,
     fakers::Command,
@@ -40,6 +41,7 @@ mod imp {
         pub distro: RefCell<Option<KnownDistro>>,
         pub apps: Query<TypedListStore<glib::BoxedAnyObject>>,
         pub binaries: Query<TypedListStore<glib::BoxedAnyObject>>,
+        pub usage: Query<Usage>,
     }
 
     impl Default for Container {
@@ -72,6 +74,7 @@ mod imp {
                             None
                         }
                     }),
+                usage: Query::new("usage".into(), || async { Ok(Usage::default()) }),
             }
         }
     }
@@ -138,6 +141,35 @@ impl Container {
             }
         });
 
+        let this_clone = this.clone();
+        this.usage().set_fetcher(move || {
+            let this = this_clone.clone();
+            async move {
+                let root_store = this.root_store();
+                let runtime = root_store.get_container_runtime().await?;
+                let runner = root_store.imp().command_runner.get().unwrap();
+
+                let mut cmd = Command::new(runtime.as_str());
+                cmd.arg("stats");
+                cmd.arg("--no-stream");
+                cmd.arg("--format");
+                cmd.arg("json");
+                cmd.arg(this.name());
+                cmd.stdout = crate::fakers::FdMode::Pipe;
+                cmd.stderr = crate::fakers::FdMode::Pipe;
+
+                let output = runner.output(cmd).await?;
+                if !output.status.success() {
+                    return Err(anyhow::anyhow!("Failed to get stats"));
+                }
+
+                let stdout = String::from_utf8(output.stdout)?;
+                let usages: Vec<Usage> = serde_json::from_str(&stdout)?;
+
+                usages.into_iter().next().ok_or_else(|| anyhow::anyhow!("No stats found"))
+            }
+        });
+
         this
     }
 
@@ -170,6 +202,10 @@ impl Container {
         self.imp().binaries.clone()
     }
 
+    pub fn usage(&self) -> Query<Usage> {
+        self.imp().usage.clone()
+    }
+
     pub fn upgrade(&self) -> DistroboxTask {
         let this = self.clone();
         self.root_store()

src/container_stats.rs

@@ -0,0 +1,32 @@
+use serde::Deserialize;
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum ContainerRuntime {
+    Podman,
+    Docker,
+}
+
+impl ContainerRuntime {
+    pub fn as_str(&self) -> &'static str {
+        match self {
+            ContainerRuntime::Podman => "podman",
+            ContainerRuntime::Docker => "docker",
+        }
+    }
+}
+
+#[derive(Debug, Clone, Deserialize, Default)]
+pub struct Usage {
+    #[serde(rename = "mem_usage", alias = "MemUsage")]
+    pub mem_usage: String,
+    #[serde(rename = "mem_percent", alias = "MemPerc")]
+    pub mem_perc: String,
+    #[serde(rename = "cpu_percent", alias = "CPU")]
+    pub cpu_perc: String,
+    #[serde(rename = "net_io", alias = "NetIO")]
+    pub net_io: String,
+    #[serde(rename = "block_io", alias = "BlockIO")]
+    pub block_io: String,
+    #[serde(rename = "pids", alias = "PIDs")]
+    pub pids: String,
+}

src/main.rs

@@ -21,6 +21,7 @@
 mod application;
 mod config;
 mod container;
+mod container_stats;
 mod dialogs;
 mod distro_icon;
 mod distrobox;

src/store/root_store.rs

@@ -16,6 +16,7 @@ use tracing::info;
 use tracing::{debug, warn};
 
 use crate::container::Container;
+use crate::container_stats::ContainerRuntime;
 use crate::distrobox;
 use crate::distrobox::CreateArgs;
 use crate::distrobox::Distrobox;
@@ -126,6 +127,7 @@ mod imp {
         pub distrobox: OnceCell<crate::distrobox::Distrobox>,
         pub terminal_repository: RefCell<TerminalRepository>,
         pub command_runner: OnceCell<CommandRunner>,
+        pub container_runtime: RefCell<Option<ContainerRuntime>>,
 
         pub distrobox_version: Query<String>,
         pub images_query: Query<Vec<String>>,
@@ -152,6 +154,7 @@ mod imp {
             Self {
                 containers: TypedListStore::new(),
                 command_runner: OnceCell::new(),
+                container_runtime: RefCell::new(None),
                 terminal_repository: RefCell::new(TerminalRepository::new(
                     CommandRunner::new_null(),
                 )),
@@ -668,6 +671,38 @@ impl RootStore {
             }
         }
     }
+
+    pub async fn get_container_runtime(&self) -> anyhow::Result<ContainerRuntime> {
+        if let Some(runtime) = *self.imp().container_runtime.borrow() {
+            return Ok(runtime);
+        }
+
+        let runner = self.imp().command_runner.get().unwrap();
+        let mut cmd = Command::new("podman");
+        cmd.arg("--version");
+        cmd.stdout = FdMode::Pipe;
+        cmd.stderr = FdMode::Pipe;
+
+        let podman_check = runner.output(cmd).await;
+
+        let runtime = if podman_check.is_ok() && podman_check.unwrap().status.success() {
+            ContainerRuntime::Podman
+        } else {
+            let mut cmd = Command::new("docker");
+            cmd.arg("--version");
+            cmd.stdout = FdMode::Pipe;
+            cmd.stderr = FdMode::Pipe;
+            let docker_check = runner.output(cmd).await;
+            if docker_check.is_ok() && docker_check.unwrap().status.success() {
+                ContainerRuntime::Docker
+            } else {
+                ContainerRuntime::Podman
+            }
+        };
+
+        *self.imp().container_runtime.borrow_mut() = Some(runtime);
+        Ok(runtime)
+    }
 }
 
 impl Default for RootStore {

src/window.rs

@@ -459,12 +459,30 @@ impl DistroShelfWindow {
         status_row.add_suffix(&status_child);
         status_group.add(&status_row);
 
+        // Usage stats row
+        let usage_row = adw::ActionRow::new();
+        usage_row.set_title("Resources");
+        usage_row.set_subtitle(&format!("CPU: 0.0% • Mem: 0 (0%)"));
+        status_group.add(&usage_row);
+
+
+        let usage_query = container.usage();
+        usage_query.connect_success(clone!(
+            #[weak] usage_row,
+            move |usage| {
+                usage_row.set_subtitle(&format!("CPU: {} • Mem: {} ({})", usage.cpu_perc, usage.mem_usage, usage.mem_perc));
+            }
+        ));
+
         reaction! {
             (container.status_detail(), container.status_tag()),
             move |(detail, tag): (String, String)| {
                 let text = format!("{tag}: {detail}");
                 status_row.set_subtitle(&text);
                 stop_btn.set_visible(tag == "up");
+                if tag == "up" {
+                    usage_query.fetch();
+                }
             }
         };