use super::ctl::Ctl;
use super::funcs::next_oid;
use ctl_error::SysctlError;
pub struct CtlIter {
base: Ctl,
current: Ctl,
}
impl CtlIter {
pub fn root() -> Self {
CtlIter {
base: Ctl::Oid(vec![]),
current: Ctl::Oid(vec![1]),
}
}
pub fn below(node: Ctl) -> Self {
CtlIter {
base: node.clone(),
current: node,
}
}
}
impl Iterator for CtlIter {
type Item = Result<Ctl, SysctlError>;
fn next(&mut self) -> Option<Self::Item> {
let oid = match next_oid(self.current.oid()?) {
Ok(Some(o)) => o,
Err(e) => return Some(Err(e)),
Ok(None) => return None,
};
let cont = oid.starts_with(self.base.oid()?);
self.current = Ctl::Oid(oid);
match cont {
true => Some(Ok(self.current.clone())),
false => None,
}
}
}
impl IntoIterator for Ctl {
type Item = Result<Ctl, SysctlError>;
type IntoIter = CtlIter;
fn into_iter(self) -> Self::IntoIter {
CtlIter::below(self)
}
}
#[cfg(test)]
mod tests {
use crate::Sysctl;
#[test]
fn ctl_iter_iterate_all() {
let root = super::CtlIter::root();
let all_ctls: Vec<super::Ctl> = root.into_iter().filter_map(Result::ok).collect();
assert_ne!(all_ctls.len(), 0);
for ctl in &all_ctls {
println!("{:?}", ctl.name());
}
}
}