62 lines
2.4 KiB
Rust
62 lines
2.4 KiB
Rust
macro_rules! regex {
|
|
($re:literal $(,)?) => {{
|
|
static RE: once_cell::sync::OnceCell<regex::Regex> = once_cell::sync::OnceCell::new();
|
|
RE.get_or_init(|| regex::Regex::new($re).expect("invalid regex"))
|
|
}};
|
|
}
|
|
|
|
pub fn filenamify<S: AsRef<str>>(input: S) -> String {
|
|
let reserved = regex!("[<>:\"/\\\\|?*\u{0000}-\u{001F}\u{007F}\u{0080}-\u{009F}]+");
|
|
let windows_reserved = regex!("^(con|prn|aux|nul|com\\d|lpt\\d)$");
|
|
let outer_periods = regex!("^\\.+|\\.+$");
|
|
|
|
let replacement = "_";
|
|
|
|
let input = reserved.replace_all(input.as_ref(), replacement);
|
|
let input = outer_periods.replace_all(input.as_ref(), replacement);
|
|
|
|
let mut result = input.into_owned();
|
|
if windows_reserved.is_match(result.as_str()) {
|
|
result.push_str(replacement);
|
|
}
|
|
|
|
result
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::filenamify;
|
|
|
|
#[test]
|
|
fn test_filenamify() {
|
|
assert_eq!(filenamify("foo/bar"), "foo_bar");
|
|
assert_eq!(filenamify("foo//bar"), "foo_bar");
|
|
assert_eq!(filenamify("//foo//bar//"), "_foo_bar_");
|
|
assert_eq!(filenamify("foo\\bar"), "foo_bar");
|
|
assert_eq!(filenamify("foo\\\\\\bar"), "foo_bar");
|
|
assert_eq!(filenamify(r"foo\\bar"), "foo_bar");
|
|
assert_eq!(filenamify(r"foo\\\\\\bar"), "foo_bar");
|
|
assert_eq!(filenamify("////foo////bar////"), "_foo_bar_");
|
|
assert_eq!(filenamify("foo\u{0000}bar"), "foo_bar");
|
|
assert_eq!(filenamify("\"foo<>bar*"), "_foo_bar_");
|
|
assert_eq!(filenamify("."), "_");
|
|
assert_eq!(filenamify(".."), "_");
|
|
assert_eq!(filenamify("./"), "__");
|
|
assert_eq!(filenamify("../"), "__");
|
|
assert_eq!(filenamify("../../foo/bar"), "__.._foo_bar");
|
|
assert_eq!(filenamify("foo.bar."), "foo.bar_");
|
|
assert_eq!(filenamify("foo.bar.."), "foo.bar_");
|
|
assert_eq!(filenamify("foo.bar..."), "foo.bar_");
|
|
assert_eq!(filenamify("con"), "con_");
|
|
assert_eq!(filenamify("com1"), "com1_");
|
|
assert_eq!(filenamify(":nul|"), "_nul_");
|
|
assert_eq!(filenamify("foo/bar/nul"), "foo_bar_nul");
|
|
assert_eq!(filenamify("file:///file.tar.gz"), "file_file.tar.gz");
|
|
assert_eq!(filenamify("http://www.google.com"), "http_www.google.com");
|
|
assert_eq!(
|
|
filenamify("https://www.youtube.com/watch?v=dQw4w9WgXcQ"),
|
|
"https_www.youtube.com_watch_v=dQw4w9WgXcQ"
|
|
);
|
|
}
|
|
}
|