Add config

Author: owjs3901

crates/vespertide-config/src/config.rs

@@ -26,6 +26,10 @@ pub struct SeaOrmConfig {
     /// Default: `Camel` (generates `#[serde(rename_all = "camelCase")]`)
     #[serde(default = "default_enum_naming_case")]
     pub enum_naming_case: NameCase,
+    /// Naming case for serde rename_all attribute on generated models.
+    /// Default: `Camel` (generates `#[serde(rename_all = "camelCase")]`)
+    #[serde(default = "default_model_naming_case")]
+    pub model_naming_case: NameCase,
 }
 
 fn default_extra_enum_derives() -> Vec<String> {
@@ -40,12 +44,17 @@ fn default_enum_naming_case() -> NameCase {
     NameCase::Camel
 }
 
+fn default_model_naming_case() -> NameCase {
+    NameCase::Camel
+}
+
 impl Default for SeaOrmConfig {
     fn default() -> Self {
         Self {
             extra_enum_derives: default_extra_enum_derives(),
-            extra_model_derives: Vec::new(),
+            extra_model_derives: default_extra_model_derives(),
             enum_naming_case: default_enum_naming_case(),
+            model_naming_case: default_model_naming_case(),
         }
     }
 }
@@ -65,6 +74,11 @@ impl SeaOrmConfig {
     pub fn enum_naming_case(&self) -> NameCase {
         self.enum_naming_case
     }
+
+    /// Get the naming case for serde rename_all attribute on generated models.
+    pub fn model_naming_case(&self) -> NameCase {
+        self.model_naming_case
+    }
 }
 
 /// Top-level vespertide configuration.
@@ -195,7 +209,11 @@ mod tests {
             config.seaorm.extra_enum_derives,
             vec!["vespera::Schema".to_string()]
         );
-        assert!(config.seaorm.extra_model_derives.is_empty());
+        assert_eq!(
+            config.seaorm.extra_model_derives,
+            vec!["vespera::Schema".to_string()]
+        );
+        assert_eq!(config.seaorm.model_naming_case, NameCase::Camel);
         assert_eq!(config.prefix, "");
     }
 

crates/vespertide-config/src/lib.rs

@@ -41,7 +41,7 @@ mod tests {
     fn seaorm_config_default_has_vespera_schema() {
         let cfg = SeaOrmConfig::default();
         assert_eq!(cfg.extra_enum_derives(), &["vespera::Schema".to_string()]);
-        assert!(cfg.extra_model_derives().is_empty());
+        assert_eq!(cfg.extra_model_derives(), &["vespera::Schema".to_string()]);
     }
 
     #[test]
@@ -70,7 +70,8 @@ mod tests {
         let json = r#"{}"#;
         let cfg: SeaOrmConfig = serde_json::from_str(json).unwrap();
         assert_eq!(cfg.extra_enum_derives(), &["vespera::Schema".to_string()]);
-        assert!(cfg.extra_model_derives().is_empty());
+        assert_eq!(cfg.extra_model_derives(), &["vespera::Schema".to_string()]);
+        assert_eq!(cfg.model_naming_case(), NameCase::Camel);
     }
 
     #[test]

crates/vespertide-exporter/src/seaorm/mod.rs

@@ -103,7 +103,15 @@ pub fn render_entity_with_config(
     }
 
     // Build model derive line with optional extra derives
-    let mut model_derives = vec!["Clone", "Debug", "PartialEq", "Eq", "DeriveEntityModel"];
+    let mut model_derives = vec![
+        "Clone",
+        "Debug",
+        "PartialEq",
+        "Eq",
+        "DeriveEntityModel",
+        "Serialize",
+        "Deserialize",
+    ];
     let extra_model_derives: Vec<&str> = config
         .extra_model_derives()
         .iter()
@@ -120,6 +128,10 @@ pub fn render_entity_with_config(
 
     lines.push("#[sea_orm::model]".into());
     lines.push(format!("#[derive({})]", model_derives.join(", ")));
+    lines.push(format!(
+        "#[serde(rename_all = \"{}\")]",
+        config.model_naming_case().serde_rename_all()
+    ));
     lines.push(format!(
         "#[sea_orm(table_name = \"{}{}\")]",
         prefix, table.name

crates/vespertide-exporter/src/seaorm/snapshots/vespertide_exporter__seaorm__tests__integer_enum_default_value_snapshots@1.snap

@@ -16,6 +16,7 @@ pub enum TaskStatus {
 
 #[sea_orm::model]
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
 #[sea_orm(table_name = "tasks")]
 pub struct Model {
     #[sea_orm(primary_key)]

crates/vespertide-exporter/src/seaorm/snapshots/vespertide_exporter__seaorm__tests__integer_enum_default_value_snapshots@pending_status.snap

@@ -16,6 +16,7 @@ pub enum TaskStatus {
 
 #[sea_orm::model]
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
 #[sea_orm(table_name = "tasks")]
 pub struct Model {
     #[sea_orm(primary_key)]

crates/vespertide-exporter/src/seaorm/snapshots/vespertide_exporter__seaorm__tests__render_entity_snapshots@params_basic_single_pk.snap

@@ -7,6 +7,7 @@ use serde::{Deserialize, Serialize};
 
 #[sea_orm::model]
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
 #[sea_orm(table_name = "users")]
 pub struct Model {
     #[sea_orm(primary_key)]

crates/vespertide-exporter/src/seaorm/snapshots/vespertide_exporter__seaorm__tests__render_entity_snapshots@params_composite_pk.snap

@@ -7,6 +7,7 @@ use serde::{Deserialize, Serialize};
 
 #[sea_orm::model]
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
 #[sea_orm(table_name = "accounts")]
 pub struct Model {
     #[sea_orm(primary_key, auto_increment = false)]

crates/vespertide-exporter/src/seaorm/snapshots/vespertide_exporter__seaorm__tests__render_entity_snapshots@params_enum_multiple_columns.snap

@@ -31,6 +31,7 @@ pub enum AvailabilityStatus {
 
 #[sea_orm::model]
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
 #[sea_orm(table_name = "products")]
 pub struct Model {
     #[sea_orm(primary_key)]

crates/vespertide-exporter/src/seaorm/snapshots/vespertide_exporter__seaorm__tests__render_entity_snapshots@params_enum_nullable.snap

@@ -21,6 +21,7 @@ pub enum TaskPriority {
 
 #[sea_orm::model]
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
 #[sea_orm(table_name = "tasks")]
 pub struct Model {
     #[sea_orm(primary_key)]

crates/vespertide-exporter/src/seaorm/snapshots/vespertide_exporter__seaorm__tests__render_entity_snapshots@params_enum_shared.snap

@@ -19,6 +19,7 @@ pub enum DocStatus {
 
 #[sea_orm::model]
 #[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Serialize, Deserialize)]
+#[serde(rename_all = "camelCase")]
 #[sea_orm(table_name = "documents")]
 pub struct Model {
     #[sea_orm(primary_key)]