mirror of
https://github.com/denoland/deno.git
synced 2024-12-25 08:39:09 -05:00
92 lines
2.5 KiB
Rust
92 lines
2.5 KiB
Rust
|
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
|
||
|
|
||
|
use std::path::PathBuf;
|
||
|
|
||
|
#[cfg(not(windows))]
|
||
|
#[inline]
|
||
|
pub fn strip_unc_prefix(path: PathBuf) -> PathBuf {
|
||
|
path
|
||
|
}
|
||
|
|
||
|
/// Strips the unc prefix (ex. \\?\) from Windows paths.
|
||
|
#[cfg(windows)]
|
||
|
pub fn strip_unc_prefix(path: PathBuf) -> PathBuf {
|
||
|
use std::path::Component;
|
||
|
use std::path::Prefix;
|
||
|
|
||
|
let mut components = path.components();
|
||
|
match components.next() {
|
||
|
Some(Component::Prefix(prefix)) => {
|
||
|
match prefix.kind() {
|
||
|
// \\?\device
|
||
|
Prefix::Verbatim(device) => {
|
||
|
let mut path = PathBuf::new();
|
||
|
path.push(format!(r"\\{}\", device.to_string_lossy()));
|
||
|
path.extend(components.filter(|c| !matches!(c, Component::RootDir)));
|
||
|
path
|
||
|
}
|
||
|
// \\?\c:\path
|
||
|
Prefix::VerbatimDisk(_) => {
|
||
|
let mut path = PathBuf::new();
|
||
|
path.push(prefix.as_os_str().to_string_lossy().replace(r"\\?\", ""));
|
||
|
path.extend(components);
|
||
|
path
|
||
|
}
|
||
|
// \\?\UNC\hostname\share_name\path
|
||
|
Prefix::VerbatimUNC(hostname, share_name) => {
|
||
|
let mut path = PathBuf::new();
|
||
|
path.push(format!(
|
||
|
r"\\{}\{}\",
|
||
|
hostname.to_string_lossy(),
|
||
|
share_name.to_string_lossy()
|
||
|
));
|
||
|
path.extend(components.filter(|c| !matches!(c, Component::RootDir)));
|
||
|
path
|
||
|
}
|
||
|
_ => path,
|
||
|
}
|
||
|
}
|
||
|
_ => path,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[cfg(test)]
|
||
|
mod test {
|
||
|
#[cfg(windows)]
|
||
|
#[test]
|
||
|
fn test_strip_unc_prefix() {
|
||
|
use std::path::PathBuf;
|
||
|
|
||
|
run_test(r"C:\", r"C:\");
|
||
|
run_test(r"C:\test\file.txt", r"C:\test\file.txt");
|
||
|
|
||
|
run_test(r"\\?\C:\", r"C:\");
|
||
|
run_test(r"\\?\C:\test\file.txt", r"C:\test\file.txt");
|
||
|
|
||
|
run_test(r"\\.\C:\", r"\\.\C:\");
|
||
|
run_test(r"\\.\C:\Test\file.txt", r"\\.\C:\Test\file.txt");
|
||
|
|
||
|
run_test(r"\\?\UNC\localhost\", r"\\localhost");
|
||
|
run_test(r"\\?\UNC\localhost\c$\", r"\\localhost\c$");
|
||
|
run_test(
|
||
|
r"\\?\UNC\localhost\c$\Windows\file.txt",
|
||
|
r"\\localhost\c$\Windows\file.txt",
|
||
|
);
|
||
|
run_test(r"\\?\UNC\wsl$\deno.json", r"\\wsl$\deno.json");
|
||
|
|
||
|
run_test(r"\\?\server1", r"\\server1");
|
||
|
run_test(r"\\?\server1\e$\", r"\\server1\e$\");
|
||
|
run_test(
|
||
|
r"\\?\server1\e$\test\file.txt",
|
||
|
r"\\server1\e$\test\file.txt",
|
||
|
);
|
||
|
|
||
|
fn run_test(input: &str, expected: &str) {
|
||
|
assert_eq!(
|
||
|
super::strip_unc_prefix(PathBuf::from(input)),
|
||
|
PathBuf::from(expected)
|
||
|
);
|
||
|
}
|
||
|
}
|
||
|
}
|