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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85

use crate::Operator;

use super::{Context, RefOperator, Value};

/// An operator for outputting.
pub struct Output<F>(F);

impl<I, O, F> Operator<Value<I>> for Output<F>
where
    F: FnMut(I, &mut Context) -> O,
{
    type Output = Value<O>;

    fn next(&mut self, input: Value<I>) -> Self::Output {
        input.map(&mut self.0)
    }
}

/// Create an operator for outputting from a closure.
pub fn output<I, O>(
    f: impl FnMut(I, &mut Context) -> O,
) -> Output<impl FnMut(I, &mut Context) -> O> {
    Output(f)
}

/// An output operator that insert the output into the context.
pub struct InsertEnvAndOutput<R>(R);

impl<T, R, Out> Operator<Value<T>> for InsertEnvAndOutput<R>
where
    R: for<'a> RefOperator<'a, T, Output = Out>,
    Out: Clone + Send + Sync + 'static,
{
    type Output = Value<Out>;

    #[inline]
    fn next(&mut self, mut input: Value<T>) -> Self::Output {
        let value = self.0.next(input.as_ref());
        input.context_mut().env_mut().insert(value.clone());
        input.map(|_, _| value)
    }
}

/// Create an output operator that insert the output into the `env` context.
pub fn insert_env_and_output<I, O, R, F>(operator: F) -> InsertEnvAndOutput<R>
where
    R: for<'a> RefOperator<'a, I, Output = O>,
    O: Clone + Send + Sync + 'static,
    F: FnOnce() -> R,
{
    InsertEnvAndOutput(operator())
}

/// An output operator that insert the output into the context.
pub struct InsertAndOutput<R>(R);

impl<T, R, Out, Data> Operator<Value<T>> for InsertAndOutput<R>
where
    R: for<'a> RefOperator<'a, T, Output = (Out, Option<Data>)>,
    Out: Clone + Send + Sync + 'static,
    Data: Send + Sync + 'static,
{
    type Output = Value<Out>;

    #[inline]
    fn next(&mut self, mut input: Value<T>) -> Self::Output {
        let (value, data) = self.0.next(input.as_ref());
        input.context_mut().env_mut().insert(value.clone());
        if let Some(data) = data {
            input.context_mut().data_mut().insert(data);
        }
        input.map(|_, _| value)
    }
}

/// Create an output operator that insert the output into the `env` and `data` context.
pub fn insert_and_output<I, O, D, R, F>(operator: F) -> InsertAndOutput<R>
where
    R: for<'a> RefOperator<'a, I, Output = (O, Option<D>)>,
    O: Clone + Send + Sync + 'static,
    D: Send + Sync + 'static,
    F: FnOnce() -> R,
{
    InsertAndOutput(operator())
}