94 lines
3.2 KiB
Rust
94 lines
3.2 KiB
Rust
use proc_macro2::TokenStream;
|
|
use proc_macro_crate::{crate_name, FoundCrate};
|
|
use quote::{format_ident, quote, ToTokens};
|
|
use syn::{Field, Ident, LitStr};
|
|
|
|
pub(crate) fn import_ruma_common() -> TokenStream {
|
|
if let Ok(FoundCrate::Name(name)) = crate_name("ruma-common") {
|
|
let import = format_ident!("{name}");
|
|
quote! { ::#import }
|
|
} else if let Ok(FoundCrate::Name(name)) = crate_name("ruma") {
|
|
let import = format_ident!("{name}");
|
|
quote! { ::#import }
|
|
} else if let Ok(FoundCrate::Name(name)) = crate_name("matrix-sdk") {
|
|
let import = format_ident!("{name}");
|
|
quote! { ::#import::ruma }
|
|
} else if let Ok(FoundCrate::Name(name)) = crate_name("matrix-sdk-appservice") {
|
|
let import = format_ident!("{name}");
|
|
quote! { ::#import::ruma }
|
|
} else {
|
|
quote! { ::ruma_common }
|
|
}
|
|
}
|
|
|
|
pub(crate) fn import_ruma_events() -> TokenStream {
|
|
if let Ok(FoundCrate::Name(name)) = crate_name("ruma-events") {
|
|
let import = format_ident!("{name}");
|
|
quote! { ::#import }
|
|
} else if let Ok(FoundCrate::Name(name)) = crate_name("ruma") {
|
|
let import = format_ident!("{name}");
|
|
quote! { ::#import::events }
|
|
} else if let Ok(FoundCrate::Name(name)) = crate_name("matrix-sdk") {
|
|
let import = format_ident!("{name}");
|
|
quote! { ::#import::ruma::events }
|
|
} else if let Ok(FoundCrate::Name(name)) = crate_name("matrix-sdk-appservice") {
|
|
let import = format_ident!("{name}");
|
|
quote! { ::#import::ruma::events }
|
|
} else {
|
|
quote! { ::ruma_events }
|
|
}
|
|
}
|
|
|
|
/// CamelCase's a field ident like "foo_bar" to "FooBar".
|
|
pub(crate) fn to_camel_case(name: &Ident) -> Ident {
|
|
let span = name.span();
|
|
let name = name.to_string();
|
|
|
|
let s: String = name
|
|
.split('_')
|
|
.map(|s| s.chars().next().unwrap().to_uppercase().to_string() + &s[1..])
|
|
.collect();
|
|
Ident::new(&s, span)
|
|
}
|
|
|
|
/// Splits the given string on `.` and `_` removing the `m.` then camel casing to give a Rust type
|
|
/// name.
|
|
pub(crate) fn m_prefix_name_to_type_name(name: &LitStr) -> syn::Result<Ident> {
|
|
let span = name.span();
|
|
let name = name.value();
|
|
|
|
let name = name.strip_prefix("m.").ok_or_else(|| {
|
|
syn::Error::new(
|
|
span,
|
|
format!("well-known matrix events have to start with `m.` found `{name}`"),
|
|
)
|
|
})?;
|
|
|
|
let s: String = name
|
|
.strip_suffix(".*")
|
|
.unwrap_or(name)
|
|
.split(&['.', '_'] as &[char])
|
|
.map(|s| s.chars().next().unwrap().to_uppercase().to_string() + &s[1..])
|
|
.collect();
|
|
|
|
Ok(Ident::new(&s, span))
|
|
}
|
|
|
|
/// Wrapper around [`syn::Field`] that emits the field without its visibility,
|
|
/// thus making it private.
|
|
pub struct PrivateField<'a>(pub &'a Field);
|
|
|
|
impl ToTokens for PrivateField<'_> {
|
|
fn to_tokens(&self, tokens: &mut TokenStream) {
|
|
let Field { attrs, vis: _, mutability, ident, colon_token, ty } = self.0;
|
|
assert_eq!(*mutability, syn::FieldMutability::None);
|
|
|
|
for attr in attrs {
|
|
attr.to_tokens(tokens);
|
|
}
|
|
ident.to_tokens(tokens);
|
|
colon_token.to_tokens(tokens);
|
|
ty.to_tokens(tokens);
|
|
}
|
|
}
|