use super::{
CallableWithSelector,
ImplItem,
ItemImpl,
};
use crate::ir;
pub struct IterConstructors<'a> {
item_impl: &'a ir::ItemImpl,
impl_items: core::slice::Iter<'a, ImplItem>,
}
impl<'a> IterConstructors<'a> {
pub(super) fn new(item_impl: &'a ItemImpl) -> Self {
Self {
item_impl,
impl_items: item_impl.items.iter(),
}
}
}
impl<'a> Iterator for IterConstructors<'a> {
type Item = CallableWithSelector<'a, ir::Constructor>;
fn next(&mut self) -> Option<Self::Item> {
'repeat: loop {
match self.impl_items.next() {
None => return None,
Some(impl_item) => {
if let Some(constructor) = impl_item.filter_map_constructor() {
return Some(CallableWithSelector::new(
self.item_impl,
constructor,
))
}
continue 'repeat
}
}
}
}
}
pub struct IterMessages<'a> {
item_impl: &'a ir::ItemImpl,
impl_items: core::slice::Iter<'a, ImplItem>,
}
impl<'a> IterMessages<'a> {
pub(super) fn new(item_impl: &'a ItemImpl) -> Self {
Self {
item_impl,
impl_items: item_impl.items.iter(),
}
}
}
impl<'a> Iterator for IterMessages<'a> {
type Item = CallableWithSelector<'a, ir::Message>;
fn next(&mut self) -> Option<Self::Item> {
'repeat: loop {
match self.impl_items.next() {
None => return None,
Some(impl_item) => {
if let Some(message) = impl_item.filter_map_message() {
return Some(CallableWithSelector::new(self.item_impl, message))
}
continue 'repeat
}
}
}
}
}