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
use std::io::{self, Write};
use std::fs::{File, OpenOptions};
use std::path::PathBuf;
use std::env;
use std::fmt;
use dirs;
pub fn check_or_set<T, U>(var: T, value: U) -> io::Result<()>
where T: fmt::Display + AsRef<std::ffi::OsStr>,
U: fmt::Display,
{
env::var(&var)
.map(|_|())
.or_else(|_| set(var, value))
}
pub fn append<T: fmt::Display>(var: T, value: T) -> io::Result<()> {
let mut profile = get_profile()?;
writeln!(profile, "\nexport {}=\"{}:${}\"", var, value, var)?;
profile.flush()
}
pub fn set<T: fmt::Display, U: fmt::Display>(var: T, value: U) -> io::Result<()> {
let mut profile = get_profile()?;
writeln!(profile, "\nexport {}={}", var, value)?;
profile.flush()
}
fn get_profile() -> io::Result<File> {
dirs::home_dir()
.ok_or_else(|| io::Error::new(io::ErrorKind::Other, "No home directory"))
.and_then(find_profile)
}
#[cfg(target_family = "unix")]
fn find_profile(mut profile: PathBuf) -> io::Result<File> {
profile.push(".bash_profile");
let mut oo = OpenOptions::new();
oo.append(true)
.create(false);
oo.open(profile.clone())
.or_else(|_|{
profile.pop();
profile.push(".bash_login");
oo.open(profile.clone())
})
.or_else(|_|{
profile.pop();
profile.push(".profile");
oo.open(profile.clone())
})
.or_else(|_|{
profile.pop();
profile.push(".bash_profile");
oo.create(true);
oo.open(profile.clone())
})
}