diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-17 12:02:58 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-17 12:02:58 +0000 |
commit | 698f8c2f01ea549d77d7dc3338a12e04c11057b9 (patch) | |
tree | 173a775858bd501c378080a10dca74132f05bc50 /vendor/dirs-sys-next/src/lib.rs | |
parent | Initial commit. (diff) | |
download | rustc-698f8c2f01ea549d77d7dc3338a12e04c11057b9.tar.xz rustc-698f8c2f01ea549d77d7dc3338a12e04c11057b9.zip |
Adding upstream version 1.64.0+dfsg1.upstream/1.64.0+dfsg1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'vendor/dirs-sys-next/src/lib.rs')
-rw-r--r-- | vendor/dirs-sys-next/src/lib.rs | 198 |
1 files changed, 198 insertions, 0 deletions
diff --git a/vendor/dirs-sys-next/src/lib.rs b/vendor/dirs-sys-next/src/lib.rs new file mode 100644 index 000000000..f76862e85 --- /dev/null +++ b/vendor/dirs-sys-next/src/lib.rs @@ -0,0 +1,198 @@ +#![warn(rust_2018_idioms)] + +use std::ffi::OsString; +use std::path::PathBuf; + +// we don't need to explicitly handle empty strings in the code above, +// because an empty string is not considered to be a absolute path here. +pub fn is_absolute_path(path: OsString) -> Option<PathBuf> { + let path = PathBuf::from(path); + if path.is_absolute() { + Some(path) + } else { + None + } +} + +#[cfg(all(unix, not(target_os = "redox")))] +mod target_unix_not_redox { + + use std::env; + use std::ffi::{CStr, OsString}; + use std::mem; + use std::os::unix::ffi::OsStringExt; + use std::path::PathBuf; + use std::ptr; + + // https://github.com/rust-lang/rust/blob/ef3e3863939217678e5f7e755c4234d224107c64/library/std/src/sys/unix/os.rs#L587 + pub fn home_dir() -> Option<PathBuf> { + return env::var_os("HOME") + .and_then(|h| if h.is_empty() { None } else { Some(h) }) + .or_else(|| unsafe { fallback() }) + .map(PathBuf::from); + + #[cfg(any(target_os = "android", target_os = "ios", target_os = "emscripten"))] + unsafe fn fallback() -> Option<OsString> { + None + } + #[cfg(not(any(target_os = "android", target_os = "ios", target_os = "emscripten")))] + unsafe fn fallback() -> Option<OsString> { + let amt = match libc::sysconf(libc::_SC_GETPW_R_SIZE_MAX) { + n if n < 0 => 512 as usize, + n => n as usize, + }; + let mut buf = Vec::with_capacity(amt); + let mut passwd: libc::passwd = mem::zeroed(); + let mut result = ptr::null_mut(); + match libc::getpwuid_r(libc::getuid(), &mut passwd, buf.as_mut_ptr(), buf.capacity(), &mut result) { + 0 if !result.is_null() => { + let ptr = passwd.pw_dir as *const _; + let bytes = CStr::from_ptr(ptr).to_bytes(); + if bytes.is_empty() { + None + } else { + Some(OsStringExt::from_vec(bytes.to_vec())) + } + } + _ => None, + } + } + } +} + +#[cfg(all(unix, not(target_os = "redox")))] +pub use self::target_unix_not_redox::home_dir; + +#[cfg(target_os = "redox")] +mod target_redox { + + use std::path::PathBuf; + + use redox_users::{All, AllUsers, Config}; + + pub fn home_dir() -> Option<PathBuf> { + let current_uid = redox_users::get_uid().ok()?; + let users = AllUsers::basic(Config::default()).ok()?; + let user = users.get_by_id(current_uid)?; + + Some(PathBuf::from(user.home.clone())) + } +} + +#[cfg(target_os = "redox")] +pub use self::target_redox::home_dir; + +#[cfg(all(unix, not(any(target_os = "macos", target_os = "ios"))))] +mod xdg_user_dirs; + +#[cfg(all(unix, not(any(target_os = "macos", target_os = "ios"))))] +mod target_unix_not_mac { + + use std::collections::HashMap; + use std::env; + use std::path::{Path, PathBuf}; + + use super::xdg_user_dirs; + use super::{home_dir, is_absolute_path}; + + fn user_dir_file(home_dir: &Path) -> PathBuf { + env::var_os("XDG_CONFIG_HOME") + .and_then(is_absolute_path) + .unwrap_or_else(|| home_dir.join(".config")) + .join("user-dirs.dirs") + } + + // this could be optimized further to not create a map and instead retrieve the requested path only + pub fn user_dir(user_dir_name: &str) -> Option<PathBuf> { + if let Some(home_dir) = home_dir() { + xdg_user_dirs::single(&home_dir, &user_dir_file(&home_dir), user_dir_name).remove(user_dir_name) + } else { + None + } + } + + pub fn user_dirs(home_dir_path: &Path) -> HashMap<String, PathBuf> { + xdg_user_dirs::all(home_dir_path, &user_dir_file(home_dir_path)) + } +} + +#[cfg(all(unix, not(any(target_os = "macos", target_os = "ios"))))] +pub use self::target_unix_not_mac::{user_dir, user_dirs}; + +#[cfg(target_os = "windows")] +mod target_windows { + + use std::ffi::OsString; + use std::os::windows::ffi::OsStringExt; + use std::path::PathBuf; + use std::ptr; + use std::slice; + + use winapi::shared::winerror; + use winapi::um::{combaseapi, knownfolders, shlobj, shtypes, winbase, winnt}; + + pub fn known_folder(folder_id: shtypes::REFKNOWNFOLDERID) -> Option<PathBuf> { + unsafe { + let mut path_ptr: winnt::PWSTR = ptr::null_mut(); + let result = shlobj::SHGetKnownFolderPath(folder_id, 0, ptr::null_mut(), &mut path_ptr); + if result == winerror::S_OK { + let len = winbase::lstrlenW(path_ptr) as usize; + let path = slice::from_raw_parts(path_ptr, len); + let ostr: OsString = OsStringExt::from_wide(path); + combaseapi::CoTaskMemFree(path_ptr as *mut winapi::ctypes::c_void); + Some(PathBuf::from(ostr)) + } else { + None + } + } + } + + pub fn known_folder_profile() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Profile) + } + + pub fn known_folder_roaming_app_data() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_RoamingAppData) + } + + pub fn known_folder_local_app_data() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_LocalAppData) + } + + pub fn known_folder_music() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Music) + } + + pub fn known_folder_desktop() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Desktop) + } + + pub fn known_folder_documents() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Documents) + } + + pub fn known_folder_downloads() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Downloads) + } + + pub fn known_folder_pictures() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Pictures) + } + + pub fn known_folder_public() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Public) + } + pub fn known_folder_templates() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Templates) + } + pub fn known_folder_videos() -> Option<PathBuf> { + known_folder(&knownfolders::FOLDERID_Videos) + } +} + +#[cfg(target_os = "windows")] +pub use self::target_windows::{ + known_folder, known_folder_desktop, known_folder_documents, known_folder_downloads, known_folder_local_app_data, + known_folder_music, known_folder_pictures, known_folder_profile, known_folder_public, + known_folder_roaming_app_data, known_folder_templates, known_folder_videos, +}; |