use effectful::{
effective::{Canonical, Effective},
environment::{Environment, InEnvironment},
higher_ranked::Rank1,
DynBind,
};
use crate::{
any::type_name,
protocol::{walker::hint::HintMeta, DynVisitor},
Status,
};
use super::VisitResult;
pub trait Recoverable<'src, E: Environment>: DynBind<E> {
fn visit<'r>(
&'r mut self,
scope: DynRecoverableScope<'r, 'src, E>,
) -> Canonical<'r, VisitResult, E>;
}
impl<'u, 'src, E: 'static> type_name::Lower<'u, 'src, &'u &'src ()>
for dyn Recoverable<'static, E>
{
type Lowered = dyn Recoverable<'src, E> + 'u;
}
impl<'u, 'src, E: 'static> type_name::Raise<'u, 'src, &'u &'src ()>
for dyn Recoverable<'src, E> + 'u
{
type Raised = dyn Recoverable<'static, E>;
}
impl<E: Environment> HintMeta for dyn Recoverable<'static, E> {
type Known = Rank1<()>;
type Hint = Rank1<()>;
}
impl<E: Environment> InEnvironment for dyn Recoverable<'static, E> {
type Env = E;
}
pub trait RecoverableScope<'src, E: Environment>: DynBind<E> {
fn new_walk<'r>(&'r mut self, visitor: DynVisitor<'r, 'src, E>) -> Canonical<'r, Status, E>;
}
pub type DynRecoverableScope<'r, 'src, E> = &'r mut dyn RecoverableScope<'src, E>;
pub fn visit_recoverable<'r, 'src, E: Environment>(
visitor: DynVisitor<'r, 'src, E>,
scope: DynRecoverableScope<'r, 'src, E>,
) -> Canonical<'r, VisitResult, E> {
if let Some(object) = visitor
.into_inner()
.as_any_trait_mut()
.upcast_mut::<dyn Recoverable<'src, E> + 'r>()
{
// Allow the visitor to give a hint if it wants.
object.visit(scope)
} else {
// If the visitor doesn't support request hint then we continue.
E::value(VisitResult::Skipped(())).cast()
}
}