refactor: Formatted files using rustfmt

This commit is contained in:
2025-03-06 15:32:59 -07:00
parent f97d46cec3
commit bdad723aef
3 changed files with 26 additions and 22 deletions
@@ -3,32 +3,33 @@ mod macro_models;
use crate::macro_models::DisplayStyleArgs; use crate::macro_models::DisplayStyleArgs;
use darling::FromVariant; use darling::FromVariant;
use quote::quote; use quote::quote;
use syn::{parse_macro_input, Data, DeriveInput}; use syn::{Data, DeriveInput, parse_macro_input};
/// Derive macro for the EnumDisplayStyle trait. /// Derive macro for the EnumDisplayStyle trait.
/// ///
/// # Example /// # Example
/// ///
/// Using default values for the display style: /// Using default values for the display style:
/// ///
/// ``` /// ```
/// use enum_display_style_derive::EnumDisplayStyle; /// use enum_display_style_derive::EnumDisplayStyle;
/// ///
/// #[derive(EnumDisplayStyle)] /// #[derive(EnumDisplayStyle)]
/// enum Weekend { /// enum Weekend {
/// Saturday, /// Saturday,
/// Sunday, /// Sunday,
/// } /// }
/// ///
/// assert_eq!(Weekend::Saturday.to_display_str(), "Saturday"); /// assert_eq!(Weekend::Saturday.to_display_str(), "Saturday");
/// assert_eq!(Weekend::Sunday.to_display_str(), "Sunday"); /// assert_eq!(Weekend::Sunday.to_display_str(), "Sunday");
/// ///
/// ``` /// ```
/// ///
/// Using custom values for the display style: /// Using custom values for the display style:
/// ///
/// ``` /// ```
/// use enum_display_style_derive::EnumDisplayStyle; /// use enum_display_style_derive::EnumDisplayStyle;
/// ///
/// #[derive(EnumDisplayStyle)] /// #[derive(EnumDisplayStyle)]
/// enum MonitorStatus { /// enum MonitorStatus {
/// #[display_style(name = "Monitor Transactions")] /// #[display_style(name = "Monitor Transactions")]
@@ -36,7 +37,7 @@ use syn::{parse_macro_input, Data, DeriveInput};
/// #[display_style(name = "Don't Monitor Transactions")] /// #[display_style(name = "Don't Monitor Transactions")]
/// None, /// None,
/// } /// }
/// ///
/// assert_eq!(MonitorStatus::Active.to_display_str(), "Monitor Transactions"); /// assert_eq!(MonitorStatus::Active.to_display_str(), "Monitor Transactions");
/// assert_eq!(MonitorStatus::None.to_display_str(), "Don't Monitor Transactions"); /// assert_eq!(MonitorStatus::None.to_display_str(), "Don't Monitor Transactions");
/// ``` /// ```
@@ -52,8 +53,11 @@ pub fn enum_display_style_derive(input: proc_macro::TokenStream) -> proc_macro::
for variant in variants { for variant in variants {
let variant_ident = &variant.ident; let variant_ident = &variant.ident;
let variant_display_name = DisplayStyleArgs::from_variant(variant).unwrap().name.unwrap_or_else(|| variant_ident.to_string()); let variant_display_name = DisplayStyleArgs::from_variant(variant)
.unwrap()
.name
.unwrap_or_else(|| variant_ident.to_string());
match_arms.push(quote! { match_arms.push(quote! {
#enum_name::#variant_ident => #variant_display_name, #enum_name::#variant_ident => #variant_display_name,
}); });
@@ -70,4 +74,4 @@ pub fn enum_display_style_derive(input: proc_macro::TokenStream) -> proc_macro::
} }
} }
.into() .into()
} }
@@ -3,5 +3,5 @@ use darling::FromVariant;
#[derive(Debug, FromVariant)] #[derive(Debug, FromVariant)]
#[darling(attributes(display_style))] #[darling(attributes(display_style))]
pub struct DisplayStyleArgs { pub struct DisplayStyleArgs {
pub name: Option<String> pub name: Option<String>,
} }
+6 -6
View File
@@ -3,13 +3,13 @@ use pretty_assertions::assert_str_eq;
#[test] #[test]
fn test_derive_enum_display_style() { fn test_derive_enum_display_style() {
assert_str_eq!(TestEnum::Test.to_display_str(), "Testing 123"); assert_str_eq!(TestEnum::Test.to_display_str(), "Testing 123");
assert_str_eq!(TestEnum::Ignored.to_display_str(), "Ignored"); assert_str_eq!(TestEnum::Ignored.to_display_str(), "Ignored");
} }
#[derive(EnumDisplayStyle)] #[derive(EnumDisplayStyle)]
pub enum TestEnum { pub enum TestEnum {
#[display_style(name = "Testing 123")] #[display_style(name = "Testing 123")]
Test, Test,
Ignored, Ignored,
} }