houseflow-macros 0.1.0

Macros for Houseflow
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

use proc_macro::TokenStream;
use quote::quote;
use syn::ItemFn;

#[proc_macro_attribute]
pub fn decoder(_args: TokenStream, item: TokenStream) -> TokenStream {
    let mut input = syn::parse_macro_input!(item as ItemFn);
    let attrs = &input.attrs;
    let sig = &mut input.sig;
    let vis = &input.vis;
    let body = &input.block;

    (quote! {
        #(#attrs)*
        #vis #sig {
            if buf.remaining() < Self::MIN_SIZE {
                return Err(DecodeError::InvalidSize {
                    expected: Self::MIN_SIZE,
                    received: buf.remaining(),
                });
            } else {
                #body
            }
        }
    })
    .into()
}