3 files changed, 262 insertions, 0 deletions
diff --git a/alacritty/src/macos/locale.rs b/alacritty/src/macos/locale.rs
new file mode 100644
index 00000000..6c02c12c
--- /dev/null
+++ b/alacritty/src/macos/locale.rs
@@ -0,0 +1,100 @@
+#![allow(clippy::let_unit_value)]
+
+use std::env;
+use std::ffi::{CStr, CString};
+use std::os::raw::c_char;
+use std::slice;
+use std::str;
+
+use libc::{setlocale, LC_ALL, LC_CTYPE};
+use log::debug;
+use objc::runtime::{Class, Object};
+use objc::{msg_send, sel, sel_impl};
+
+const FALLBACK_LOCALE: &str = "UTF-8";
+
+pub fn set_locale_environment() {
+    let env_locale_c = CString::new("").unwrap();
+    let env_locale_ptr = unsafe { setlocale(LC_ALL, env_locale_c.as_ptr()) };
+    if !env_locale_ptr.is_null() {
+        let env_locale = unsafe { CStr::from_ptr(env_locale_ptr).to_string_lossy() };
+
+        // Assume `C` locale means unchanged, since it is the default anyways.
+        if env_locale != "C" {
+            debug!("Using environment locale: {}", env_locale);
+            return;
+        }
+    }
+
+    let system_locale = system_locale();
+
+    // Set locale to system locale.
+    let system_locale_c = CString::new(system_locale.clone()).expect("nul byte in system locale");
+    let lc_all = unsafe { setlocale(LC_ALL, system_locale_c.as_ptr()) };
+
+    // Check if system locale was valid or not.
+    if lc_all.is_null() {
+        // Use fallback locale.
+        debug!("Using fallback locale: {}", FALLBACK_LOCALE);
+
+        let fallback_locale_c = CString::new(FALLBACK_LOCALE).unwrap();
+        unsafe { setlocale(LC_CTYPE, fallback_locale_c.as_ptr()) };
+
+        env::set_var("LC_CTYPE", FALLBACK_LOCALE);
+    } else {
+        // Use system locale.
+        debug!("Using system locale: {}", system_locale);
+
+        env::set_var("LC_ALL", system_locale);
+    }
+}
+
+/// Determine system locale based on language and country code.
+fn system_locale() -> String {
+    unsafe {
+        let locale_class = Class::get("NSLocale").unwrap();
+        let locale: *const Object = msg_send![locale_class, currentLocale];
+        let _: () = msg_send![locale_class, release];
+
+        // `localeIdentifier` returns extra metadata with the locale (including currency and
+        // collator) on newer versions of macOS. This is not a valid locale, so we use
+        // `languageCode` and `countryCode`, if they're available (macOS 10.12+):
+        //
+        // https://developer.apple.com/documentation/foundation/nslocale/1416263-localeidentifier?language=objc
+        // https://developer.apple.com/documentation/foundation/nslocale/1643060-countrycode?language=objc
+        // https://developer.apple.com/documentation/foundation/nslocale/1643026-languagecode?language=objc
+        let is_language_code_supported: bool =
+            msg_send![locale, respondsToSelector: sel!(languageCode)];
+        let is_country_code_supported: bool =
+            msg_send![locale, respondsToSelector: sel!(countryCode)];
+        let locale_id = if is_language_code_supported && is_country_code_supported {
+            let language_code: *const Object = msg_send![locale, languageCode];
+            let language_code_str = nsstring_as_str(language_code).to_owned();
+            let _: () = msg_send![language_code, release];
+
+            let country_code: *const Object = msg_send![locale, countryCode];
+            let country_code_str = nsstring_as_str(country_code).to_owned();
+            let _: () = msg_send![country_code, release];
+
+            format!("{}_{}.UTF-8", &language_code_str, &country_code_str)
+        } else {
+            let identifier: *const Object = msg_send![locale, localeIdentifier];
+            let identifier_str = nsstring_as_str(identifier).to_owned();
+            let _: () = msg_send![identifier, release];
+
+            identifier_str + ".UTF-8"
+        };
+
+        let _: () = msg_send![locale, release];
+
+        locale_id
+    }
+}
+
+const UTF8_ENCODING: usize = 4;
+
+unsafe fn nsstring_as_str<'a>(nsstring: *const Object) -> &'a str {
+    let cstr: *const c_char = msg_send![nsstring, UTF8String];
+    let len: usize = msg_send![nsstring, lengthOfBytesUsingEncoding: UTF8_ENCODING];
+    str::from_utf8(slice::from_raw_parts(cstr as *const u8, len)).unwrap()
+}
diff --git a/alacritty/src/macos/mod.rs b/alacritty/src/macos/mod.rs
new file mode 100644
index 00000000..1d630730
--- /dev/null
+++ b/alacritty/src/macos/mod.rs
@@ -0,0 +1,2 @@
+pub mod locale;
+pub mod proc;
diff --git a/alacritty/src/macos/proc.rs b/alacritty/src/macos/proc.rs
new file mode 100644
index 00000000..be7e4891
--- /dev/null
+++ b/alacritty/src/macos/proc.rs
@@ -0,0 +1,160 @@
+use std::ffi::{CStr, CString, IntoStringError};
+use std::fmt::{self, Display, Formatter};
+use std::io;
+use std::mem::{self, MaybeUninit};
+use std::os::raw::{c_int, c_void};
+use std::path::PathBuf;
+
+/// Error during working directory retrieval.
+#[derive(Debug)]
+pub enum Error {
+    Io(io::Error),
+
+    /// Error converting into utf8 string.
+    IntoString(IntoStringError),
+
+    /// Expected return size didn't match libproc's.
+    InvalidSize,
+}
+
+impl std::error::Error for Error {
+    fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
+        match self {
+            Error::InvalidSize => None,
+            Error::Io(err) => err.source(),
+            Error::IntoString(err) => err.source(),
+        }
+    }
+}
+
+impl Display for Error {
+    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
+        match self {
+            Error::InvalidSize => write!(f, "Invalid proc_pidinfo return size"),
+            Error::Io(err) => write!(f, "Error getting current working directory: {}", err),
+            Error::IntoString(err) => {
+                write!(f, "Error when parsing current working directory: {}", err)
+            },
+        }
+    }
+}
+
+impl From<io::Error> for Error {
+    fn from(val: io::Error) -> Self {
+        Error::Io(val)
+    }
+}
+
+impl From<IntoStringError> for Error {
+    fn from(val: IntoStringError) -> Self {
+        Error::IntoString(val)
+    }
+}
+
+pub fn cwd(pid: c_int) -> Result<PathBuf, Error> {
+    let mut info = MaybeUninit::<sys::proc_vnodepathinfo>::uninit();
+    let info_ptr = info.as_mut_ptr() as *mut c_void;
+    let size = mem::size_of::<sys::proc_vnodepathinfo>() as c_int;
+
+    let c_str = unsafe {
+        let pidinfo_size = sys::proc_pidinfo(pid, sys::PROC_PIDVNODEPATHINFO, 0, info_ptr, size);
+        match pidinfo_size {
+            c if c < 0 => return Err(io::Error::last_os_error().into()),
+            s if s != size => return Err(Error::InvalidSize),
+            _ => CStr::from_ptr(info.assume_init().pvi_cdir.vip_path.as_ptr()),
+        }
+    };
+
+    Ok(CString::from(c_str).into_string().map(PathBuf::from)?)
+}
+
+/// Bindings for libproc.
+#[allow(non_camel_case_types)]
+mod sys {
+    use std::os::raw::{c_char, c_int, c_longlong, c_void};
+
+    pub const PROC_PIDVNODEPATHINFO: c_int = 9;
+
+    type gid_t = c_int;
+    type off_t = c_longlong;
+    type uid_t = c_int;
+    type fsid_t = fsid;
+
+    #[repr(C)]
+    #[derive(Debug, Copy, Clone)]
+    pub struct fsid {
+        pub val: [i32; 2usize],
+    }
+
+    #[repr(C)]
+    #[derive(Debug, Copy, Clone)]
+    pub struct vinfo_stat {
+        pub vst_dev: u32,
+        pub vst_mode: u16,
+        pub vst_nlink: u16,
+        pub vst_ino: u64,
+        pub vst_uid: uid_t,
+        pub vst_gid: gid_t,
+        pub vst_atime: i64,
+        pub vst_atimensec: i64,
+        pub vst_mtime: i64,
+        pub vst_mtimensec: i64,
+        pub vst_ctime: i64,
+        pub vst_ctimensec: i64,
+        pub vst_birthtime: i64,
+        pub vst_birthtimensec: i64,
+        pub vst_size: off_t,
+        pub vst_blocks: i64,
+        pub vst_blksize: i32,
+        pub vst_flags: u32,
+        pub vst_gen: u32,
+        pub vst_rdev: u32,
+        pub vst_qspare: [i64; 2usize],
+    }
+
+    #[repr(C)]
+    #[derive(Debug, Copy, Clone)]
+    pub struct vnode_info {
+        pub vi_stat: vinfo_stat,
+        pub vi_type: c_int,
+        pub vi_pad: c_int,
+        pub vi_fsid: fsid_t,
+    }
+
+    #[repr(C)]
+    #[derive(Copy, Clone)]
+    pub struct vnode_info_path {
+        pub vip_vi: vnode_info,
+        pub vip_path: [c_char; 1024usize],
+    }
+
+    #[repr(C)]
+    #[derive(Copy, Clone)]
+    pub struct proc_vnodepathinfo {
+        pub pvi_cdir: vnode_info_path,
+        pub pvi_rdir: vnode_info_path,
+    }
+
+    extern "C" {
+        pub fn proc_pidinfo(
+            pid: c_int,
+            flavor: c_int,
+            arg: u64,
+            buffer: *mut c_void,
+            buffersize: c_int,
+        ) -> c_int;
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    use std::env;
+    use std::process;
+
+    #[test]
+    fn cwd_matches_current_dir() {
+        assert_eq!(cwd(process::id() as i32).ok(), env::current_dir().ok());
+    }
+}