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

use super::*;

pub struct Utf16;

impl ForeignType for Utf16 {
    fn assignable_from(&self, type_: &Type) -> bool {
        match type_ {
            Type::Array(inner) => {
                let inner = inner.element.type_.borrow();
                Type::Scalar(ScalarType::U16).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);
        let len = arguments.first().expect("missing len argument");
        quote! {
            let #output_ref = {
                let t_count = #len as usize;
                let mut t: Vec<u16> = 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() * 2;
                    let ptr = t.as_ptr() as *mut u8;
                    slice::from_raw_parts_mut(ptr, len)
                };
                #source.read_exact(&mut t_borrow2[..])#async_?;
                String::from_utf16(&t[..])?
            };
        }
    }

    fn encoding_gen(
        &self,
        target: TokenStream,
        field_ref: TokenStream,
        _arguments: Vec<TokenStream>,
        is_async: bool,
    ) -> TokenStream {
        let async_ = map_async(is_async);
        quote! {
            {
                for utf16 in #field_ref.encode_utf16() {
                    #target.write_all(&utf16.to_be_bytes()[..])#async_?;
                }
            }
        }
    }

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

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