81 lines
2.4 KiB
Rust
81 lines
2.4 KiB
Rust
/// The titles alacritty saves before it drops the oldest.
|
|
const SAVED: usize = 4096;
|
|
|
|
/// A program's title as alacritty keeps it, read in the output's order: OSC 0 and 2 set it, CSI 22 t saves it, CSI 23 t brings it back, a full reset empties both.
|
|
#[derive(Default)]
|
|
pub(crate) struct Titles {
|
|
/// Empty while the program names none.
|
|
title: String,
|
|
saved: Vec<String>,
|
|
}
|
|
|
|
impl Titles {
|
|
/// Sets the title an OSC 0 or 2 body past its number names: its UTF-8 parameters joined by `;` and trimmed, as alacritty joins them.
|
|
pub fn set(&mut self, rest: &[u8]) -> String {
|
|
let parts: Vec<&str> = rest
|
|
.split(|&byte| byte == b';')
|
|
.filter_map(|part| std::str::from_utf8(part).ok())
|
|
.collect();
|
|
self.title = parts.join(";").trim().to_owned();
|
|
self.title.clone()
|
|
}
|
|
|
|
/// Saves the title: CSI 22 t.
|
|
pub fn push(&mut self) {
|
|
if self.saved.len() == SAVED {
|
|
self.saved.remove(0);
|
|
}
|
|
self.saved.push(self.title.clone());
|
|
}
|
|
|
|
/// Brings the last title saved back: CSI 23 t; none when none was saved.
|
|
pub fn pop(&mut self) -> Option<String> {
|
|
let title = self.saved.pop()?;
|
|
self.title.clone_from(&title);
|
|
Some(title)
|
|
}
|
|
|
|
/// A full reset empties the title and those saved.
|
|
pub fn reset(&mut self) {
|
|
self.title.clear();
|
|
self.saved.clear();
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn a_title_reads_as_alacritty_joins_it() {
|
|
let mut titles = Titles::default();
|
|
assert_eq!(titles.set(b" a;b "), "a;b");
|
|
assert_eq!(
|
|
titles.set(b"a;\xff;b"),
|
|
"a;b",
|
|
"a part not in UTF-8 passes by"
|
|
);
|
|
assert_eq!(titles.set(b""), "");
|
|
}
|
|
|
|
#[test]
|
|
fn a_title_saved_comes_back_and_a_reset_empties_both() {
|
|
let mut titles = Titles::default();
|
|
assert_eq!(titles.pop(), None, "none saved");
|
|
titles.set(b"shell");
|
|
titles.push();
|
|
titles.set(b"vim");
|
|
assert_eq!(titles.pop().as_deref(), Some("shell"));
|
|
assert_eq!(titles.pop(), None);
|
|
titles.push();
|
|
titles.reset();
|
|
assert_eq!(titles.pop(), None, "a reset empties those saved");
|
|
for at in 0..=SAVED {
|
|
titles.set(at.to_string().as_bytes());
|
|
titles.push();
|
|
}
|
|
assert_eq!(titles.saved.len(), SAVED);
|
|
assert_eq!(titles.saved[0], "1", "the oldest went");
|
|
}
|
|
}
|