40 lines
1.1 KiB
Rust
40 lines
1.1 KiB
Rust
|
use std::path::PathBuf;
|
||
|
|
|
||
|
|
use crate::error::NotfilesError;
|
||
|
|
|
||
|
|
/// Expand a leading `~` to the user's home directory.
|
||
|
|
pub fn expand_tilde(path: &str) -> Result<PathBuf, NotfilesError> {
|
||
|
|
if path == "~" {
|
||
|
|
return dirs::home_dir().ok_or_else(|| NotfilesError::Path("cannot determine home directory".into()));
|
||
|
|
}
|
||
|
|
if let Some(rest) = path.strip_prefix("~/") {
|
||
|
|
let home = dirs::home_dir()
|
||
|
|
.ok_or_else(|| NotfilesError::Path("cannot determine home directory".into()))?;
|
||
|
|
return Ok(home.join(rest));
|
||
|
|
}
|
||
|
|
Ok(PathBuf::from(path))
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
mod tests {
|
||
|
|
use super::*;
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_expand_tilde_home() {
|
||
|
|
let home = dirs::home_dir().unwrap();
|
||
|
|
assert_eq!(expand_tilde("~").unwrap(), home);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_expand_tilde_subpath() {
|
||
|
|
let home = dirs::home_dir().unwrap();
|
||
|
|
assert_eq!(expand_tilde("~/foo/bar").unwrap(), home.join("foo/bar"));
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn test_expand_tilde_absolute() {
|
||
|
|
assert_eq!(expand_tilde("/usr/bin").unwrap(), PathBuf::from("/usr/bin"));
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|