events-macros: Shorten expand_event_enums through import-rename

This commit is contained in:
Jonas Platte 2021-10-06 20:52:56 +02:00
parent ce8cbff3fd
commit 653f03325b
No known key found for this signature in database
GPG Key ID: CC154DE0E30B7C67

View File

@ -44,6 +44,8 @@ const EVENT_FIELDS: &[(&str, EventKindFn)] = &[
/// Create a content enum from `EventEnumInput`.
pub fn expand_event_enums(input: &EventEnumDecl) -> syn::Result<TokenStream> {
use EventKindVariation as V;
let ruma_events = crate::import_ruma_events();
let name = &input.name;
@ -52,30 +54,13 @@ pub fn expand_event_enums(input: &EventEnumDecl) -> syn::Result<TokenStream> {
let variants: Vec<_> =
input.events.iter().map(EventEnumEntry::to_variant).collect::<syn::Result<_>>()?;
let event_enum =
expand_event_enum(name, &events, attrs, &variants, &EventKindVariation::Full, &ruma_events);
let event_enum = expand_event_enum(name, &events, attrs, &variants, &V::Full, &ruma_events);
let sync_event_enum =
expand_event_enum(name, &events, attrs, &variants, &EventKindVariation::Sync, &ruma_events);
let stripped_event_enum = expand_event_enum(
name,
&events,
attrs,
&variants,
&EventKindVariation::Stripped,
&ruma_events,
);
let initial_event_enum = expand_event_enum(
name,
&events,
attrs,
&variants,
&EventKindVariation::Initial,
&ruma_events,
);
expand_event_enum(name, &events, attrs, &variants, &V::Sync, &ruma_events);
let stripped_event_enum =
expand_event_enum(name, &events, attrs, &variants, &V::Stripped, &ruma_events);
let initial_event_enum =
expand_event_enum(name, &events, attrs, &variants, &V::Initial, &ruma_events);
let redacted_event_enums =
expand_redacted_event_enum(name, &events, attrs, &variants, &ruma_events);
let event_content_enum = expand_content_enum(name, &events, attrs, &variants, &ruma_events);