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

mod ast;
mod from_packet;
mod to_packet;

extern crate proc_macro;

use syn::{parse_macro_input, DeriveInput};

#[proc_macro_derive(FromPacket, attributes(message))]
pub fn derive_from_packet(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let e = parse_macro_input!(input as DeriveInput);
    from_packet::derive(e)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}

#[proc_macro_derive(ToPacket, attributes(message))]
pub fn derive_to_packet(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let e = parse_macro_input!(input as DeriveInput);
    to_packet::derive(e)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}

#[proc_macro_derive(ToHeaders, attributes(header))]
pub fn derive_to_headers(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let e = parse_macro_input!(input as DeriveInput);
    to_packet::headers(e)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}

#[proc_macro_derive(FromHeaders, attributes(header))]
pub fn derive_from_headers(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    let e = parse_macro_input!(input as DeriveInput);
    from_packet::headers(e)
        .unwrap_or_else(|err| err.to_compile_error())
        .into()
}