radn-rs/src/rstd/inject.rs

42 lines
1.3 KiB
Rust

use std::rc::Rc;
use crate::rcore::*;
use crate::rstd::{singular::*, *};
pub trait Inject<'a, Ctx: Context<'a>>: 'a + Sized {
fn inject<A: 'a>(&self, fa: Wrapped<'a, Ctx, A>) -> Wrapped<'a, Ctx, A>;
fn inject_mentionable<A: Mentionable<'a, Ctx>>(
self: Rc<Self>,
a: &A,
) -> ParseResultA<'a, Ctx, A> {
let factory = a.factory();
let inject = self;
let resolver = SingularResolver::from_mentionable(a).into_rc();
let resolver = InjectedResolver { resolver, inject }.into_rc();
let bytes = a.bytes();
factory.parse_slice(&bytes, &resolver)
}
}
struct InjectedResolver<'a, Ctx: Context<'a>, F: Inject<'a, Ctx>> {
resolver: Rc<dyn Resolver<'a, Ctx>>,
inject: Rc<F>,
}
impl<'a, Ctx: Context<'a>, F: Inject<'a, Ctx>> Resolver<'a, Ctx> for InjectedResolver<'a, Ctx, F> {
fn resolve(self: Rc<Self>, address: Address) -> HashResolution<'a, Ctx> {
let inject = self.inject.clone();
self.inject.inject(
self.resolver
.clone()
.resolve_map(address, |resolved| match resolved {
Ok((source, resolver)) => {
Ok((source, InjectedResolver { resolver, inject }.into_rc()))
}
Err(e) => Err(e),
}),
)
}
}