count_tts 0.2.0

A simple macro for counting tokens.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

#![no_std]

use proc_macro::{Literal, TokenStream, TokenTree};

/// Expands to the number of token trees in the macro arguments
///
/// Example:
///
/// ```
/// use count_tts::count_tts;
///
/// assert_eq!(count_tts!(), 0);
/// assert_eq!(count_tts!(a b c), 3);
/// assert_eq!(count_tts!(a (b c)), 2);
/// assert_eq!(count_tts!(a, b, c), 5);
/// ```
#[proc_macro]
pub fn count_tts(input: TokenStream) -> TokenStream {
    let count = input.into_iter().count();
    TokenTree::Literal(Literal::usize_unsuffixed(count)).into()
}