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
|
use std::env;
const COLORTERM: &str = "COLORTERM";
const BAT_THEME: &str = "BAT_THEME";
const GIT_CONFIG_PARAMETERS: &str = "GIT_CONFIG_PARAMETERS";
const GIT_PREFIX: &str = "GIT_PREFIX";
const DELTA_FEATURES: &str = "DELTA_FEATURES";
const DELTA_NAVIGATE: &str = "DELTA_NAVIGATE";
const DELTA_EXPERIMENTAL_MAX_LINE_DISTANCE_FOR_NAIVELY_PAIRED_LINES: &str =
"DELTA_EXPERIMENTAL_MAX_LINE_DISTANCE_FOR_NAIVELY_PAIRED_LINES";
const DELTA_PAGER: &str = "DELTA_PAGER";
#[derive(Default, Clone)]
pub struct DeltaEnv {
pub bat_theme: Option<String>,
pub colorterm: Option<String>,
pub current_dir: Option<std::path::PathBuf>,
pub experimental_max_line_distance_for_naively_paired_lines: Option<String>,
pub features: Option<String>,
pub git_config_parameters: Option<String>,
pub git_prefix: Option<String>,
pub hostname: Option<String>,
pub navigate: Option<String>,
pub pagers: (Option<String>, Option<String>),
}
impl DeltaEnv {
/// Create a structure with current environment variable
pub fn init() -> Self {
let bat_theme = env::var(BAT_THEME).ok();
let colorterm = env::var(COLORTERM).ok();
let experimental_max_line_distance_for_naively_paired_lines =
env::var(DELTA_EXPERIMENTAL_MAX_LINE_DISTANCE_FOR_NAIVELY_PAIRED_LINES).ok();
let features = env::var(DELTA_FEATURES).ok();
let git_config_parameters = env::var(GIT_CONFIG_PARAMETERS).ok();
let git_prefix = env::var(GIT_PREFIX).ok();
let hostname = hostname();
let navigate = env::var(DELTA_NAVIGATE).ok();
let current_dir = env::current_dir().ok();
let pagers = (
env::var(DELTA_PAGER).ok(),
// We're using `bat::config::get_pager_executable` here instead of just returning
// the pager from the environment variables, because we want to make sure
// that the pager is a valid pager from env and handle the case of
// the PAGER being set to something invalid like "most" and "more".
bat::config::get_pager_executable(None),
);
Self {
bat_theme,
colorterm,
current_dir,
experimental_max_line_distance_for_naively_paired_lines,
features,
git_config_parameters,
git_prefix,
hostname,
navigate,
pagers,
}
}
}
fn hostname() -> Option<String> {
grep_cli::hostname().ok()?.to_str().map(|s| s.to_string())
}
#[cfg(test)]
pub mod tests {
use super::DeltaEnv;
use lazy_static::lazy_static;
use std::env;
use std::sync::{Arc, Mutex};
lazy_static! {
static ref ENV_ACCESS: Arc<Mutex<()>> = Arc::new(Mutex::new(()));
}
#[test]
fn test_env_parsing() {
let _guard = ENV_ACCESS.lock().unwrap();
let feature = "Awesome Feature";
env::set_var("DELTA_FEATURES", feature);
let env = DeltaEnv::init();
assert_eq!(env.features, Some(feature.into()));
// otherwise `current_dir` is not used in the test cfg:
assert_eq!(env.current_dir, env::current_dir().ok());
}
#[test]
fn test_env_parsing_with_pager_set_to_bat() {
let _guard = ENV_ACCESS.lock().unwrap();
env::set_var("PAGER", "bat");
let env = DeltaEnv::init();
assert_eq!(
env.pagers.1,
Some("bat".into()),
"Expected env.pagers.1 == Some(bat) but was {:?}",
env.pagers.1
);
}
#[test]
fn test_env_parsing_with_pager_set_to_more() {
let _guard = ENV_ACCESS.lock().unwrap();
env::set_var("PAGER", "more");
let env = DeltaEnv::init();
assert_eq!(env.pagers.1, Some("less".into()));
}
#[test]
fn test_env_parsing_with_pager_set_to_most() {
let _guard = ENV_ACCESS.lock().unwrap();
env::set_var("PAGER", "most");
let env = DeltaEnv::init();
assert_eq!(env.pagers.1, Some("less".into()));
}
}
|
