From 3c58ffbb367cd49f7c7fa65de8829b20e63ef78a Mon Sep 17 00:00:00 2001
From: Nikkuss <Scuggo@pm.me>
Date: Mon, 14 Apr 2025 13:50:36 +0400
Subject: [PATCH] add model template

---
 templates/model.jinja    | 19 +++++++++++++++++++
 tests/src/models/user.rs | 11 ++++++++++-
 2 files changed, 29 insertions(+), 1 deletion(-)

diff --git a/templates/model.jinja b/templates/model.jinja
index a17eb2d..3f8c541 100644
--- a/templates/model.jinja
+++ b/templates/model.jinja
@@ -1 +1,20 @@
+{% if config.prelude %}
+use {{prelude_path}}::*;
+use sea_orm::ActiveModelBehavior;
+
+#[async_trait::async_trait]
+impl ActiveModelBehavior for {{pascalCase(table_name)}}ActiveModel {}
+
+impl {{pascalCase(table_name)}}Model {}
+
+impl {{pascalCase(table_name)}}ActiveModel {}
+{% else %}
 use {{entities_path}}::{{table_name}}::{ActiveModel, Model, Entity};
+
+#[async_trait::async_trait]
+impl ActiveModelBehavior for ActiveModel {}
+
+impl Model {}
+
+impl ActiveModel {}
+{% endif %}
diff --git a/tests/src/models/user.rs b/tests/src/models/user.rs
index 49f1a64..c7b188f 100644
--- a/tests/src/models/user.rs
+++ b/tests/src/models/user.rs
@@ -1 +1,10 @@
-use super::_entities::user::{ActiveModel, Model, Entity};
\ No newline at end of file
+
+use super::prelude::*;
+use sea_orm::ActiveModelBehavior;
+
+#[async_trait::async_trait]
+impl ActiveModelBehavior for UserActiveModel {}
+
+impl UserModel {}
+
+impl UserActiveModel {}