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

//! # Naia Derive
//! Procedural macros to simplify implementation of Naia Replicate &
//! Protocolize traits

#![deny(trivial_casts, trivial_numeric_casts, unstable_features)]

mod channel_index;
mod protocolize;
mod replicate;

use channel_index::channels_impl;
use protocolize::protocolize_impl;
use replicate::replicate_impl;

/// Derives the Protocolize trait for a given enum
#[proc_macro_derive(Protocolize)]
pub fn protocolize_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    protocolize_impl(input)
}

/// Derives the Replicate trait for a given struct
#[proc_macro_derive(Replicate, attributes(protocol_path))]
pub fn replicate_derive(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
    replicate_impl(input)
}

#[proc_macro_attribute]
pub fn derive_channels(
    first_input: proc_macro::TokenStream,
    second_input: proc_macro::TokenStream,
) -> proc_macro::TokenStream {
    channels_impl(first_input, second_input)
}