36 lines
1.1 KiB
Rust
36 lines
1.1 KiB
Rust
|
use proc_macro::{token_stream::IntoIter, TokenTree};
|
||
|
|
use std::iter::Peekable;
|
||
|
|
|
||
|
|
pub fn skip_pub(mut it: Peekable<IntoIter>) -> Peekable<IntoIter> {
|
||
|
|
if let Some(TokenTree::Ident(ident)) = it.next() {
|
||
|
|
if ident.to_string().as_str() != "pub" {
|
||
|
|
panic!("Expect to find keyword pub but was found {:?}", ident)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
panic!("Expect to find keyword pub but nothing was found")
|
||
|
|
}
|
||
|
|
it
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn skip_enum(mut it: Peekable<IntoIter>) -> Peekable<IntoIter> {
|
||
|
|
if let Some(TokenTree::Ident(ident)) = it.next() {
|
||
|
|
if ident.to_string().as_str() != "enum" {
|
||
|
|
panic!("Expect to find keyword enum but was found {:?}", ident)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
panic!("Expect to find keyword enum but nothing was found")
|
||
|
|
}
|
||
|
|
it
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn skip_struct(mut it: Peekable<IntoIter>) -> Peekable<IntoIter> {
|
||
|
|
if let Some(TokenTree::Ident(ident)) = it.next() {
|
||
|
|
if ident.to_string().as_str() != "struct" {
|
||
|
|
panic!("Expect to find keyword struct but was found {:?}", ident)
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
panic!("Expect to find keyword struct but nothing was found")
|
||
|
|
}
|
||
|
|
it
|
||
|
|
}
|