organic/src/context/list.rs

use std::fmt::Debug;

#[derive(Debug, Clone)]
pub struct List<'parent, T> {
    data: T,
    parent: Link<'parent, T>,
}

type Link<'parent, T> = Option<&'parent List<'parent, T>>;

impl<'parent, T> List<'parent, T> {
    pub fn new(first_item: T) -> Self {
        Self {
            data: first_item,
            parent: None,
        }
    }

    pub fn get_data(&self) -> &T {
        &self.data
    }

    pub fn get_parent(&'parent self) -> Link<'parent, T> {
        self.parent
    }

    pub fn iter(&self) -> Iter<'_, T> {
        Iter { next: Some(self) }
    }

    pub fn iter_list(&self) -> IterList<'_, T> {
        IterList { next: Some(self) }
    }

    pub fn push(&'parent self, item: T) -> Self {
        Self {
            data: item,
            parent: Some(self),
        }
    }
}

pub struct Iter<'a, T> {
    next: Link<'a, T>,
}

impl<'a, T> Iterator for Iter<'a, T> {
    type Item = &'a T;

    fn next(&mut self) -> Option<Self::Item> {
        let ret = self.next.map(|link| link.get_data());
        self.next = self.next.map(|link| link.get_parent()).flatten();
        ret
    }
}

pub struct IterList<'a, T> {
    next: Link<'a, T>,
}

impl<'a, T> Iterator for IterList<'a, T> {
    type Item = &'a List<'a, T>;

    fn next(&mut self) -> Option<Self::Item> {
        let ret = self.next;
        self.next = self.next.map(|this| this.get_parent()).flatten();
        ret
    }
}