From 0f5430ab9eb7049c26ee65e27f14812f91a232d5 Mon Sep 17 00:00:00 2001 From: mo8it Date: Fri, 1 Mar 2024 00:55:00 +0100 Subject: Optimize get_truncated_path --- helix-stdx/src/path.rs | 23 +++++++++++------------ 1 file changed, 11 insertions(+), 12 deletions(-) diff --git a/helix-stdx/src/path.rs b/helix-stdx/src/path.rs index c514f9f2..ff2bffae 100644 --- a/helix-stdx/src/path.rs +++ b/helix-stdx/src/path.rs @@ -181,21 +181,20 @@ where /// pub fn get_truncated_path(path: impl AsRef) -> PathBuf { let cwd = current_working_dir(); - let path = path - .as_ref() - .strip_prefix(cwd) - .unwrap_or_else(|_| path.as_ref()); + let path = path.as_ref(); + let path = path.strip_prefix(cwd).unwrap_or(path); let file = path.file_name().unwrap_or_default(); let base = path.parent().unwrap_or_else(|| Path::new("")); - let mut ret = PathBuf::new(); + let mut ret = PathBuf::with_capacity(file.len()); + // A char can't be directly pushed to a PathBuf + let mut first_char_buffer = String::new(); for d in base { - ret.push( - d.to_string_lossy() - .chars() - .next() - .unwrap_or_default() - .to_string(), - ); + let Some(first_char) = d.to_string_lossy().chars().next() else { + break; + }; + first_char_buffer.push(first_char); + ret.push(&first_char_buffer); + first_char_buffer.clear(); } ret.push(file); ret -- cgit v1.2.3-70-g09d2