use std::fmt;
use predicates_core::reflection;
pub trait CaseTreeExt {
fn tree(&self) -> CaseTree;
}
impl<'a> CaseTreeExt for reflection::Case<'a> {
fn tree(&self) -> CaseTree {
CaseTree(convert(self))
}
}
type CaseTreeInner = treeline::Tree<Box<dyn fmt::Display>>;
fn convert(case: &reflection::Case<'_>) -> CaseTreeInner {
let mut leaves: Vec<CaseTreeInner> = vec![];
leaves.extend(case.predicate().iter().flat_map(|pred| {
pred.parameters().map(|item| {
let root: Box<dyn fmt::Display> = Box::new(item.to_string());
treeline::Tree::new(root, vec![])
})
}));
leaves.extend(case.products().map(|item| {
let root: Box<dyn fmt::Display> = Box::new(item.to_string());
treeline::Tree::new(root, vec![])
}));
leaves.extend(case.children().map(|item| convert(item)));
let root = Box::new(case.predicate().map(|p| p.to_string()).unwrap_or_default());
CaseTreeInner::new(root, leaves)
}
#[allow(missing_debug_implementations)]
pub struct CaseTree(CaseTreeInner);
impl fmt::Display for CaseTree {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
self.0.fmt(f)
}
}