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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
#![cfg(target_os = "linux")]
#![deny(missing_docs)]
use std::env;
use std::path::PathBuf;
use std::process::Command;
use BaseDirs;
use UserDirs;
use ProjectDirs;
impl BaseDirs {
pub fn new() -> BaseDirs {
let home_dir = env::home_dir().unwrap();
let cache_dir = env::var("XDG_CACHE_HOME") .ok().and_then(is_absolute_path).unwrap_or(home_dir.join(".cache"));
let config_dir = env::var("XDG_CONFIG_HOME").ok().and_then(is_absolute_path).unwrap_or(home_dir.join(".config"));
let data_dir = env::var("XDG_DATA_HOME") .ok().and_then(is_absolute_path).unwrap_or(home_dir.join(".local/share"));
let data_local_dir = data_dir.clone();
let runtime_dir = env::var("XDG_RUNTIME_DIR").ok().and_then(is_absolute_path);
let executable_dir =
env::var("XDG_BIN_HOME").ok().and_then(is_absolute_path).unwrap_or({
let mut new_dir = data_dir.clone(); new_dir.pop(); new_dir.push("bin"); new_dir });
BaseDirs {
home_dir: home_dir,
cache_dir: cache_dir,
config_dir: config_dir,
data_dir: data_dir,
data_local_dir: data_local_dir,
executable_dir: Some(executable_dir),
runtime_dir: runtime_dir
}
}
}
impl UserDirs {
pub fn new() -> UserDirs {
let home_dir = env::home_dir().unwrap();
let data_dir = env::var("XDG_DATA_HOME").ok().and_then(is_absolute_path).unwrap_or(home_dir.join(".local/share"));
let font_dir = data_dir.join("fonts");
UserDirs {
home_dir: home_dir,
audio_dir: run_xdg_user_dir_command("MUSIC"),
desktop_dir: run_xdg_user_dir_command("DESKTOP"),
document_dir: run_xdg_user_dir_command("DOCUMENTS"),
download_dir: run_xdg_user_dir_command("DOWNLOAD"),
font_dir: Some(font_dir),
picture_dir: run_xdg_user_dir_command("PICTURES"),
public_dir: run_xdg_user_dir_command("PUBLICSHARE"),
template_dir: run_xdg_user_dir_command("TEMPLATES"),
video_dir: run_xdg_user_dir_command("VIDEOS")
}
}
}
#[deny(missing_docs)]
impl ProjectDirs {
pub fn from_path(project_path: PathBuf) -> ProjectDirs {
let home_dir = env::home_dir().unwrap();
let cache_dir = env::var("XDG_CACHE_HOME") .ok().and_then(is_absolute_path).unwrap_or(home_dir.join(".cache")).join(&project_path);
let config_dir = env::var("XDG_CONFIG_HOME").ok().and_then(is_absolute_path).unwrap_or(home_dir.join(".config")).join(&project_path);
let data_dir = env::var("XDG_DATA_HOME") .ok().and_then(is_absolute_path).unwrap_or(home_dir.join(".local/share")).join(&project_path);
let data_local_dir = data_dir.clone();
let runtime_dir = env::var("XDG_RUNTIME_DIR").ok().and_then(is_absolute_path).map(|o| o.join(&project_path));
ProjectDirs {
project_path: project_path,
cache_dir: cache_dir,
config_dir: config_dir,
data_dir: data_dir,
data_local_dir: data_local_dir,
runtime_dir: runtime_dir
}
}
#[allow(unused_variables)]
pub fn from(qualifier: &str, organization: &str, application: &str) -> ProjectDirs {
ProjectDirs::from_path(PathBuf::from(&trim_and_lowercase_then_replace_spaces(application, "")))
}
}
fn is_absolute_path(path: String) -> Option<PathBuf> {
let path = PathBuf::from(path);
if path.is_absolute() {
Some(path)
} else {
None
}
}
fn run_xdg_user_dir_command(arg: &str) -> Option<PathBuf> {
use std::ffi::OsString;
use std::os::unix::ffi::OsStringExt;
let mut out = Command::new("xdg-user-dir").arg(arg).output().ok()?.stdout;
let out_len = out.len();
out.truncate(out_len - 1);
Some(PathBuf::from(OsString::from_vec(out)))
}
fn trim_and_lowercase_then_replace_spaces(name: &str, replacement: &str) -> String {
let mut buf = String::with_capacity(name.len());
let mut parts = name.split_whitespace();
let mut current_part = parts.next();
let replace = !replacement.is_empty();
while current_part.is_some() {
let value = current_part.unwrap().to_lowercase();
buf.push_str(&value);
current_part = parts.next();
if replace && current_part.is_some() {
buf.push_str(replacement);
}
}
buf
}
#[cfg(test)]
mod tests {
use lin::trim_and_lowercase_then_replace_spaces;
#[test]
fn test_trim_and_lowercase_then_replace_spaces() {
let input1 = "Bar App";
let actual1 = trim_and_lowercase_then_replace_spaces(input1, "-");
let expected1 = "bar-app";
assert_eq!(expected1, actual1);
let input2 = "BarApp-Foo";
let actual2 = trim_and_lowercase_then_replace_spaces(input2, "-");
let expected2 = "barapp-foo";
assert_eq!(expected2, actual2);
let input3 = " Bar App ";
let actual3 = trim_and_lowercase_then_replace_spaces(input3, "-");
let expected3 = "bar-app";
assert_eq!(expected3, actual3);
let input4 = " Bar App ";
let actual4 = trim_and_lowercase_then_replace_spaces(input4, "-");
let expected4 = "bar-app";
assert_eq!(expected4, actual4);
}
#[test]
fn test_file_user_dirs_exists() {
let base_dirs = ::BaseDirs::new();
let user_dirs_file = base_dirs.config_dir().join("user-dirs.dirs");
println!("{:?} exists: {:?}", user_dirs_file, user_dirs_file.exists());
}
}