From 402f37bae40ae39c20a3e9fabfe9bf016042fae2 Mon Sep 17 00:00:00 2001
From: robcholz <84130577+robcholz@users.noreply.github.com>
Date: Sat, 7 Feb 2026 17:36:07 -0500
Subject: [PATCH] feat: added vm info

---
 src/bin/vibebox-cli.rs |  5 +++--
 src/tui.rs             | 27 +++++++++++++++++++++------
 2 files changed, 24 insertions(+), 8 deletions(-)

diff --git a/src/bin/vibebox-cli.rs b/src/bin/vibebox-cli.rs
index 9dff02d..24d97c6 100644
--- a/src/bin/vibebox-cli.rs
+++ b/src/bin/vibebox-cli.rs
@@ -79,12 +79,13 @@ fn main() -> Result<()> {
         mounts: config.box_cfg.mounts.clone(),
     };
 
+    let auto_shutdown_ms = config.supervisor.auto_shutdown_ms;
     let vm_info = VmInfo {
-        version: env!("CARGO_PKG_VERSION").to_string(),
         max_memory_mb: vm_args.ram_bytes / (1024 * 1024),
         cpu_cores: vm_args.cpu_count,
+        system_name: "Debian".to_string(), // TODO: read system name from the VM.
+        auto_shutdown_ms,
     };
-    let auto_shutdown_ms = config.supervisor.auto_shutdown_ms;
     if let Ok(manager) = SessionManager::new() {
         if let Err(err) = manager.update_global_sessions(&cwd) {
             tracing::warn!(error = %err, "failed to update a global session list");
diff --git a/src/tui.rs b/src/tui.rs
index 1a4371c..fcd231e 100644
--- a/src/tui.rs
+++ b/src/tui.rs
@@ -34,12 +34,14 @@ const ASCII_BANNER: [&str; 7] = [
     "  ╚═══╝  ╚═╝╚═════╝ ╚══════╝╚═════╝  ╚═════╝ ╚═╝  ╚═╝",
     "",
 ];
+const INFO_LINE_COUNT: u16 = 5;
 
 #[derive(Debug, Clone)]
 pub struct VmInfo {
-    pub version: String,
     pub max_memory_mb: u64,
     pub cpu_cores: usize,
+    pub system_name: String,
+    pub auto_shutdown_ms: u64,
 }
 
 #[derive(Debug)]
@@ -155,10 +157,14 @@ fn compute_page_layout(app: &AppState, width: u16) -> PageLayout {
 fn header_height() -> u16 {
     let banner_height = ASCII_BANNER.len() as u16;
     let welcome_height = 1;
-    let info_height = 4;
+    let info_height = info_block_height();
     welcome_height + banner_height + info_height
 }
 
+fn info_block_height() -> u16 {
+    INFO_LINE_COUNT.saturating_add(1)
+}
+
 pub fn render_tui_once(app: &mut AppState) -> Result<()> {
     let (width, _) = crossterm::terminal::size()?;
     if width == 0 {
@@ -532,13 +538,15 @@ fn render_header(buffer: &mut Buffer, area: Rect, app: &AppState) {
         .constraints([
             Constraint::Length(1),
             Constraint::Length(ASCII_BANNER.len() as u16),
-            Constraint::Length(4),
+            Constraint::Length(info_block_height()),
         ])
         .split(area);
 
+    let version = env!("CARGO_PKG_VERSION");
+
     let welcome = Line::from(vec![
         Span::raw("Welcome to Vibebox v"),
-        Span::styled(&app.vm_info.version, Style::default().fg(Color::Yellow)),
+        Span::styled(version, Style::default().fg(Color::Yellow)),
     ]);
 
     Paragraph::new(welcome).render(header_chunks[0], buffer);
@@ -557,8 +565,8 @@ fn render_header(buffer: &mut Buffer, area: Rect, app: &AppState) {
             Span::styled(app.cwd.to_string_lossy(), Style::default().fg(Color::Cyan)),
         ]),
         Line::from(vec![
-            Span::raw("VM Version: "),
-            Span::styled(&app.vm_info.version, Style::default().fg(Color::Green)),
+            Span::raw("System: "),
+            Span::styled(&app.vm_info.system_name, Style::default().fg(Color::Green)),
         ]),
         Line::from(vec![
             Span::raw("CPU / Memory: "),
@@ -570,6 +578,13 @@ fn render_header(buffer: &mut Buffer, area: Rect, app: &AppState) {
                 Style::default().fg(Color::Green),
             ),
         ]),
+        Line::from(vec![
+            Span::raw("Auto Shutdown: "),
+            Span::styled(
+                format!("{} ms", app.vm_info.auto_shutdown_ms),
+                Style::default().fg(Color::Green),
+            ),
+        ]),
     ];
 
     Paragraph::new(info_lines)