use anyhow::{Context, Result, anyhow};
use std::path::{Path, PathBuf};
use std::process::Command;
use std::time::{SystemTime, UNIX_EPOCH};
use toml_edit::{DocumentMut, InlineTable, Item, Table, TableLike, Value};
const KEYBINDS_TABLE: &str = "keybinds";
const ACTION_FIELD: &str = "action";
const REPEAT_FIELD: &str = "repeat";
#[derive(Debug, PartialEq, Eq)]
pub struct KeybindEdit {
pub chord: String,
pub new_text: String,
pub previous: Option<String>,
}
fn normalized_chord(chord: &str) -> Vec<String> {
let mut parts: Vec<String> = chord
.split('+')
.map(|part| part.trim().to_lowercase())
.filter(|part| !part.is_empty())
.collect();
let key = parts.pop();
parts.sort();
parts.extend(key);
parts
}
pub fn same_chord(left: &str, right: &str) -> bool {
normalized_chord(left) == normalized_chord(right)
}
fn parse(config: &str) -> Result<DocumentMut> {
config
.parse::<DocumentMut>()
.context("the Umbriel config is not valid TOML, so it was not changed")
}
fn keybinds_table(document: &mut DocumentMut) -> Result<&mut dyn TableLike> {
if document.get(KEYBINDS_TABLE).is_none() {
let mut table = Table::new();
table.decor_mut().set_prefix("\n");
document.insert(KEYBINDS_TABLE, Item::Table(table));
}
document
.get_mut(KEYBINDS_TABLE)
.and_then(Item::as_table_like_mut)
.ok_or_else(|| anyhow!("`keybinds` in the Umbriel config is not a table"))
}
fn existing_chord(table: &dyn TableLike, chord: &str) -> Option<String> {
table
.iter()
.map(|(key, _)| key.to_owned())
.find(|key| same_chord(key, chord))
}
fn bound_action(item: &Item) -> Option<String> {
match item.as_value()? {
Value::String(action) => Some(action.value().clone()),
Value::InlineTable(table) => table
.get(ACTION_FIELD)
.and_then(Value::as_str)
.map(str::to_owned),
_ => None,
}
}
fn require_text<'a>(value: &'a str, what: &str) -> Result<&'a str> {
let value = value.trim();
if value.is_empty() {
return Err(anyhow!("the {what} must not be empty"));
}
Ok(value)
}
fn replacement_value(existing: Option<&Value>, action: &str, repeat: Option<bool>) -> Value {
let decor = existing.map(|value| value.decor().clone());
let mut value = match existing {
Some(Value::InlineTable(table)) => {
let mut table = table.clone();
replace_field(&mut table, ACTION_FIELD, Value::from(action));
if let Some(repeat) = repeat {
replace_field(&mut table, REPEAT_FIELD, Value::from(repeat));
}
Value::InlineTable(table)
}
_ => match repeat {
Some(repeat) => {
let mut table = InlineTable::new();
table.insert(ACTION_FIELD, Value::from(action));
table.insert(REPEAT_FIELD, Value::from(repeat));
table.fmt();
Value::InlineTable(table)
}
None => Value::from(action),
},
};
if let Some(decor) = decor {
*value.decor_mut() = decor;
}
value
}
fn replace_field(table: &mut InlineTable, field: &str, mut replacement: Value) {
if let Some(existing) = table.get(field) {
*replacement.decor_mut() = existing.decor().clone();
table.insert(field, replacement);
} else {
table.insert(field, replacement);
table.fmt();
}
}
pub fn plan_set_keybind(
config: &str,
chord: &str,
action: &str,
repeat: Option<bool>,
) -> Result<KeybindEdit> {
let chord = require_text(chord, "key chord")?;
let action = require_text(action, "action")?;
let mut document = parse(config)?;
let table = keybinds_table(&mut document)?;
let (key, previous) = match existing_chord(table, chord) {
Some(key) => {
let item = table.get_mut(&key).expect("chord found above");
let previous = bound_action(item);
let value = replacement_value(item.as_value(), action, repeat);
*item = Item::Value(value);
(key, previous)
}
None => {
table.insert(chord, Item::Value(replacement_value(None, action, repeat)));
(chord.to_owned(), None)
}
};
Ok(KeybindEdit {
chord: key,
new_text: document.to_string(),
previous,
})
}
pub fn plan_remove_keybind(config: &str, chord: &str) -> Result<KeybindEdit> {
let chord = require_text(chord, "key chord")?;
let mut document = parse(config)?;
let not_bound =
|| anyhow!("{chord} is not bound in the Umbriel config, so nothing was removed");
let table = document
.get_mut(KEYBINDS_TABLE)
.and_then(Item::as_table_like_mut)
.ok_or_else(not_bound)?;
let key = existing_chord(table, chord).ok_or_else(not_bound)?;
let previous = remove_keeping_comments_above(table, &key)
.as_ref()
.and_then(bound_action);
Ok(KeybindEdit {
chord: key,
new_text: document.to_string(),
previous,
})
}
pub fn remove_keeping_comments_above(table: &mut dyn TableLike, key: &str) -> Option<Item> {
let leading_comments = table
.key(key)
.and_then(|key| key.leaf_decor().prefix())
.and_then(|prefix| prefix.as_str())
.map(str::to_owned)
.unwrap_or_default();
let next_key = table
.iter()
.map(|(name, _)| name.to_owned())
.skip_while(|name| name != key)
.nth(1);
let removed = table.remove(key);
if let Some(next_key) = next_key
&& let Some(mut next) = table.key_mut(&next_key)
{
let next_prefix = next
.leaf_decor()
.prefix()
.and_then(|prefix| prefix.as_str())
.unwrap_or_default()
.to_owned();
next.leaf_decor_mut()
.set_prefix(format!("{leading_comments}{next_prefix}"));
}
removed
}
pub fn backup_order(name: &str, config_name: &str) -> Option<(u64, u64)> {
let stamp = name.strip_prefix(config_name)?.strip_prefix(".bak-")?;
let digits = |text: &str| {
(!text.is_empty() && text.bytes().all(|byte| byte.is_ascii_digit()))
.then(|| text.parse::<u64>().ok())
.flatten()
};
match stamp.split_once('-') {
Some((seconds, attempt)) => Some((digits(seconds)?, digits(attempt)?)),
None => Some((digits(stamp)?, 1)),
}
}
pub fn next_backup_attempt(names: &[String], config_name: &str, stamp: u64) -> u64 {
names
.iter()
.filter_map(|name| backup_order(name, config_name))
.filter(|(seconds, _)| *seconds == stamp)
.map(|(_, attempt)| attempt + 1)
.max()
.unwrap_or(1)
}
pub fn backups_to_remove(
names: &[String],
config_name: &str,
keep_newest: usize,
just_written: &str,
) -> Vec<String> {
let mut backups: Vec<(u64, u64, &String)> = names
.iter()
.filter(|name| name.as_str() != just_written)
.filter_map(|name| {
backup_order(name, config_name).map(|(seconds, attempt)| (seconds, attempt, name))
})
.collect();
backups.sort_unstable_by_key(|(seconds, attempt, _)| std::cmp::Reverse((*seconds, *attempt)));
backups
.into_iter()
.skip(keep_newest.saturating_sub(1))
.map(|(_, _, name)| name.clone())
.collect()
}
pub fn changed_lines(before: &str, after: &str) -> String {
let old: Vec<&str> = before.lines().collect();
let new: Vec<&str> = after.lines().collect();
let common_prefix = old
.iter()
.zip(&new)
.take_while(|(left, right)| left == right)
.count();
let common_suffix = old[common_prefix..]
.iter()
.rev()
.zip(new[common_prefix..].iter().rev())
.take_while(|(left, right)| left == right)
.count();
let removed = &old[common_prefix..old.len() - common_suffix];
let added = &new[common_prefix..new.len() - common_suffix];
removed
.iter()
.map(|line| format!("- {line}"))
.chain(added.iter().map(|line| format!("+ {line}")))
.collect::<Vec<_>>()
.join("\n")
}
pub fn included_files(config: &str) -> Vec<String> {
let Ok(document) = config.parse::<DocumentMut>() else {
return Vec::new();
};
let files_in = |item: Option<&Item>| -> Vec<String> {
item.and_then(|item| item.get("files"))
.and_then(Item::as_array)
.map(|files| {
files
.iter()
.filter_map(Value::as_str)
.map(str::to_owned)
.collect()
})
.unwrap_or_default()
};
let include = document.get("include");
let mut files = files_in(include);
files.extend(files_in(
include.and_then(|include| include.get("optional")),
));
files
}
fn user_config_file() -> Result<PathBuf> {
let path = crate::compositor::user_config_path(
std::env::var("XDG_CONFIG_HOME").ok().as_deref(),
std::env::var("HOME").ok().as_deref(),
)
.ok_or_else(|| {
anyhow!("cannot locate the Umbriel config: neither XDG_CONFIG_HOME nor HOME is set")
})?;
if !path.is_file() {
return Err(anyhow!(
"there is no user config at {}; Umbriel is running on its packaged defaults. Copy \
/usr/share/umbriel/config.toml to {} first, then retry.",
path.display(),
path.display()
));
}
std::fs::canonicalize(&path).with_context(|| format!("cannot resolve {}", path.display()))
}
fn resolve_included(config_file: &Path, listed: &str) -> PathBuf {
if let Some(relative_to_home) = listed.strip_prefix("~/")
&& let Ok(home) = std::env::var("HOME")
{
return PathBuf::from(home).join(relative_to_home);
}
let listed = PathBuf::from(listed);
if listed.is_absolute() {
return listed;
}
config_file
.parent()
.map(|directory| directory.join(&listed))
.unwrap_or(listed)
}
fn binds_chord(config: &str, chord: &str) -> bool {
config
.parse::<DocumentMut>()
.ok()
.and_then(|document| {
document
.get(KEYBINDS_TABLE)
.and_then(Item::as_table_like)
.map(|table| existing_chord(table, chord).is_some())
})
.unwrap_or(false)
}
fn refuse_chords_owned_by_included_files(
config_file: &Path,
config: &str,
chord: &str,
) -> Result<()> {
for listed in included_files(config) {
let included = resolve_included(config_file, &listed);
if let Ok(text) = std::fs::read_to_string(&included)
&& binds_chord(&text, chord)
{
return Err(anyhow!(
"{chord} is bound in the included file {}; editing the main config would leave \
two bindings for it, so nothing was changed. Edit that file instead.",
included.display()
));
}
}
Ok(())
}
fn readable_validation_report(raw: &str, candidate: &Path, config_file: &Path) -> String {
let candidate_name = candidate.display().to_string();
let config_name = config_file.display().to_string();
raw.lines()
.filter(|line| !line.contains('\u{1b}') && !line.trim().is_empty())
.map(|line| line.replace(&candidate_name, &config_name))
.collect::<Vec<_>>()
.join("\n")
}
fn validate(config_file: &Path) -> Result<std::result::Result<(), String>> {
let output = Command::new("umbriel")
.args(["config", "validate", "-c"])
.arg(config_file)
.output()
.map_err(|cause| {
anyhow!("cannot run `umbriel config validate` ({cause}), so the config was not changed")
})?;
let report = String::from_utf8_lossy(&[output.stdout, output.stderr].concat()).into_owned();
Ok(if output.status.success() {
Ok(())
} else {
Err(report)
})
}
fn unix_seconds() -> u64 {
SystemTime::now()
.duration_since(UNIX_EPOCH)
.map(|elapsed| elapsed.as_secs())
.unwrap_or_default()
}
fn sibling(config_file: &Path, suffix: &str) -> PathBuf {
let name = config_file
.file_name()
.map(|name| name.to_string_lossy().into_owned())
.unwrap_or_else(|| "config.toml".to_owned());
config_file.with_file_name(format!("{name}{suffix}"))
}
fn names_beside(config_file: &Path) -> Vec<String> {
config_file
.parent()
.map(|folder| {
std::fs::read_dir(folder)
.into_iter()
.flatten()
.flatten()
.map(|entry| entry.file_name().to_string_lossy().into_owned())
.collect()
})
.unwrap_or_default()
}
fn config_name(config_file: &Path) -> String {
config_file
.file_name()
.map(|name| name.to_string_lossy().into_owned())
.unwrap_or_default()
}
fn write_new_backup(config_file: &Path, original: &str) -> Result<PathBuf> {
use std::io::Write as _;
let stamp = unix_seconds();
let first = next_backup_attempt(&names_beside(config_file), &config_name(config_file), stamp);
for attempt in first.. {
let suffix = if attempt == 1 {
format!(".bak-{stamp}")
} else {
format!(".bak-{stamp}-{attempt}")
};
let backup = sibling(config_file, &suffix);
match std::fs::OpenOptions::new()
.write(true)
.create_new(true)
.open(&backup)
{
Ok(mut file) => {
file.write_all(original.as_bytes())
.with_context(|| format!("cannot write the backup {}", backup.display()))?;
if let Ok(metadata) = std::fs::metadata(config_file) {
let _ = std::fs::set_permissions(&backup, metadata.permissions());
}
return Ok(backup);
}
Err(error) if error.kind() == std::io::ErrorKind::AlreadyExists => continue,
Err(error) => {
return Err(error)
.with_context(|| format!("cannot create the backup {}", backup.display()));
}
}
}
unreachable!("an unused backup name is always found")
}
pub struct PreparedChange {
pub config_file: PathBuf,
original: String,
new_text: String,
pub summary: String,
pub change: String,
pub caution: Option<String>,
always_ask_because: Option<String>,
}
impl PreparedChange {
pub fn asking_every_time_because(mut self, reason: Option<&str>) -> Self {
self.always_ask_because = reason.map(str::to_owned);
self
}
pub fn reason_to_ask_anyway(&self) -> Option<&str> {
self.caution
.as_deref()
.or(self.always_ask_because.as_deref())
}
pub fn preview(&self) -> String {
let caution = self
.caution
.as_deref()
.map(|caution| format!("\nCaution: {caution}"))
.unwrap_or_default();
format!(
"{} in {}\n{}{caution}",
self.summary,
self.config_file.display(),
self.change
)
}
}
fn candidate_path(config_file: &Path) -> PathBuf {
config_file.with_file_name(format!(
".{}.umbriel-mcp-{}",
config_file
.file_name()
.map(|name| name.to_string_lossy().into_owned())
.unwrap_or_default(),
std::process::id()
))
}
fn check_with_umbriel(config_file: &Path, new_text: &str) -> Result<()> {
let candidate = candidate_path(config_file);
std::fs::write(&candidate, new_text)
.with_context(|| format!("cannot write {}", candidate.display()))?;
let verdict = validate(&candidate);
let _ = std::fs::remove_file(&candidate);
let report = match verdict? {
Ok(()) => return Ok(()),
Err(report) => readable_validation_report(&report, &candidate, config_file),
};
let already_invalid = matches!(validate(config_file), Ok(Err(_)));
Err(if already_invalid {
anyhow!(
"{} already fails `umbriel config validate`, so the change could not be checked \
and nothing was written. Fix the existing problems first:\n{report}",
config_file.display()
)
} else {
anyhow!(
"Umbriel rejected the change, so {} was not modified:\n{report}",
config_file.display()
)
})
}
pub fn load_user_config() -> Result<(PathBuf, String)> {
let config_file = user_config_file()?;
let original = std::fs::read_to_string(&config_file)
.with_context(|| format!("cannot read {}", config_file.display()))?;
Ok((config_file, original))
}
pub fn prepare(
config_file: PathBuf,
original: String,
new_text: String,
summary: String,
caution: Option<String>,
) -> Result<Option<PreparedChange>> {
if new_text == original {
return Ok(None);
}
check_with_umbriel(&config_file, &new_text)?;
let change = changed_lines(&original, &new_text);
Ok(Some(PreparedChange {
config_file,
original,
new_text,
summary,
change,
caution,
always_ask_because: None,
}))
}
#[derive(Clone, Copy, Debug)]
pub struct BackupPolicy {
pub keep_backup: bool,
pub keep_newest: usize,
}
fn prune_backups(config_file: &Path, just_written: &Path, keep_newest: usize) -> usize {
let Some(folder) = config_file.parent() else {
return 0;
};
let just_written = just_written
.file_name()
.map(|name| name.to_string_lossy().into_owned())
.unwrap_or_default();
backups_to_remove(
&names_beside(config_file),
&config_name(config_file),
keep_newest,
&just_written,
)
.iter()
.filter(|name| std::fs::remove_file(folder.join(name)).is_ok())
.count()
}
pub fn commit(prepared: &PreparedChange, policy: BackupPolicy) -> Result<String> {
let keep_backup = policy.keep_backup;
let config_file = &prepared.config_file;
let current = std::fs::read_to_string(config_file)
.with_context(|| format!("cannot read {}", config_file.display()))?;
if current != prepared.original {
return Err(anyhow!(
"{} changed after this change was previewed, so nothing was written; \
make the request again to preview it against the current file",
config_file.display()
));
}
let candidate = candidate_path(config_file);
let finish = || -> Result<Option<PathBuf>> {
std::fs::write(&candidate, &prepared.new_text)
.with_context(|| format!("cannot write {}", candidate.display()))?;
let backup = if keep_backup {
Some(write_new_backup(config_file, &prepared.original)?)
} else {
None
};
let permissions = std::fs::metadata(config_file)?.permissions();
std::fs::set_permissions(&candidate, permissions)?;
std::fs::rename(&candidate, config_file)
.with_context(|| format!("cannot replace {}", config_file.display()))?;
Ok(backup)
};
let backup = finish().inspect_err(|_| {
let _ = std::fs::remove_file(&candidate);
})?;
let backup_note = match backup {
Some(backup) => {
let pruned = match prune_backups(config_file, &backup, policy.keep_newest) {
0 => String::new(),
1 => format!(
" Removed 1 older backup, keeping the newest {}.",
policy.keep_newest
),
count => format!(
" Removed {count} older backups, keeping the newest {}.",
policy.keep_newest
),
};
format!(
"Backup of the previous config: {}.{pruned}",
backup.display()
)
}
None => "No backup was kept, as requested.".to_owned(),
};
Ok(format!(
"Applied: {} in {}. Umbriel applies it as soon as the file is saved.\n{backup_note}\n\
Change:\n{}",
prepared.summary,
config_file.display(),
prepared.change
))
}
fn was(previous: Option<&str>) -> String {
previous
.map(|previous| format!(" (was {previous})"))
.unwrap_or_default()
}
pub fn prepare_set_keybind(
chord: &str,
action: &str,
repeat: Option<bool>,
) -> Result<Option<PreparedChange>> {
let (config_file, original) = load_user_config()?;
refuse_chords_owned_by_included_files(&config_file, &original, chord)?;
let edit = plan_set_keybind(&original, chord, action, repeat)?;
let summary = format!(
"Bind {} to {}{}",
edit.chord,
action.trim(),
was(edit.previous.as_deref())
);
prepare(config_file, original, edit.new_text, summary, None)
}
pub fn prepare_remove_keybind(chord: &str) -> Result<PreparedChange> {
let (config_file, original) = load_user_config()?;
refuse_chords_owned_by_included_files(&config_file, &original, chord)?;
let edit = plan_remove_keybind(&original, chord)?;
let summary = format!(
"Remove the binding for {}{}",
edit.chord,
was(edit.previous.as_deref())
);
prepare(config_file, original, edit.new_text, summary, None)?.ok_or_else(|| {
anyhow!("{chord} is not bound in the Umbriel config, so nothing was removed")
})
}
#[cfg(test)]
mod tests {
use super::{
backups_to_remove, changed_lines, included_files, next_backup_attempt, plan_remove_keybind,
plan_set_keybind, same_chord,
};
const CONFIG: &str = r#"# My Umbriel config
[general]
autostart = ["noctalia"]
[keybinds]
# launchers
"Mod+T" = { action = "spawn:kitty", repeat = false }
"Mod+Q" = "window-close" # close
"Mod+Shift+Escape" = { action = "shortcuts-inhibit-toggle", allow_when_inhibited = true }
[layout]
gap = 8
"#;
#[test]
fn chords_match_regardless_of_modifier_order_and_case() {
assert!(same_chord("Mod+Shift+T", "shift+mod+t"));
assert!(same_chord("Mod+Return", "Mod + Return"));
assert!(!same_chord("Mod+T", "Mod+Shift+T"));
assert!(!same_chord("Mod+T", "Mod+Y"));
}
#[test]
fn a_new_keybind_is_added_inside_the_keybinds_table_and_nothing_else_changes() {
let edit = plan_set_keybind(CONFIG, "Mod+F", "window-toggle-fullscreen", None).unwrap();
assert_eq!(edit.previous, None);
assert_eq!(edit.chord, "Mod+F");
assert_eq!(
changed_lines(CONFIG, &edit.new_text),
"+ \"Mod+F\" = \"window-toggle-fullscreen\""
);
let keybinds_at = edit.new_text.find("[keybinds]").unwrap();
let layout_at = edit.new_text.find("[layout]").unwrap();
let added_at = edit.new_text.find("\"Mod+F\"").unwrap();
assert!(keybinds_at < added_at && added_at < layout_at);
assert!(edit.new_text.contains("# launchers"));
assert!(edit.new_text.contains("# close"));
}
#[test]
fn a_repeat_setting_writes_the_table_form() {
let edit =
plan_set_keybind(CONFIG, "Mod+F", "window-toggle-fullscreen", Some(false)).unwrap();
assert!(
edit.new_text
.contains(r#""Mod+F" = { action = "window-toggle-fullscreen", repeat = false }"#),
"{}",
edit.new_text
);
}
#[test]
fn rebinding_an_existing_chord_replaces_it_in_place_and_reports_the_old_binding() {
let edit = plan_set_keybind(CONFIG, "mod+q", "window-kill", None).unwrap();
assert_eq!(edit.chord, "Mod+Q");
assert_eq!(edit.previous.as_deref(), Some("window-close"));
assert_eq!(edit.new_text.matches("Mod+Q").count(), 1);
assert_eq!(
changed_lines(CONFIG, &edit.new_text),
"- \"Mod+Q\" = \"window-close\" # close\n+ \"Mod+Q\" = \"window-kill\" # close"
);
}
#[test]
fn rebinding_a_table_entry_keeps_its_other_settings() {
let edit = plan_set_keybind(CONFIG, "Mod+Shift+Escape", "session-quit", None).unwrap();
assert_eq!(edit.previous.as_deref(), Some("shortcuts-inhibit-toggle"));
assert!(
edit.new_text.contains(
r#""Mod+Shift+Escape" = { action = "session-quit", allow_when_inhibited = true }"#
),
"{}",
edit.new_text
);
}
#[test]
fn a_config_without_keybinds_gets_a_keybinds_table() {
let edit = plan_set_keybind("[layout]\ngap = 8\n", "Mod+T", "spawn:foot", None).unwrap();
assert!(
edit.new_text
.contains("[keybinds]\n\"Mod+T\" = \"spawn:foot\""),
"{}",
edit.new_text
);
assert!(edit.new_text.starts_with("[layout]\ngap = 8\n"));
}
#[test]
fn removing_a_keybind_drops_only_that_line() {
let edit = plan_remove_keybind(CONFIG, "Mod+T").unwrap();
assert_eq!(edit.previous.as_deref(), Some("spawn:kitty"));
assert_eq!(
changed_lines(CONFIG, &edit.new_text),
"- \"Mod+T\" = { action = \"spawn:kitty\", repeat = false }"
);
}
#[test]
fn removing_an_unbound_chord_is_an_error_naming_it() {
let error = plan_remove_keybind(CONFIG, "Mod+Z")
.unwrap_err()
.to_string();
assert!(
error.contains("Mod+Z") && error.contains("not bound"),
"{error}"
);
}
#[test]
fn malformed_input_is_refused() {
assert!(plan_set_keybind("[keybinds\n", "Mod+T", "spawn:x", None).is_err());
assert!(plan_set_keybind(CONFIG, " ", "window-close", None).is_err());
assert!(plan_set_keybind(CONFIG, "Mod+T", " ", None).is_err());
assert!(plan_set_keybind("keybinds = 3\n", "Mod+T", "window-close", None).is_err());
}
#[test]
fn included_files_are_listed_from_both_include_tables() {
let config =
"[include]\nfiles = [\"binds.toml\"]\n\n[include.optional]\nfiles = [\"local.toml\"]\n";
assert_eq!(included_files(config), vec!["binds.toml", "local.toml"]);
assert!(included_files(CONFIG).is_empty());
}
fn names(list: &[&str]) -> Vec<String> {
list.iter().map(|name| (*name).to_owned()).collect()
}
#[test]
fn only_the_newest_backups_are_kept() {
let existing = names(&[
"config.toml.bak-1700000003",
"config.toml.bak-1700000001",
"config.toml.bak-1700000003-2",
"config.toml.bak-1700000002",
"config.toml",
]);
let mut removed =
backups_to_remove(&existing, "config.toml", 2, "config.toml.bak-1700000003-2");
removed.sort();
assert_eq!(
removed,
["config.toml.bak-1700000001", "config.toml.bak-1700000002"]
);
}
#[test]
fn files_umbriel_mcp_did_not_name_are_never_removed() {
let existing = names(&[
"config.toml.bak-1700000001",
"config.toml.bak-1700000002",
"config.toml.bak",
"config.toml.bak-old",
"config.toml.bak-1700000000-x",
"other.toml.bak-1600000000",
"notes.txt",
]);
assert_eq!(
backups_to_remove(&existing, "config.toml", 1, "config.toml.bak-1700000002"),
["config.toml.bak-1700000001"]
);
}
#[test]
fn nothing_is_removed_while_within_the_limit() {
let existing = names(&["config.toml.bak-1", "config.toml.bak-2"]);
assert!(backups_to_remove(&existing, "config.toml", 5, "config.toml.bak-2").is_empty());
}
#[test]
fn the_backup_just_written_is_never_removed() {
let existing = names(&[
"config.toml.bak-1700000000-3",
"config.toml.bak-1700000000-2",
"config.toml.bak-1700000000",
]);
let removed = backups_to_remove(&existing, "config.toml", 2, "config.toml.bak-1700000000");
assert_eq!(removed, ["config.toml.bak-1700000000-2"]);
}
#[test]
fn a_new_backup_numbers_after_every_backup_from_the_same_second() {
let existing = names(&["config.toml.bak-1700000000-3", "config.toml.bak-1699999999"]);
assert_eq!(
next_backup_attempt(&existing, "config.toml", 1_700_000_000),
4
);
assert_eq!(
next_backup_attempt(&existing, "config.toml", 1_700_000_001),
1
);
assert_eq!(
next_backup_attempt(
&names(&["config.toml.bak-1700000000"]),
"config.toml",
1_700_000_000
),
2
);
}
}