chris/ruwuma
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

events-macros: Refactor accessor method codegen

Browse Source
This commit is contained in:
Jonas Platte 2021-10-06 23:02:38 +02:00
parent 4229adfc05
commit c14a0edf15
No known key found for this signature in database
GPG Key ID: CC154DE0E30B7C67
1 changed files with 33 additions and 57 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
crates/ruma-events-macros/src/event_enum.rs
View File

@ -522,23 +522,13 @@ fn accessor_methods(
variants: &[EventEnumVariant],
ruma_events: &TokenStream,
) -> Option<TokenStream> {
use EventKindVariation as V;
let ident = kind.to_event_enum_ident(var)?;
// matching `EventKindVariation`s
if let V::Redacted | V::RedactedSync = var {
return redacted_accessor_methods(kind, var, variants, ruma_events);
}
let ident = kind.to_event_enum_ident(var).unwrap();
let methods = EVENT_FIELDS.iter().map(|(name, has_field)| {
generate_accessor(name, kind, var, *has_field, variants, ruma_events)
});
let content_enum = kind.to_content_enum();
let self_variants: Vec<_> = variants.iter().map(|v| v.match_arm(quote! { Self })).collect();
let content_variants: Vec<_> = variants.iter().map(|v| v.ctor(&content_enum)).collect();
let event_type = quote! {
/// Returns the `type` of this event.
@ -552,21 +542,13 @@ fn accessor_methods(
}
};
let content = quote! {
/// Returns the content enum for this event.
pub fn content(&self) -> #content_enum {
match self {
#( #self_variants(event) => #content_variants(event.content.clone()), )*
Self::_Custom(event) => #content_enum::_Custom {
event_type: #ruma_events::EventContent::event_type(&event.content).to_owned(),
},
}
}
};
let content_accessors = (!var.is_redacted()).then(|| {
let content_enum = kind.to_content_enum();
let content_variants: Vec<_> = variants.iter().map(|v| v.ctor(&content_enum)).collect();
let prev_content = has_prev_content_field(kind, var).then(|| {
quote! {
/// Returns the content enum for this events prev_content.
/// Returns the previous content for this event.
pub fn prev_content(&self) -> Option<#content_enum> {
match self {
#(
@ -584,33 +566,27 @@ fn accessor_methods(
}
});
Some(quote! {
#[automatically_derived]
impl #ident {
#content
#event_type
#prev_content
#( #methods )*
quote! {
/// Returns the content for this event.
pub fn content(&self) -> #content_enum {
match self {
#( #self_variants(event) => #content_variants(event.content.clone()), )*
Self::_Custom(event) => #content_enum::_Custom {
event_type: #ruma_events::EventContent::event_type(&event.content)
.to_owned(),
},
}
})
}
/// Redacted events do NOT generate `content` or `prev_content` methods like
/// un-redacted events; otherwise, they are the same.
fn redacted_accessor_methods(
kind: EventKind,
var: EventKindVariation,
variants: &[EventEnumVariant],
ruma_events: &TokenStream,
) -> Option<TokenStream> {
let ident = kind.to_event_enum_ident(var)?;
let methods = EVENT_FIELDS.iter().map(|(name, has_field)| {
generate_accessor(name, kind, var, *has_field, variants, ruma_events)
#prev_content
}
});
Some(quote! {
#[automatically_derived]
impl #ident {
#event_type
#content_accessors
#( #methods )*
}
})