feat: enable refresh & reset through cli

Author: cimandef

vectorctl-cli/src/commands/mod.rs

@@ -51,6 +51,10 @@ pub enum MigrateSubcommands {
         #[arg(long, required = false)]
         to: Option<String>,
     },
+    #[command()]
+    Refresh,
+    #[command()]
+    Reset,
     #[command(about = "Get migration status")]
     Status,
 }
@@ -77,7 +81,9 @@ pub async fn run_migrate_command(
         sub @ Some(MigrateSubcommands::Generate { .. })
         | sub @ Some(MigrateSubcommands::Up { .. })
         | sub @ Some(MigrateSubcommands::Down { .. })
-        | sub @ Some(MigrateSubcommands::Status) => {
+        | sub @ Some(MigrateSubcommands::Status)
+        | sub @ Some(MigrateSubcommands::Refresh)
+        | sub @ Some(MigrateSubcommands::Reset) => {
             let (cmd_str, extra_args) = match sub {
                 Some(MigrateSubcommands::Generate { name, message }) => ("generate", {
                     let mut args = vec![name];
@@ -100,6 +106,8 @@ pub async fn run_migrate_command(
                         .collect(),
                 ),
                 Some(MigrateSubcommands::Status) => ("status", vec![]),
+                Some(MigrateSubcommands::Refresh) => ("refresh", vec![]),
+                Some(MigrateSubcommands::Reset) => ("reset", vec![]),
                 _ => ("up", vec![]),
             };
 

vectorctl-migration/src/cli.rs

@@ -130,6 +130,8 @@ where
         }
         Some(MigrateSubcommands::Up { to }) => M::up(&context, to).await?,
         Some(MigrateSubcommands::Down { to }) => M::down(&context, to).await?,
+        Some(MigrateSubcommands::Refresh) => M::refresh(&context).await?,
+        Some(MigrateSubcommands::Reset) => M::reset(&context).await?,
         Some(MigrateSubcommands::Status) => M::status(&context).await?,
         None => M::up(&context, None).await?,
     }