1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
use std::path::PathBuf;
/// Format path str
pub fn format_path_str(path: impl Into<String>) -> Result<String, std::io::Error> {
let path_str = path.into();
// ANSI Strip
let cleaned = strip_ansi_escapes::strip(&path_str);
let path_without_ansi = String::from_utf8(cleaned)
.map_err(|e| std::io::Error::new(std::io::ErrorKind::InvalidData, e))?;
let path_with_forward_slash = path_without_ansi.replace('\\', "/");
let mut result = String::new();
let mut prev_char = '\0';
for c in path_with_forward_slash.chars() {
if c == '/' && prev_char == '/' {
continue;
}
result.push(c);
prev_char = c;
}
let unfriendly_chars = ['*', '?', '"', '<', '>', '|'];
result = result
.chars()
.filter(|c| !unfriendly_chars.contains(c))
.collect();
if result.ends_with('/') {
Ok(result)
} else {
Ok(result)
}
}
pub fn format_path(path: impl Into<PathBuf>) -> Result<PathBuf, std::io::Error> {
let path_str = format_path_str(path.into().display().to_string())?;
Ok(PathBuf::from(path_str))
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_format_path() -> Result<(), std::io::Error> {
assert_eq!(format_path_str("C:\\Users\\\\test")?, "C:/Users/test");
assert_eq!(
format_path_str("/path/with/*unfriendly?chars")?,
"/path/with/unfriendlychars"
);
assert_eq!(format_path_str("\x1b[31m/path\x1b[0m")?, "/path");
assert_eq!(format_path_str("/home/user/dir/")?, "/home/user/dir/");
assert_eq!(
format_path_str("/home/user/file.txt")?,
"/home/user/file.txt"
);
Ok(())
}
}
|
