feat(enterprise): add database migration commands (#158)

- Get migration status by UID
- Export database data
- Import database data with JSON configuration
- Support for stdin input via '-' flag
- Comprehensive mdBook documentation

Author: joshrotenberg

crates/redisctl/src/cli.rs

@@ -1027,6 +1027,10 @@ pub enum EnterpriseCommands {
     #[command(subcommand)]
     Logs(crate::commands::enterprise::logs::LogsCommands),
 
+    /// Migration operations
+    #[command(subcommand)]
+    Migration(crate::commands::enterprise::migration::MigrationCommands),
+
     /// Module management operations
     #[command(subcommand)]
     Module(crate::commands::enterprise::module::ModuleCommands),

crates/redisctl/src/commands/enterprise/migration.rs

@@ -0,0 +1,158 @@
+use anyhow::Context;
+use clap::Subcommand;
+
+use crate::{cli::OutputFormat, connection::ConnectionManager, error::Result as CliResult};
+
+#[allow(dead_code)]
+pub async fn handle_migration_command(
+    conn_mgr: &ConnectionManager,
+    profile_name: Option<&str>,
+    migration_cmd: MigrationCommands,
+    output_format: OutputFormat,
+    query: Option<&str>,
+) -> CliResult<()> {
+    migration_cmd
+        .execute(conn_mgr, profile_name, output_format, query)
+        .await
+}
+
+#[derive(Debug, Clone, Subcommand)]
+pub enum MigrationCommands {
+    /// Get migration status
+    Get {
+        /// Migration UID
+        uid: u64,
+    },
+
+    /// Export database data
+    Export {
+        /// Database UID
+        bdb_uid: u64,
+    },
+
+    /// Import database data
+    Import {
+        /// Database UID
+        bdb_uid: u64,
+        /// Import data (use @filename or - for stdin)
+        #[arg(short, long)]
+        data: String,
+    },
+}
+
+impl MigrationCommands {
+    #[allow(dead_code)]
+    pub async fn execute(
+        &self,
+        conn_mgr: &ConnectionManager,
+        profile_name: Option<&str>,
+        output_format: OutputFormat,
+        query: Option<&str>,
+    ) -> CliResult<()> {
+        handle_migration_command_impl(conn_mgr, profile_name, self, output_format, query).await
+    }
+}
+
+#[allow(dead_code)]
+async fn handle_migration_command_impl(
+    conn_mgr: &ConnectionManager,
+    profile_name: Option<&str>,
+    command: &MigrationCommands,
+    output_format: OutputFormat,
+    query: Option<&str>,
+) -> CliResult<()> {
+    let client = conn_mgr.create_enterprise_client(profile_name).await?;
+
+    match command {
+        MigrationCommands::Get { uid } => {
+            let response: serde_json::Value = client
+                .get(&format!("/v1/migrations/{}", uid))
+                .await
+                .context(format!("Failed to get migration {}", uid))?;
+
+            let output_data = if let Some(q) = query {
+                super::utils::apply_jmespath(&response, q)?
+            } else {
+                response
+            };
+
+            super::utils::print_formatted_output(output_data, output_format)?;
+        }
+        MigrationCommands::Export { bdb_uid } => {
+            let response: serde_json::Value = client
+                .post(
+                    &format!("/v1/bdbs/{}/actions/export", bdb_uid),
+                    &serde_json::json!({}),
+                )
+                .await
+                .context(format!("Failed to export database {}", bdb_uid))?;
+
+            let output_data = if let Some(q) = query {
+                super::utils::apply_jmespath(&response, q)?
+            } else {
+                response
+            };
+
+            super::utils::print_formatted_output(output_data, output_format)?;
+        }
+        MigrationCommands::Import { bdb_uid, data } => {
+            let payload = super::utils::read_json_data(data)?;
+
+            let response: serde_json::Value = client
+                .post(&format!("/v1/bdbs/{}/actions/import", bdb_uid), &payload)
+                .await
+                .context(format!("Failed to import data to database {}", bdb_uid))?;
+
+            let output_data = if let Some(q) = query {
+                super::utils::apply_jmespath(&response, q)?
+            } else {
+                response
+            };
+
+            super::utils::print_formatted_output(output_data, output_format)?;
+        }
+    }
+
+    Ok(())
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_migration_command_parsing() {
+        use clap::Parser;
+
+        #[derive(Parser)]
+        struct TestCli {
+            #[command(subcommand)]
+            cmd: MigrationCommands,
+        }
+
+        // Test get command
+        let cli = TestCli::parse_from(["test", "get", "1"]);
+        if let MigrationCommands::Get { uid } = cli.cmd {
+            assert_eq!(uid, 1);
+        } else {
+            panic!("Expected Get command");
+        }
+
+        // Test export command
+        let cli = TestCli::parse_from(["test", "export", "2"]);
+        if let MigrationCommands::Export { bdb_uid } = cli.cmd {
+            assert_eq!(bdb_uid, 2);
+        } else {
+            panic!("Expected Export command");
+        }
+
+        // Test import command
+        let cli = TestCli::parse_from(["test", "import", "3", "--data", "@import.json"]);
+        if let MigrationCommands::Import { bdb_uid, data } = cli.cmd {
+            assert_eq!(bdb_uid, 3);
+            assert_eq!(data, "@import.json");
+        } else {
+            panic!("Expected Import command");
+        }
+    }
+}

crates/redisctl/src/commands/enterprise/mod.rs

@@ -15,6 +15,7 @@ pub mod endpoint;
 pub mod job_scheduler;
 pub mod logs;
 pub mod logs_impl;
+pub mod migration;
 pub mod module;
 pub mod module_impl;
 pub mod node;

crates/redisctl/src/main.rs

@@ -330,6 +330,16 @@ async fn execute_enterprise_command(
             )
             .await
         }
+        Migration(migration_cmd) => {
+            commands::enterprise::migration::handle_migration_command(
+                conn_mgr,
+                profile,
+                migration_cmd.clone(),
+                output,
+                query,
+            )
+            .await
+        }
         Module(module_cmd) => {
             commands::enterprise::module_impl::handle_module_commands(
                 conn_mgr, profile, module_cmd, output, query,

docs/src/SUMMARY.md

@@ -39,6 +39,7 @@
 - [Usage Reports](./enterprise/usage-report.md)
 - [Modules](./enterprise/modules.md)
 - [Logs](./enterprise/logs.md)
+- [Database Migration](./enterprise/migration.md)
 - [Active-Active (CRDB)](./enterprise/crdb.md)
 - [CRDB Tasks](./enterprise/crdb-tasks.md)
 - [Actions (Tasks)](./enterprise/actions.md)