ferriorm_codegen/
enums.rs

1//! Generates the `enums.rs` module from schema enum definitions.
2//!
3//! Each schema enum becomes a Rust enum with `sqlx::Type`, `Serialize`,
4//! `Deserialize`, and `Display` derives, ready for use in queries and
5//! serialization.
6
7use ferriorm_core::schema::Enum;
8use proc_macro2::TokenStream;
9use quote::{format_ident, quote};
10
11/// Generate the `enums.rs` module containing all enum definitions.
12pub fn generate_enums_module(enums: &[Enum]) -> TokenStream {
13    if enums.is_empty() {
14        return quote! {};
15    }
16
17    let enum_defs: Vec<TokenStream> = enums.iter().map(generate_enum).collect();
18
19    quote! {
20        use serde::{Deserialize, Serialize};
21
22        #(#enum_defs)*
23    }
24}
25
26fn generate_enum(e: &Enum) -> TokenStream {
27    let name = format_ident!("{}", e.name);
28    let db_name = &e.db_name;
29    let variants: Vec<_> = e.variants.iter().map(|v| format_ident!("{}", v)).collect();
30
31    quote! {
32        #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize, sqlx::Type)]
33        #[sqlx(type_name = #db_name, rename_all = "snake_case")]
34        pub enum #name {
35            #(#variants),*
36        }
37
38        impl std::fmt::Display for #name {
39            fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
40                match self {
41                    #(Self::#variants => write!(f, stringify!(#variants))),*
42                }
43            }
44        }
45    }
46}
ferriorm_codegen/enums.rs

ferriorm_codegen/
enums.rs
