kitchen-freezer/src/utils.rs
2021-04-20 00:22:39 +02:00

113 lines
3.1 KiB
Rust

///! Some utility functions
use std::collections::HashMap;
use std::sync::{Arc, Mutex};
use minidom::Element;
use crate::traits::CompleteCalendar;
use crate::traits::DavCalendar;
use crate::calendar::CalendarId;
use crate::Item;
use crate::item::SyncStatus;
/// Walks an XML tree and returns every element that has the given name
pub fn find_elems<S: AsRef<str>>(root: &Element, searched_name: S) -> Vec<&Element> {
let searched_name = searched_name.as_ref();
let mut elems: Vec<&Element> = Vec::new();
for el in root.children() {
if el.name() == searched_name {
elems.push(el);
} else {
let ret = find_elems(el, searched_name);
elems.extend(ret);
}
}
elems
}
/// Walks an XML tree until it finds an elements with the given name
pub fn find_elem<S: AsRef<str>>(root: &Element, searched_name: S) -> Option<&Element> {
let searched_name = searched_name.as_ref();
if root.name() == searched_name {
return Some(root);
}
for el in root.children() {
if el.name() == searched_name {
return Some(el);
} else {
let ret = find_elem(el, searched_name);
if ret.is_some() {
return ret;
}
}
}
None
}
pub fn print_xml(element: &Element) {
use std::io::Write;
let mut writer = std::io::stdout();
let mut xml_writer = minidom::quick_xml::Writer::new_with_indent(
std::io::stdout(),
0x20, 4
);
let _ = element.to_writer(&mut xml_writer);
let _ = writer.write(&[0x0a]);
}
/// A debug utility that pretty-prints calendars
pub async fn print_calendar_list<C>(cals: &HashMap<CalendarId, Arc<Mutex<C>>>)
where
C: CompleteCalendar,
{
for (id, cal) in cals {
println!("CAL {}", id);
match cal.lock().unwrap().get_items().await {
Err(_err) => continue,
Ok(map) => {
for (_, item) in map {
print_task(item);
}
},
}
}
}
/// A debug utility that pretty-prints calendars
pub async fn print_dav_calendar_list<C>(cals: &HashMap<CalendarId, Arc<Mutex<C>>>)
where
C: DavCalendar,
{
for (id, cal) in cals {
println!("CAL {}", id);
match cal.lock().unwrap().get_item_version_tags().await {
Err(_err) => continue,
Ok(map) => {
for (id, version_tag) in map {
println!(" * {} (version {:?})", id, version_tag);
}
},
}
}
}
pub fn print_task(item: &Item) {
match item {
Item::Task(task) => {
let completion = if task.completed() { "" } else { " " };
let sync = match task.sync_status() {
SyncStatus::NotSynced => ".",
SyncStatus::Synced(_) => "=",
SyncStatus::LocallyModified(_) => "~",
SyncStatus::LocallyDeleted(_) => "x",
};
println!(" {}{} {}\t{}", completion, sync, task.name(), task.id());
},
_ => return,
}
}