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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
use crate::ParseResult;
use proc_macro2::Ident;
use quote::{quote, ToTokens, TokenStreamExt};
use syn::parse::{Parse, ParseStream};
#[derive(Debug, Clone)]
pub struct RustPath(Vec<Ident>);
impl RustPath {
pub fn new_from_ident(id: Ident) -> Self {
Self(vec![id])
}
#[must_use]
pub fn append(&self, id: Ident) -> Self {
Self(self.0.iter().cloned().chain(std::iter::once(id)).collect())
}
pub fn get_final_ident(&self) -> &Ident {
self.0.last().unwrap()
}
}
impl ToTokens for RustPath {
fn to_tokens(&self, tokens: &mut proc_macro2::TokenStream) {
let mut it = self.0.iter();
let mut id = it.next();
while id.is_some() {
id.unwrap().to_tokens(tokens);
let next = it.next();
if next.is_some() {
tokens.append_all(quote! { :: });
}
id = next;
}
}
}
impl Parse for RustPath {
fn parse(input: ParseStream) -> ParseResult<Self> {
let id: Ident = input.parse()?;
let mut p = RustPath::new_from_ident(id);
while input.parse::<Option<syn::token::Colon2>>()?.is_some() {
let id: Ident = input.parse()?;
p = p.append(id);
}
Ok(p)
}
}