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

macros: Allow EventContent derive macro to work on enums

Browse Source 
Requires to disable all automatic type generation.
This commit is contained in:
Kévin Commaille 2023-08-24 11:20:32 +02:00 committed by Kévin Commaille
parent e11d30934b
commit a70f99a233
1 changed files with 43 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
crates/ruma-macros/src/events/event_content.rs
View File

@ -297,12 +297,30 @@ pub fn expand_event_content(
let ident = &input.ident; let ident = &input.ident;
let fields = match &input.data { let fields = match &input.data {
syn::Data::Struct(syn::DataStruct { fields, .. }) => fields.iter(), syn::Data::Struct(syn::DataStruct { fields, .. }) => Some(fields.iter()),
_ => { _ => {
return Err(syn::Error::new( if event_kind.is_some_and(|kind| needs_redacted(is_custom_redacted, kind)) {
Span::call_site(), return Err(syn::Error::new(
"event content types need to be structs", Span::call_site(),
)); "To generate a redacted event content, the event content type needs to be a struct. Disable this with the custom_redacted attribute",
));
}
if event_kind.is_some_and(|kind| needs_possibly_redacted(is_custom_redacted, kind)) {
return Err(syn::Error::new(
Span::call_site(),
"To generate a possibly redacted event content, the event content type needs to be a struct. Disable this with the custom_possibly_redacted attribute",
));
}
if has_without_relation {
return Err(syn::Error::new(
Span::call_site(),
"To generate an event content without relation, the event content type needs to be a struct. Disable this by removing the without_relation attribute",
));
}
None
} }
}; };
@ -312,7 +330,7 @@ pub fn expand_event_content(
generate_redacted_event_content( generate_redacted_event_content(
ident, ident,
&input.vis, &input.vis,
fields.clone(), fields.clone().unwrap(),
&event_type, &event_type,
kind, kind,
state_key_type.as_ref(), state_key_type.as_ref(),
@ -330,7 +348,7 @@ pub fn expand_event_content(
generate_possibly_redacted_event_content( generate_possibly_redacted_event_content(
ident, ident,
&input.vis, &input.vis,
fields.clone(), fields.clone().unwrap(),
&event_type, &event_type,
state_key_type.as_ref(), state_key_type.as_ref(),
unsigned_type.clone(), unsigned_type.clone(),
@ -341,8 +359,13 @@ pub fn expand_event_content(
}); });
let event_content_without_relation = has_without_relation.then(|| { let event_content_without_relation = has_without_relation.then(|| {
generate_event_content_without_relation(ident, &input.vis, fields.clone(), ruma_events) generate_event_content_without_relation(
.unwrap_or_else(syn::Error::into_compile_error) ident,
&input.vis,
fields.clone().unwrap(),
ruma_events,
)
.unwrap_or_else(syn::Error::into_compile_error)
}); });
let event_content_impl = generate_event_content_impl( let event_content_impl = generate_event_content_impl(
@ -444,7 +467,7 @@ fn generate_redacted_event_content<'a>(
let redacted_event_content = generate_event_content_impl( let redacted_event_content = generate_event_content_impl(
&redacted_ident, &redacted_ident,
vis, vis,
kept_redacted_fields.iter(), Some(kept_redacted_fields.iter()),
event_type, event_type,
Some(event_kind), Some(event_kind),
EventKindContentVariation::Redacted, EventKindContentVariation::Redacted,
@ -599,7 +622,7 @@ fn generate_possibly_redacted_event_content<'a>(
let possibly_redacted_event_content = generate_event_content_impl( let possibly_redacted_event_content = generate_event_content_impl(
&possibly_redacted_ident, &possibly_redacted_ident,
vis, vis,
possibly_redacted_fields.iter(), Some(possibly_redacted_fields.iter()),
event_type, event_type,
Some(EventKind::State), Some(EventKind::State),
EventKindContentVariation::PossiblyRedacted, EventKindContentVariation::PossiblyRedacted,
@ -787,7 +810,7 @@ impl fmt::Display for EventKindContentVariation {
fn generate_event_content_impl<'a>( fn generate_event_content_impl<'a>(
ident: &Ident, ident: &Ident,
vis: &syn::Visibility, vis: &syn::Visibility,
mut fields: impl Iterator<Item = &'a Field>, mut fields: Option<impl Iterator<Item = &'a Field>>,
event_type: &LitStr, event_type: &LitStr,
event_kind: Option<EventKind>, event_kind: Option<EventKind>,
variation: EventKindContentVariation, variation: EventKindContentVariation,
@ -805,6 +828,13 @@ fn generate_event_content_impl<'a>(
.value() .value()
.strip_suffix('*') .strip_suffix('*')
.map(|type_prefix| { .map(|type_prefix| {
let Some(fields) = &mut fields else {
return Err(syn::Error::new_spanned(
event_type,
"event type with a `.*` suffix is required to be a struct",
));
};
let type_fragment_field = fields let type_fragment_field = fields
.find_map(|f| { .find_map(|f| {
f.attrs.iter().filter(|a| a.path().is_ident("ruma_event")).find_map(|attr| { f.attrs.iter().filter(|a| a.path().is_ident("ruma_event")).find_map(|attr| {
@ -923,6 +953,7 @@ fn generate_event_content_impl<'a>(
[#(#type_prefixes,)*] [#(#type_prefixes,)*]
}; };
let fields_without_type_fragment = fields let fields_without_type_fragment = fields
.unwrap()
.filter(|f| { .filter(|f| {
!f.attrs.iter().any(|a| { !f.attrs.iter().any(|a| {
a.path().is_ident("ruma_event") a.path().is_ident("ruma_event")