1use {
2 proc_macro::TokenStream,
3 quote::quote,
4 syn::{parse_macro_input, DeriveInput},
5};
6
7#[proc_macro_derive(IntoTrieIter)]
8pub fn derive_into_trie_iter(input: TokenStream) -> TokenStream {
9 let input = parse_macro_input!(input as DeriveInput);
10 let ident = &input.ident;
11
12 let output = quote! {
13
14 impl<'a> IntoIterator for #ident<'a> {
15 type Item = Vec<usize>;
16 type IntoIter = TrieIteratorWrapper<Self>;
17
18 fn into_iter(self) -> Self::IntoIter {
19 TrieIteratorWrapper::new(self)
20 }
21 }
22
23 };
24
25 output.into()
26}