From f229e3044048d57bddf7f14d7ff3d8d296039018 Mon Sep 17 00:00:00 2001
From: 魏曹先生 <1992414357@qq.com>
Date: Thu, 6 Nov 2025 22:06:30 +0800
Subject: feat: add display utilities and table component

- Create display module with SimpleTable structure
- Add size_str function for file size formatting
- Support ANSI escape sequence filtering in display width
- Provide table formatting with proper column alignment
---
 src/utils/display.rs | 136 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 136 insertions(+)
 create mode 100644 src/utils/display.rs

(limited to 'src/utils/display.rs')

diff --git a/src/utils/display.rs b/src/utils/display.rs
new file mode 100644
index 0000000..533d46d
--- /dev/null
+++ b/src/utils/display.rs
@@ -0,0 +1,136 @@
+pub struct SimpleTable {
+    items: Vec<String>,
+    line: Vec<Vec<String>>,
+    length: Vec<usize>,
+    padding: usize,
+}
+
+impl SimpleTable {
+    /// Create a new Table
+    pub fn new(items: Vec<impl Into<String>>) -> Self {
+        Self::new_with_padding(items, 2)
+    }
+
+    /// Create a new Table with padding
+    pub fn new_with_padding(items: Vec<impl Into<String>>, padding: usize) -> Self {
+        let items: Vec<String> = items.into_iter().map(|v| v.into()).collect();
+        let mut length = Vec::with_capacity(items.len());
+
+        for item in &items {
+            length.push(display_width(item));
+        }
+
+        SimpleTable {
+            items,
+            padding,
+            line: Vec::new(),
+            length,
+        }
+    }
+
+    /// Push a new row of items to the table
+    pub fn push_item(&mut self, items: Vec<impl Into<String>>) {
+        let items: Vec<String> = items.into_iter().map(|v| v.into()).collect();
+
+        let mut processed_items = Vec::with_capacity(self.items.len());
+
+        for i in 0..self.items.len() {
+            if i < items.len() {
+                processed_items.push(items[i].clone());
+            } else {
+                processed_items.push(String::new());
+            }
+        }
+
+        for (i, d) in processed_items.iter().enumerate() {
+            let d_len = display_width(d);
+            if d_len > self.length[i] {
+                self.length[i] = d_len;
+            }
+        }
+
+        self.line.push(processed_items);
+    }
+
+    /// Get the current maximum column widths
+    fn get_column_widths(&self) -> &[usize] {
+        &self.length
+    }
+}
+
+impl std::fmt::Display for SimpleTable {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        let column_widths = self.get_column_widths();
+
+        // Build the header row
+        let header: Vec<String> = self
+            .items
+            .iter()
+            .enumerate()
+            .map(|(i, item)| {
+                let target_width = column_widths[i] + self.padding;
+                let current_width = display_width(item);
+                let space_count = target_width - current_width;
+                let space = " ".repeat(space_count);
+                let result = format!("{}{}", item, space);
+                result
+            })
+            .collect();
+        writeln!(f, "{}", header.join(""))?;
+
+        // Build each data row
+        for row in &self.line {
+            let formatted_row: Vec<String> = row
+                .iter()
+                .enumerate()
+                .map(|(i, cell)| {
+                    let target_width = column_widths[i] + self.padding;
+                    let current_width = display_width(cell);
+                    let space_count = target_width - current_width;
+                    let spaces = " ".repeat(space_count);
+                    let result = format!("{}{}", cell, spaces);
+                    result
+                })
+                .collect();
+            writeln!(f, "{}", formatted_row.join(""))?;
+        }
+
+        Ok(())
+    }
+}
+
+fn display_width(s: &str) -> usize {
+    // Filter out ANSI escape sequences before calculating width
+    let filtered_bytes = strip_ansi_escapes::strip(s);
+    let filtered_str = match std::str::from_utf8(&filtered_bytes) {
+        Ok(s) => s,
+        Err(_) => s, // Fallback to original string if UTF-8 conversion fails
+    };
+
+    let mut width = 0;
+    for c in filtered_str.chars() {
+        if c.is_ascii() {
+            width += 1;
+        } else {
+            width += 2;
+        }
+    }
+    width
+}
+
+pub fn size_str(total_size: usize) -> String {
+    if total_size < 1024 {
+        format!("{} B", total_size)
+    } else if total_size < 1024 * 1024 {
+        format!("{:.2} KB", total_size as f64 / 1024.0)
+    } else if total_size < 1024 * 1024 * 1024 {
+        format!("{:.2} MB", total_size as f64 / (1024.0 * 1024.0))
+    } else if total_size < 1024 * 1024 * 1024 * 1024 {
+        format!("{:.2} GB", total_size as f64 / (1024.0 * 1024.0 * 1024.0))
+    } else {
+        format!(
+            "{:.2} TB",
+            total_size as f64 / (1024.0 * 1024.0 * 1024.0 * 1024.0)
+        )
+    }
+}
-- 
cgit