potk-macros 0.1.0

potk db macros
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
28

use proc_macro::TokenStream;
use proc_macro2::TokenStream as TokenStream2;
use quote_into::quote_into;

type IdentList = syn::punctuated::Punctuated<syn::Ident, syn::Token![,]>;

pub fn perms(code: TokenStream) -> TokenStream {
    let ci = crate::crate_ident();
    let idents = syn::parse_macro_input!(code with IdentList::parse_terminated);

    let mut s = TokenStream2::new();
    let mut all = TokenStream2::new();
    let mut count = 0usize;

    for (idx, ident) in idents.iter().enumerate() {
        if ident.to_string().starts_with('_') {
            continue;
        }
        count += 1;
        let (byte, bit) = (idx / 8, (idx % 8) as u8);
        quote_into! {s += pub const #ident: #ci::Perm = (#byte, #bit); };
        quote_into! {all += #ident, };
    }

    quote_into! {s += pub const ALL: [#ci::Perm; #count] = [#all]; };

    s.into()
}