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
|
#[derive(Clone, Copy)]
pub(crate) enum ConfigEdit<'a> {
Theme(&'a str),
StatusIndicators(super::StatusIndicatorStyle),
Sound(bool),
ToastDelivery(super::ToastDelivery),
}
impl ConfigEdit<'_> {
pub(crate) fn description(self) -> &'static str {
match self {
Self::Theme(_) => "theme",
Self::StatusIndicators(_) => "status indicators",
Self::Sound(_) => "sound setting",
Self::ToastDelivery(_) => "toast setting",
}
}
pub(crate) fn apply(self, content: &str) -> String {
match self {
Self::Theme(name) => {
let content =
super::upsert_section_value(content, "theme", "name", &format!("\"{name}\""));
super::upsert_section_bool(&content, "theme", "auto_switch", false)
}
Self::StatusIndicators(style) => super::upsert_section_value(
content,
"ui",
"status_indicators",
&format!("\"{}\"", style.as_str()),
),
Self::Sound(enabled) => {
super::upsert_section_bool(content, "ui.sound", "enabled", enabled)
}
Self::ToastDelivery(delivery) => {
let value = match delivery {
super::ToastDelivery::Off => "\"off\"",
super::ToastDelivery::Herdr => "\"herdr\"",
super::ToastDelivery::Terminal => "\"terminal\"",
super::ToastDelivery::System => "\"system\"",
};
let content = super::upsert_section_value(content, "ui.toast", "delivery", value);
super::remove_section_key(&content, "ui.toast", "enabled")
}
}
}
}
pub(crate) fn update_file_at(
path: &std::path::Path,
description: &str,
update: impl FnOnce(&str) -> String,
) -> Result<(), String> {
if let Some(parent) = path.parent() {
std::fs::create_dir_all(parent)
.map_err(|error| format!("failed to create config directory: {error}"))?;
}
let content = match super::io::read_optional_config(path) {
Ok(Some(content)) => content,
Ok(None) => String::new(),
Err(error) => {
return Err(format!(
"failed to read config before saving {description}: {error}"
));
}
};
std::fs::write(path, update(&content))
.map_err(|error| format!("failed to save {description}: {error}"))
}
pub(crate) fn write_edit(edit: ConfigEdit<'_>) -> Result<(), String> {
update_file_at(&super::config_path(), edit.description(), |content| {
edit.apply(content)
})
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn update_file_at_does_not_move_a_leading_bom_into_the_file() {
let dir = std::env::temp_dir().join(format!("herdr-config-bom-{}", std::process::id()));
std::fs::create_dir_all(&dir).unwrap();
let path = dir.join("config.toml");
std::fs::write(
&path,
b"\xEF\xBB\xBF[terminal]\ndefault_shell = \"pwsh.exe\"\n",
)
.unwrap();
update_file_at(&path, "onboarding setting", |content| {
crate::config::upsert_top_level_bool(content, "onboarding", false)
})
.unwrap();
let written = std::fs::read_to_string(&path).unwrap();
let _ = std::fs::remove_dir_all(dir);
assert!(
!written.contains('\u{feff}'),
"unexpected BOM in {written:?}"
);
assert!(
toml::from_str::<toml::Value>(&written).is_ok(),
"written config is not valid TOML: {written:?}"
);
}
}
|
