feat: Support --column-extra-derives

SeaQL · May 1, 2024 · 6551ebb · 6551ebb
1 parent e664998
commit 6551ebb
Show file tree

Hide file tree

Showing 7 changed files with 325 additions and 3 deletions.
diff --git a/sea-orm-cli/src/cli.rs b/sea-orm-cli/src/cli.rs
@@ -298,6 +298,13 @@ pub enum GenerateSubcommands {
         )]
         enum_extra_attributes: Vec<String>,
 
+        #[arg(
+            long,
+            value_delimiter = ',',
+            help = "Add extra derive macros to generated column enum (comma separated), e.g. `--column-extra-derives 'async_graphql::Enum','CustomDerive'`"
+        )]
+        column_extra_derives: Vec<String>,
+
         #[arg(
             long,
             default_value = "false",

diff --git a/sea-orm-cli/src/commands/generate.rs b/sea-orm-cli/src/commands/generate.rs
@@ -33,6 +33,7 @@ pub async fn run_generate_command(
             model_extra_attributes,
             enum_extra_derives,
             enum_extra_attributes,
+            column_extra_derives,
             seaography,
         } => {
             if verbose {
@@ -187,6 +188,7 @@ pub async fn run_generate_command(
                 model_extra_attributes,
                 enum_extra_derives,
                 enum_extra_attributes,
+                column_extra_derives,
                 seaography,
             );
             let output = EntityTransformer::transform(table_stmts)?.generate(&writer_context);

diff --git a/sea-orm-codegen/src/entity/transformer.rs b/sea-orm-codegen/src/entity/transformer.rs
@@ -386,6 +386,7 @@ mod tests {
                     false,
                     &Default::default(),
                     &Default::default(),
+                    &Default::default(),
                     false,
                 )
                 .into_iter()