[linux-block.git] / rust / macros / concat_idents.rs

// SPDX-License-Identifier: GPL-2.0

use proc_macro::{token_stream, Ident, TokenStream, TokenTree};

use crate::helpers::expect_punct;

fn expect_ident(it: &mut token_stream::IntoIter) -> Ident {
    if let Some(TokenTree::Ident(ident)) = it.next() {
        ident
    } else {
        panic!("Expected Ident")
    }
}

pub(crate) fn concat_idents(ts: TokenStream) -> TokenStream {
    let mut it = ts.into_iter();
    let a = expect_ident(&mut it);
    assert_eq!(expect_punct(&mut it), ',');
    let b = expect_ident(&mut it);
    assert!(it.next().is_none(), "only two idents can be concatenated");
    let res = Ident::new(&format!("{a}{b}"), b.span());
    TokenStream::from_iter([TokenTree::Ident(res)])
}
Commit	Line	Data
60f18c22 BRB	1	// SPDX-License-Identifier: GPL-2.0
	2
	3	use proc_macro::{token_stream, Ident, TokenStream, TokenTree};
	4
	5	use crate::helpers::expect_punct;
	6
	7	fn expect_ident(it: &mut token_stream::IntoIter) -> Ident {
	8	if let Some(TokenTree::Ident(ident)) = it.next() {
	9	ident
	10	} else {
	11	panic!("Expected Ident")
	12	}
	13	}
	14
	15	pub(crate) fn concat_idents(ts: TokenStream) -> TokenStream {
	16	let mut it = ts.into_iter();
	17	let a = expect_ident(&mut it);
	18	assert_eq!(expect_punct(&mut it), ',');
	19	let b = expect_ident(&mut it);
	20	assert!(it.next().is_none(), "only two idents can be concatenated");
	21	let res = Ident::new(&format!("{a}{b}"), b.span());
	22	TokenStream::from_iter([TokenTree::Ident(res)])
	23	}