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        #![allow(unused_imports, dead_code, unused_variables, clippy::all, clippy::pedantic, clippy::nursery)]
21
22        use serde::{Deserialize, Serialize};
23        use ferriorm_runtime::prelude::sqlx;
24
25        #(#enum_defs)*
26    }
27}
28
29fn generate_enum(e: &Enum) -> TokenStream {
30    let name = format_ident!("{}", e.name);
31    let db_name = &e.db_name;
32    let variants: Vec<_> = e.variants.iter().map(|v| format_ident!("{}", v)).collect();
33
34    quote! {
35        #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Serialize, Deserialize, sqlx::Type)]
36        #[sqlx(type_name = #db_name, rename_all = "snake_case")]
37        pub enum #name {
38            #(#variants),*
39        }
40
41        impl std::fmt::Display for #name {
42            fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
43                match self {
44                    #(Self::#variants => write!(f, stringify!(#variants))),*
45                }
46            }
47        }
48    }
49}
ferriorm_codegen/enums.rs

ferriorm_codegen/
enums.rs
