unipipe 0.1.0

A simple Rust pipe abstraction that extends to iterator and stream.
Documentation 
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
86
87
88
89
90
91
92
93
94
95
96

#[macro_export]
macro_rules! extend_iterator {
    ($visibility:vis $struct_name:ident) => {
        ::paste::paste! {
            $crate::extend_iterator!($visibility $struct_name, [<$struct_name:snake>]);
        }
    };
    ($visibility:vis $struct_name:ident, $method_name:ident) => {
        ::paste::paste! {
            $visibility trait [<$struct_name UniPipeIteratorExt>]:
                Iterator<Item = <$struct_name as $crate::UniPipe>::Input> + Sized
            {
                fn $method_name(
                    mut self,
                ) -> impl Iterator<Item = <$struct_name as $crate::UniPipe>::Output> {
                    let mut pipe = <$struct_name as Default>::default();
                    let mut completed = false;

                    std::iter::from_fn(move || {
                        if completed {
                            return None;
                        }

                        while let Some(input) = self.next() {
                            if let Some(output) = pipe.next(Some(input)) {
                                return Some(output);
                            }
                        }

                        completed = true;

                        if let Some(output) = pipe.next(None) {
                            return Some(output);
                        }

                        None
                    })
                }
            }

            impl<TIterator> [<$struct_name UniPipeIteratorExt>] for TIterator where
                TIterator: Iterator<Item = <$struct_name as $crate::UniPipe>::Input>
            {}
        }
    };
}

#[macro_export]
macro_rules! extend_try_iterator {
    ($visibility:vis $struct_name:ident) => {
        ::paste::paste! {
            $crate::extend_try_iterator!($visibility $struct_name, [<try_ $struct_name:snake>]);
        }
    };
    ($visibility:vis $struct_name:ident, $method_name:ident) => {
        ::paste::paste! {
            $visibility trait [<$struct_name UniPipeIteratorTryExt>]<TError>:
                Iterator<Item = Result<<$struct_name as $crate::UniPipe>::Input, TError>> + Sized
            {
                fn $method_name(mut self) -> impl Iterator<Item = Result<<$struct_name as $crate::UniPipe>::Output, TError>> {
                    let mut pipe = <$struct_name as Default>::default();
                    let mut completed = false;

                    std::iter::from_fn(move || {
                        if completed {
                            return None;
                        }

                        while let Some(input) = self.next() {
                            match input {
                                Ok(input) => {
                                    if let Some(output) = pipe.next(Some(input)) {
                                        return Some(Ok(output));
                                    }
                                }
                                Err(error) => return Some(Err(error)),
                            }
                        }

                        completed = true;

                        if let Some(output) = pipe.next(None) {
                            return Some(Ok(output));
                        }

                        None
                    })
                }
            }

            impl<TIterator, TError> [<$struct_name UniPipeIteratorTryExt>]<TError> for TIterator where
                TIterator: Iterator<Item = Result<<$struct_name as $crate::UniPipe>::Input, TError>>
            {}
        }
    };
}