use std::sync::Arc;
use crate::prelude::{Error, PInput, PSuccess, Parser, ParserCore, Span};
#[derive(Clone)]
pub struct PMapWithSpan<P, O1, O2> {
parser: P,
f: Arc<dyn Fn(O1, Span) -> O2>,
}
pub fn pmap_with_span<P, F, O1, O2>(parser: P, f: F) -> PMapWithSpan<P, O1, O2>
where
F: Fn(O1, Span) -> O2 + 'static,
{
PMapWithSpan {
parser,
f: Arc::new(f),
}
}
impl<'a, K, P, O1, O2> ParserCore<'a, K, O2> for PMapWithSpan<P, O1, O2>
where
K: PartialEq + Clone + 'a,
O1: Clone + 'a,
P: Parser<'a, K, O1>,
{
fn parse(&self, i: PInput<'a, K>) -> Result<PSuccess<'a, K, O2>, Error<'a, K>> {
let start = i.loc;
let PSuccess { val, rest } = self.parser.parse(i)?;
let span = (start, rest.loc);
Ok(PSuccess {
val: (self.f)(val, span),
rest,
})
}
}
impl<'a, K, P, O1, O2> Parser<'a, K, O2> for PMapWithSpan<P, O1, O2>
where
K: PartialEq + Clone + 'a,
O1: Clone + 'a,
O2: Clone + 'a,
P: Parser<'a, K, O1>,
{
}