1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
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()
    }
}