use crate::*;
pub(crate) fn parse_ident_segment(input: ParseStream) -> syn::Result<String> {
let tt: proc_macro2::TokenTree = input.parse()?;
match tt {
proc_macro2::TokenTree::Ident(ident) => {
let raw: String = ident.to_string();
Ok(raw.strip_prefix("r#").unwrap_or(&raw).to_string())
}
_ => Err(input.error("expected identifier")),
}
}
pub(crate) fn parse_kebab_name(input: ParseStream) -> syn::Result<String> {
if input.peek(LitStr) {
let lit: LitStr = input.parse()?;
return Ok(lit.value());
}
let mut name: String = String::new();
while input.peek(Token![-]) {
input.parse::<Token![-]>()?;
name.push('-');
}
if !input.is_empty() && !input.peek(Token![:]) {
let first: String = parse_ident_segment(input)?;
name.push_str(&first);
}
while input.peek(Token![-]) {
input.parse::<Token![-]>()?;
name.push('-');
let next: String = parse_ident_segment(input)?;
name.push_str(&next);
}
Ok(name)
}
pub(crate) fn reconstruct_kebab_from_tokens(tokens: &proc_macro2::TokenStream) -> String {
let iter: std::iter::Peekable<proc_macro2::token_stream::IntoIter> =
tokens.clone().into_iter().peekable();
let mut result: String = String::new();
for token in iter {
match token {
proc_macro2::TokenTree::Ident(ident) => {
let raw: String = ident.to_string();
let clean: String = raw.strip_prefix("r#").unwrap_or(&raw).to_string();
result.push_str(&clean);
}
proc_macro2::TokenTree::Punct(punct) if punct.as_char() == '-' => {
result.push('-');
}
proc_macro2::TokenTree::Group(group) => {
let inner: String = reconstruct_kebab_from_tokens(&group.stream());
result.push_str(&inner);
}
proc_macro2::TokenTree::Literal(lit) => {
let lit_ts: proc_macro2::TokenStream = proc_macro2::TokenTree::Literal(lit).into();
if let Ok(lit_str) = syn::parse2::<LitStr>(lit_ts) {
result.push_str(&lit_str.value());
}
}
_ => {}
}
}
result
}