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

ferriorm_codegen/
enums.rs
