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
|
use config::{Config, File};
use std::{env, ffi::OsStr, path::PathBuf};
/// Reads in `.erdtree.toml` file.
#[cfg(windows)]
pub fn load() -> Option<Config> {
windows::load_toml()
}
#[cfg(not(any(windows, unix)))]
pub fn load() -> Option<Config> {
None
}
/// Reads in `.erdtree.toml` file.
#[cfg(unix)]
pub fn load() -> Option<Config> {
unix::load_toml()
}
/// Attempts to load in `.erdtree.toml` from `$ERDTREE_TOML_PATH`.
fn toml_from_env() -> Option<Config> {
let path = match env::var_os(super::ERDTREE_TOML_PATH).map(PathBuf::from) {
Some(config_path) => config_path,
None => return None,
};
let file = path
.file_stem()
.and_then(OsStr::to_str)
.map(File::with_name)?;
Config::builder().add_source(file).build().ok()
}
/// Concerned with how to load `.erdtree.toml` on Unix systems.
#[cfg(unix)]
mod unix {
use super::super::{
CONFIG_DIR, ERDTREE_CONFIG_FILE, ERDTREE_CONFIG_TOML, ERDTREE_DIR, HOME, XDG_CONFIG_HOME,
};
use config::{Config, File};
use std::{env, ffi::OsStr, path::PathBuf};
/// Looks for `.erdtree.toml` in the following locations in order:
///
/// - `$XDG_CONFIG_HOME/erdtree/.erdtree.toml`
/// - `$XDG_CONFIG_HOME/.erdtree.toml`
/// - `$HOME/.config/erdtree/.erdtree.toml`
/// - `$HOME/.erdtree.toml`
pub fn load_toml() -> Option<Config> {
super::toml_from_env()
.or_else(toml_from_xdg_path)
.or_else(toml_from_home)
}
/// Looks for `.erdtree.toml` in the following locations in order:
///
/// - `$XDG_CONFIG_HOME/erdtree/.erdtree.toml`
/// - `$XDG_CONFIG_HOME/.erdtree.toml`
fn toml_from_xdg_path() -> Option<Config> {
let path = match env::var_os(XDG_CONFIG_HOME).map(PathBuf::from) {
Some(dir_path) => dir_path,
None => return None,
};
let file = path
.join(ERDTREE_DIR)
.join(ERDTREE_CONFIG_TOML)
.file_stem()
.and_then(OsStr::to_str)
.map(File::with_name)?;
if let Ok(config) = Config::builder().add_source(file).build() {
return Some(config);
}
let file = path
.join(ERDTREE_CONFIG_TOML)
.file_stem()
.and_then(OsStr::to_str)
.map(File::with_name)?;
Config::builder().add_source(file).build().ok()
}
/// Looks for `.erdtree.toml` in the following locations in order:
///
/// - `$HOME/.config/erdtree/.erdtree.toml`
/// - `$HOME/.erdtree.toml`
fn toml_from_home() -> Option<Config> {
let home = match env::var_os(HOME).map(PathBuf::from) {
Some(path) => path,
None => return None, // Why don't you have `HOME` set? Weirdo.
};
let source = home
.join(CONFIG_DIR)
.join(ERDTREE_DIR)
.join(ERDTREE_CONFIG_FILE)
.to_str()
.map(File::with_name)?;
if let Ok(config) = Config::builder().add_source(source).build() {
return Some(config);
}
let file = home
.join(ERDTREE_CONFIG_TOML)
.file_stem()
.and_then(OsStr::to_str)
.map(File::with_name)?;
Config::builder().add_source(file).build().ok()
}
}
/// Concerned with how to load `.erdtree.toml` on Windows.
#[cfg(windows)]
mod windows {
use super::super::{ERDTREE_CONFIG_FILE, ERDTREE_DIR};
use config::{Config, File};
/// Try to read in config from the following location:
/// - `%APPDATA%\erdtree\.erdtree.toml`
pub fn load_toml() -> Option<Config> {
super::toml_from_env().or_else(toml_from_appdata)
}
/// Try to read in config from the following location:
/// - `%APPDATA%\erdtree\.erdtree.toml`
fn toml_from_appdata() -> Option<Config> {
let file = dirs::config_dir().and_then(|config_dir| {
config_dir
.join(ERDTREE_DIR)
.join(ERDTREE_CONFIG_FILE)
.to_str()
.map(File::with_name)
})?;
Config::builder().add_source(file).build().ok()
}
}
|
