zeroshade commented on code in PR #3099: URL: https://github.com/apache/arrow-adbc/pull/3099#discussion_r2190488905
########## rust/core/src/driver_manager.rs: ########## @@ -1310,3 +1461,489 @@ impl Drop for ManagedStatement { unsafe { method(statement.deref_mut(), null_mut()) }; } } + +const fn current_arch() -> &'static str { + #[cfg(target_arch = "x86_64")] + const ARCH: &str = "amd64"; + #[cfg(target_arch = "aarch64")] + const ARCH: &str = "arm64"; + #[cfg(not(any(target_arch = "x86_64", target_arch = "aarch64")))] + const ARCH: &str = std::env::consts::ARCH; + + #[cfg(target_os = "macos")] + const OS: &str = "osx"; + #[cfg(not(target_os = "macos"))] + const OS: &str = std::env::consts::OS; + + #[cfg(target_env = "musl")] + const EXTRA: &str = "_musl"; + #[cfg(all(target_os = "windows", target_env = "gnu"))] + const EXTRA: &str = "_mingw"; + #[cfg(not(any(target_env = "musl", all(target_os = "windows", target_env = "gnu"))))] + const EXTRA: &str = ""; + + concat!(OS, "_", ARCH, EXTRA) +} + +#[cfg(target_os = "windows")] +extern crate windows_sys as windows; + +#[cfg(target_os = "windows")] +mod target_windows { + use std::ffi::c_void; + use std::ffi::OsString; + use std::os::windows::ffi::OsStringExt; + use std::path::PathBuf; + use std::slice; + + use super::windows::Win32::UI::Shell; + + fn user_config_dir() -> Option<PathBuf> { + unsafe { + let mut path_ptr: windows::core::PWSTR = std::ptr::null_mut(); + let result = Shell::SHGetKnownFolderPath( + Shell::FOLDERID_LocalAppData, + 0, + std::ptr::null_mut(), + &mut path_ptr, + ); + + if result == 0 { + let len = windows::Win32::Globalization::lstrlenW(path_ptr) as usize; + let path = slice::from_raw_parts(path_ptr, len); + let ostr: OsString = OsStringExt::from_wide(path); + windows::Win32::System::Com::CoTaskMemFree(path_ptr as *const c_void); + Some(PathBuf::from(ostr)) + } else { + windows::Win32::System::Com::CoTaskMemFree(path_ptr as *const c_void); + None + } + } + } +} Review Comment: where do you think I figured out how to do it? :smile: Added a comment to reference `dirs`. I didn't want to rely on an archived repo/dependency -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: github-unsubscr...@arrow.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org