le-stream-derive 1.3.5

Derive macros to de-/serialize objects from/to little endian byte streams
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

use proc_macro2::TokenStream;
use quote::{quote, ToTokens};
use syn::Type;

use super::extend::Extend;

#[derive(Debug)]
pub struct IteratorTokens {
    statement: TokenStream,
    typ: TokenStream,
}

impl Extend for IteratorTokens {
    fn extend<T>(&mut self, field: T, typ: &Type)
    where
        T: ToTokens,
    {
        self.statement
            .extend(quote! { .chain(<#typ as ::le_stream::ToLeStream>::to_le_stream(#field)) });
        let current_type = &self.typ;
        self.typ =
            quote! { ::core::iter::Chain<#current_type, <#typ as ::le_stream::ToLeStream>::Iter> };
    }
}

impl Default for IteratorTokens {
    fn default() -> Self {
        Self {
            statement: quote! { ::core::iter::empty() },
            typ: quote! { ::core::iter::Empty<u8> },
        }
    }
}

impl From<IteratorTokens> for (TokenStream, TokenStream) {
    fn from(iterator: IteratorTokens) -> Self {
        (iterator.statement, iterator.typ)
    }
}

impl Extend for Option<IteratorTokens> {
    fn extend<T>(&mut self, field: T, typ: &Type)
    where
        T: ToTokens,
    {
        if let Some(iterator) = self {
            iterator.extend(field, typ);
        } else {
            self.replace(IteratorTokens {
                statement: quote! { <#typ as ::le_stream::ToLeStream >::to_le_stream(#field) },
                typ: quote! { <#typ as ::le_stream::ToLeStream >::Iter },
            });
        }
    }
}