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

#![doc = include_str!("../README.md")]

use arguments::MacroArgs;
use proc_macro::TokenStream;
use proc_macro_error::proc_macro_error;

mod argparser;
mod arguments;

mod tag_counter;
mod utils;

mod general;

/// Strict version of the [`IntegralEnum`], see module-level
/// documentation for more detailed explaination.
#[proc_macro_derive(StrictIntegralEnum, attributes(enum_impl))]
#[proc_macro_error]
pub fn define_strict_integral_enum(body: TokenStream) -> TokenStream {
    let input = syn::parse_macro_input!(body as syn::DeriveInput);
    general::impl_macro(
        "enum_impl",
        true,
        MacroArgs::new_implied(false),
        input,
    )
}

/// Macro that implements bunch of traits. See module-level
/// documentation for more detailed explaination.
#[proc_macro_derive(IntegralEnum, attributes(enum_disable))]
#[proc_macro_error]
pub fn define_integral_enum(body: TokenStream) -> TokenStream {
    let input = syn::parse_macro_input!(body as syn::DeriveInput);
    general::impl_macro(
        "enum_disable",
        false,
        MacroArgs::new_implied(true),
        input,
    )
}