use core::fmt;
use crate::{try_trait_v2::Try, DoubleEndedLender, ExactSizeLender, FusedLender, Lend, Lender, Lending};
#[derive(Clone)]
#[must_use = "lenders are lazy and do nothing unless consumed"]
pub struct Inspect<L, F> {
lender: L,
f: F,
}
impl<L, F> Inspect<L, F> {
pub(crate) fn new(lender: L, f: F) -> Inspect<L, F> {
Inspect { lender, f }
}
}
impl<I: fmt::Debug, F> fmt::Debug for Inspect<I, F> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("Inspect").field("lender", &self.lender).finish()
}
}
impl<'lend, L, F> Lending<'lend> for Inspect<L, F>
where
L: Lender,
F: FnMut(&Lend<'lend, L>),
{
type Lend = Lend<'lend, L>;
}
impl<L, F> Lender for Inspect<L, F>
where
L: Lender,
F: FnMut(&Lend<'_, L>),
{
#[inline]
fn next(&mut self) -> Option<Lend<'_, Self>> {
let next = self.lender.next();
if let Some(ref x) = next {
(self.f)(x);
}
next
}
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
self.lender.size_hint()
}
#[inline]
fn try_fold<B, Fold, R>(&mut self, init: B, mut fold: Fold) -> R
where
Self: Sized,
Fold: FnMut(B, Lend<'_, Self>) -> R,
R: Try<Output = B>,
{
let f = &mut self.f;
self.lender.try_fold(init, move |acc, x| {
(f)(&x);
fold(acc, x)
})
}
#[inline]
fn fold<B, Fold>(mut self, init: B, mut fold: Fold) -> B
where
Self: Sized,
Fold: FnMut(B, Lend<'_, Self>) -> B,
{
self.lender.fold(init, move |acc, x| {
(self.f)(&x);
fold(acc, x)
})
}
}
impl<L, F> DoubleEndedLender for Inspect<L, F>
where
L: DoubleEndedLender,
F: FnMut(&Lend<'_, L>),
{
#[inline]
fn next_back(&mut self) -> Option<Lend<'_, Self>> {
let next = self.lender.next_back();
if let Some(ref x) = next {
(self.f)(x);
}
next
}
#[inline]
fn try_rfold<B, Fold, R>(&mut self, init: B, mut fold: Fold) -> R
where
Self: Sized,
Fold: FnMut(B, Lend<'_, Self>) -> R,
R: Try<Output = B>,
{
let f = &mut self.f;
self.lender.try_rfold(init, move |acc, x| {
(f)(&x);
fold(acc, x)
})
}
#[inline]
fn rfold<B, Fold>(mut self, init: B, mut fold: Fold) -> B
where
Self: Sized,
Fold: FnMut(B, Lend<'_, Self>) -> B,
{
self.lender.rfold(init, move |acc, x| {
(self.f)(&x);
fold(acc, x)
})
}
}
impl<L: ExactSizeLender, F> ExactSizeLender for Inspect<L, F>
where
F: FnMut(&Lend<'_, L>),
{
#[inline]
fn len(&self) -> usize {
self.lender.len()
}
#[inline]
fn is_empty(&self) -> bool {
self.lender.is_empty()
}
}
impl<L: FusedLender, F> FusedLender for Inspect<L, F> where F: FnMut(&Lend<'_, L>) {}