Add extract-custom & apply-custom emty subcommands

sea-orm-cli/src/cli.rs

@@ -287,6 +287,26 @@ pub enum GenerateSubcommands {
         )]
         seaography: bool,
     },
+    #[command(about = "Extract Custom Entity Code")]
+    ExtractCustom {
+        #[arg(
+            short = 'o',
+            long,
+            default_value = "./",
+            help = "Entity file output directory"
+        )]
+        output_dir: String,
+    },
+    #[command(about = "Apply Custom Entity Code")]
+    ApplyCustom {
+        #[arg(
+            short = 'o',
+            long,
+            default_value = "./",
+            help = "Entity file output directory"
+        )]
+        output_dir: String,
+    },
 }
 
 #[derive(Copy, Clone, Debug, PartialEq, Eq, ValueEnum, Default)]

sea-orm-cli/src/commands/generate.rs

@@ -205,6 +205,18 @@ pub async fn run_generate_command(
 
             println!("... Done.");
         }
+
+        GenerateSubcommands::ExtractCustom {
+            output_dir,
+        } => {
+            println!("ExtractCustom: Output directory = {}",output_dir);
+        }
+
+        GenerateSubcommands::ApplyCustom {
+            output_dir,
+        } => {
+            println!("ApplyCustom: Output directory = {}",output_dir);
+        }
     }
 
     Ok(())
@@ -322,6 +334,8 @@ mod tests {
             "sea-orm-cli",
             "generate",
             "entity",
+            "extract",
+            "apply",
             "--database-url",
             "mysql://root:@localhost:3306/database",
         ]);