Add support for additional model field types and arrays

loco-rs · Jan 10, 2025 · 049f500 · 049f500
1 parent abe016e
commit 049f500
Show file tree

Hide file tree

Showing 6 changed files with 407 additions and 59 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -91,7 +91,7 @@ fs-err = "2.11.0"
 # mailer
 tera = "1.19.1"
 thousands = "0.2.0"
-heck = "0.4.0"
+heck = { workspace = true }
 cruet = "0.13.0"
 lettre = { version = "0.11.4", default-features = false, features = [
     "builder",
@@ -179,6 +179,7 @@ tower-http = { version = "0.6.1", features = [
     "set-header",
     "compression-full",
 ] }
+heck = "0.4.0"
 
 [dependencies.sea-orm-migration]
 optional = true

diff --git a/loco-gen/Cargo.toml b/loco-gen/Cargo.toml
@@ -23,6 +23,7 @@ regex = { workspace = true }
 tracing = { workspace = true }
 chrono = { workspace = true }
 colored = { workspace = true }
+heck = { workspace = true }
 
 clap = { version = "4.4.7", features = ["derive"] }
 duct = "0.13"

diff --git a/loco-gen/src/lib.rs b/loco-gen/src/lib.rs
@@ -107,13 +107,6 @@ impl Mappings {
             .map(|f| &f.name)
             .collect::<Vec<_>>()
     }
-    pub fn rust_fields(&self) -> Vec<&String> {
-        self.field_types
-            .iter()
-            .filter(|f| f.rust.is_some())
-            .map(|f| &f.name)
-            .collect::<Vec<_>>()
-    }
 }
 
 static MAPPINGS: OnceLock<Mappings> = OnceLock::new();