drt-sc-derive 0.0.2

Dharitri smart contract API procedural 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
29
30
31
32
33
34
35
36
37

use proc_macro2::Group;

fn token_tree_is_comma(tt: &proc_macro2::TokenTree) -> bool {
    if let proc_macro2::TokenTree::Punct(punct) = &tt {
        punct.as_char() == ','
    } else {
        false
    }
}

fn flush_token_buffer(
    output: &mut Vec<proc_macro2::TokenTree>,
    mut buffer: Vec<proc_macro2::TokenTree>,
) {
    match buffer.len() {
        0 => panic!("empty tokens not allowed in push_format macro"),
        1 => output.append(&mut buffer),
        _ => output.push(proc_macro2::TokenTree::Group(Group::new(
            proc_macro2::Delimiter::Parenthesis,
            buffer.into_iter().collect(),
        ))),
    }
}

pub fn tokenize(input: proc_macro2::TokenStream) -> Vec<proc_macro2::TokenTree> {
    let mut buffer = Vec::new();
    let mut output = Vec::new();
    for tt in input.into_iter() {
        if token_tree_is_comma(&tt) {
            flush_token_buffer(&mut output, core::mem::take(&mut buffer));
        } else {
            buffer.push(tt);
        }
    }
    flush_token_buffer(&mut output, buffer);
    output
}