use core::fmt;
use crate::{Covar, Lend, Lender, Lending, higher_order::FnMutHKAOpt};
#[derive(Clone)]
#[must_use = "lenders are lazy and do nothing unless consumed"]
pub struct MapWhile<L, P> {
pub(crate) lender: L,
pub(crate) predicate: Covar<P>,
}
impl<L, P> MapWhile<L, P> {
#[inline(always)]
pub fn into_inner(self) -> L {
self.lender
}
#[inline(always)]
pub fn into_parts(self) -> (L, Covar<P>) {
(self.lender, self.predicate)
}
}
impl<L: Lender, P> MapWhile<L, P> {
#[inline(always)]
pub(crate) fn new(lender: L, predicate: Covar<P>) -> MapWhile<L, P> {
let _ = L::__check_covariance(crate::CovariantProof::new());
MapWhile { lender, predicate }
}
}
impl<L: fmt::Debug, P> fmt::Debug for MapWhile<L, P> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("MapWhile")
.field("lender", &self.lender)
.finish_non_exhaustive()
}
}
impl<'lend, L, P> Lending<'lend> for MapWhile<L, P>
where
P: for<'all> FnMutHKAOpt<'all, Lend<'all, L>>,
L: Lender,
{
type Lend = <P as FnMutHKAOpt<'lend, Lend<'lend, L>>>::B;
}
impl<L, P> Lender for MapWhile<L, P>
where
P: for<'all> FnMutHKAOpt<'all, Lend<'all, L>>,
L: Lender,
{
crate::unsafe_assume_covariance!();
#[inline]
fn next(&mut self) -> Option<Lend<'_, Self>> {
(self.predicate.as_inner_mut())(self.lender.next()?)
}
#[inline]
fn size_hint(&self) -> (usize, Option<usize>) {
let (_, upper) = self.lender.size_hint();
(0, upper)
}
}