From b9e1964bdf75c2e06786a79426d41f0516de68b4 Mon Sep 17 00:00:00 2001 From: 魏曹先生 <1992414357@qq.com> Date: Mon, 10 Nov 2025 09:24:49 +0800 Subject: Create fs.rs --- src/utils/display.rs | 26 ++++++++++++++++++++++++++ src/utils/fs.rs | 40 ++++++++++++++++++++++++++++++++++++++++ 2 files changed, 66 insertions(+) create mode 100644 src/utils/fs.rs (limited to 'src/utils') diff --git a/src/utils/display.rs b/src/utils/display.rs index 533d46d..09efe97 100644 --- a/src/utils/display.rs +++ b/src/utils/display.rs @@ -1,3 +1,6 @@ +use colored::*; +use regex::Regex; + pub struct SimpleTable { items: Vec, line: Vec>, @@ -118,6 +121,10 @@ fn display_width(s: &str) -> usize { width } +/// Convert byte size to a human-readable string format +/// +/// Automatically selects the appropriate unit (B, KB, MB, GB, TB) based on the byte size +/// and formats it as a string with two decimal places pub fn size_str(total_size: usize) -> String { if total_size < 1024 { format!("{} B", total_size) @@ -134,3 +141,22 @@ pub fn size_str(total_size: usize) -> String { ) } } + +// Convert the Markdown formatted text into a format supported by the command line +pub fn md(text: impl AsRef) -> String { + let bold_re = Regex::new(r"\*\*(.*?)\*\*").unwrap(); + let mut result = bold_re + .replace_all(text.as_ref().trim(), |caps: ®ex::Captures| { + format!("{}", caps[1].bold()) + }) + .to_string(); + + let italic_re = Regex::new(r"\*(.*?)\*").unwrap(); + result = italic_re + .replace_all(&result, |caps: ®ex::Captures| { + format!("{}", caps[1].italic()) + }) + .to_string(); + + result +} diff --git a/src/utils/fs.rs b/src/utils/fs.rs new file mode 100644 index 0000000..0050cf1 --- /dev/null +++ b/src/utils/fs.rs @@ -0,0 +1,40 @@ +pub async fn move_across_partitions( + source_path: impl AsRef, + dest_path: impl AsRef, +) -> Result<(), std::io::Error> { + let source_path = source_path.as_ref(); + let dest_path = dest_path.as_ref(); + if !source_path.exists() { + return Err(std::io::Error::new( + std::io::ErrorKind::NotFound, + "Source file does not exist", + )); + } + + if let Ok(()) = std::fs::rename(source_path, dest_path) { + return Ok(()); + } + + std::fs::copy(source_path, dest_path)?; + std::fs::remove_file(source_path)?; + + Ok(()) +} + +pub async fn copy_across_partitions( + source_path: impl AsRef, + dest_path: impl AsRef, +) -> Result<(), std::io::Error> { + let source_path = source_path.as_ref(); + let dest_path = dest_path.as_ref(); + if !source_path.exists() { + return Err(std::io::Error::new( + std::io::ErrorKind::NotFound, + "Source file does not exist", + )); + } + + std::fs::copy(source_path, dest_path)?; + + Ok(()) +} -- cgit