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

//! Procedural macros the bern_kernel.
//!
//! This crate provides macros that:
//! - simplify the kernel usage
//! - make kernel development less tedious (used internally)
mod enum_map;

use proc_macro::TokenStream;
use quote::ToTokens;

/// Generates an enum with values and a map to match the enum to another type.
///
/// e.g.
/// ```no_run
/// enum_map!{
///     Size, u8;
///     S128 = 5, 128;
///     S256 = 6, 256;
/// }
/// ```
/// expands to
/// ```no_run
/// #[derive(Copy, Clone, Debug, Eq, PartialEq)]
/// #[repr(u8)]
/// pub enum Size {
///     S128 = 5,
///     S256 = 6,
/// }
/// impl Size {
///     pub const fn bits(self) -> u8 {
///         self as u8
///     }
/// }
///
/// #[macro_export]
/// macro_rules! size_from {
///     (128) => { Size::S128 };
///     (256) => { Size::S256 };
///     ($x:expr) => {
///         compile_error!("Invalid parameter - possible values are: 128, 256");
///     }
/// }
/// ```
/// ```
#[doc(hidden)]
#[proc_macro]
pub fn enum_map(input: TokenStream) -> TokenStream {
    let map = syn::parse_macro_input!(input as enum_map::EnumMap);
    let mut output = proc_macro2::TokenStream::new();
    map.to_tokens(&mut output);
    TokenStream::from(output)
}