chore: upgrade migate-node select to allow use of the box environment.

Author: l-monninger

migration/cli/migrate-node/src/cli/migrate/select.rs

@@ -3,6 +3,7 @@ use clap::Parser;
 use mtma_box_environment::Config as BoxEnvironmentConfig;
 use mtma_environment_types::Environmentish;
 use mtma_migrator_types::migration::Migrationish;
+use mtma_migrator_types::migrator::MovementMigrator;
 use mtma_node_null_core::Config as NullConfig;
 use mtma_provisioner_environment::Config as ProvisionerEnvironmentConfig;
 use mtma_testing_environment::Config as TestingEnvironmentConfig;
@@ -19,12 +20,12 @@ pub struct Select {
 }
 
 impl select::Select {
-	pub async fn execute(&self) -> Result<(), anyhow::Error> {
+	pub async fn get_movement_migrator(&self) -> Result<MovementMigrator, anyhow::Error> {
 		let (
 			maybe_environment_testing,
 			maybe_environment_box,
 			maybe_environment_provisioner,
-			maybe_null,
+			_maybe_null,
 		) = self.select().map_err(|e| anyhow::anyhow!("{}", e))?;
 
 		// if more than one environment is provided, we need to error
@@ -38,11 +39,51 @@ impl select::Select {
 			));
 		}
 
-		let environment_config = maybe_environment_testing.context(
-			"--environment-testing is the only supported environment; it must be provided",
-		)?;
+		match maybe_environment_testing {
+			Some(environment_testing) => {
+				let environment_config = environment_testing.build()?;
+				return environment_config
+					.build_movement_migrator()
+					.await
+					.map_err(|e| anyhow::anyhow!("{}", e));
+			}
+			None => {}
+		}
+
+		match maybe_environment_box {
+			Some(environment_box) => {
+				let environment_config = environment_box.build()?;
+				return environment_config
+					.build_movement_migrator()
+					.await
+					.map_err(|e| anyhow::anyhow!("{}", e));
+			}
+			None => {}
+		}
+
+		match maybe_environment_provisioner {
+			Some(environment_provisioner) => {
+				let environment_config = environment_provisioner.build()?;
+				return environment_config
+					.build_movement_migrator()
+					.await
+					.map_err(|e| anyhow::anyhow!("{}", e));
+			}
+			None => {}
+		}
+
+		return Err(anyhow::anyhow!("no environment provided"));
+	}
+
+	pub async fn execute(&self) -> Result<(), anyhow::Error> {
+		let (
+			_maybe_environment_testing,
+			_maybe_environment_box,
+			_maybe_environment_provisioner,
+			maybe_null,
+		) = self.select().map_err(|e| anyhow::anyhow!("{}", e))?;
 
-		let movement_migrator = environment_config.build()?.build_movement_migrator().await?;
+		let movement_migrator = self.get_movement_migrator().await?;
 
 		if let Some(null) = maybe_null {
 			let null_migration = null.build()?;