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
|
use log::warn;
/// Emits com.canonical.Unity.LauncherEntry.Update signals via gdbus.
pub struct BadgeEmitter;
impl BadgeEmitter {
/// Emit a badge update for the given desktop_id with the specified count.
pub fn emit(desktop_id: &str, count: usize) {
Self::emit_with(desktop_id, count, crate::dbus_glue::emit_badge);
}
/// Testable version that accepts a function for the actual D-Bus emission.
pub fn emit_with<F>(desktop_id: &str, count: usize, emit_fn: F)
where
F: FnOnce(&str, &str) -> Result<(), String>,
{
let visible = if count > 0 { "true" } else { "false" };
let urgent = visible;
let properties = format!(
"{{'count': <int64 {count}>, 'count-visible': <{visible}>, 'urgent': <{urgent}>}}"
);
if let Err(e) = emit_fn(desktop_id, &properties) {
warn!("badge error: {e}");
}
}
/// Clear badges for a list of desktop_ids (used on shutdown).
pub fn clear_all(desktop_ids: &[String]) {
for id in desktop_ids {
Self::emit(id, 0);
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use std::cell::RefCell;
#[test]
fn test_emit_with_count() {
let captured = RefCell::new(None);
BadgeEmitter::emit_with("application://firefox.desktop", 3, |id, props| {
*captured.borrow_mut() = Some((id.to_string(), props.to_string()));
Ok(())
});
let (id, props) = captured.borrow().clone().unwrap();
assert_eq!(id, "application://firefox.desktop");
assert!(props.contains("count': <int64 3>"));
assert!(props.contains("count-visible': <true>"));
assert!(props.contains("urgent': <true>"));
}
#[test]
fn test_emit_with_zero_count() {
let captured = RefCell::new(None);
BadgeEmitter::emit_with("application://firefox.desktop", 0, |id, props| {
*captured.borrow_mut() = Some((id.to_string(), props.to_string()));
Ok(())
});
let (_, props) = captured.borrow().clone().unwrap();
assert!(props.contains("count': <int64 0>"));
assert!(props.contains("count-visible': <false>"));
assert!(props.contains("urgent': <false>"));
}
#[test]
fn test_emit_error_is_logged_not_panicked() {
// Should not panic on error
BadgeEmitter::emit_with("application://firefox.desktop", 1, |_, _| {
Err("dbus error".into())
});
}
}
|
