protospec-build 0.3.0

One binary format language to rule them all, One binary format language to find them, One binary format language to bring them all and in the darkness bind them.
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
97
98

use super::*;

pub struct Utf8;

impl ForeignType for Utf8 {
    fn assignable_from(&self, type_: &Type) -> bool {
        match type_ {
            Type::Array(inner) => {
                let inner = inner.element.type_.borrow();
                Type::Scalar(ScalarType::U8).assignable_from(&*inner)
            }
            _ => false,
        }
    }

    fn assignable_to(&self, type_: &Type) -> bool {
        self.assignable_from(type_)
    }

    fn type_ref(&self) -> TokenStream {
        quote! { String }
    }

    fn decoding_gen(
        &self,
        source: TokenStream,
        output_ref: TokenStream,
        arguments: Vec<TokenStream>,
        is_async: bool,
    ) -> TokenStream {
        let async_ = map_async(is_async);
        if let Some(len) = arguments.first() {
            quote! {
                let #output_ref = {
                    let t_count = #len as usize;
                    let mut t: Vec<u8> = Vec::with_capacity(t_count);
                    unsafe { t.set_len(t_count); }
                    let t_borrow = &mut t[..];
                    let t_borrow2 = unsafe {
                        let len = t_borrow.len();
                        let ptr = t.as_ptr() as *mut u8;
                        slice::from_raw_parts_mut(ptr, len)
                    };
                    #source.read_exact(&mut t_borrow2[..])#async_?;
                    String::from_utf8(t)?
                };
            }
        } else {
            quote! {
                let #output_ref = {
                    let mut t: Vec<u8> = vec![];
                    #source.read_until(0u8, &mut t)#async_?;
                    if t.len() > 0 && t[t.len() - 1] == 0u8 {
                        t.truncate(t.len() - 1);
                    }
                    String::from_utf8(t)?
                };
            }
        }
    }

    fn encoding_gen(
        &self,
        target: TokenStream,
        field_ref: TokenStream,
        arguments: Vec<TokenStream>,
        is_async: bool,
    ) -> TokenStream {
        let async_ = map_async(is_async);
        if let Some(_) = arguments.first() {
            quote! {
                {
                    #target.write_all(#field_ref.as_bytes())#async_?;
                }
            }
        } else {
            quote! {
                {
                    #target.write_all(#field_ref.as_bytes())#async_?;
                    #target.write_all(&[0u8])#async_?;
                }
            }
        }
    }

    fn arguments(&self) -> Vec<TypeArgument> {
        vec![TypeArgument {
            name: "length".to_string(),
            type_: Type::Scalar(ScalarType::U64),
            default_value: Some(u64::MAX.into()),
            can_resolve_auto: true,
        }]
    }

    fn can_receive_auto(&self) -> Option<ScalarType> {
        None
    }
}