1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
// Copyright 2016 Joe Wilm, The Alacritty Project Contributors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#![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};
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()
}
|
