feat: box environment

Cargo.toml

@@ -31,7 +31,8 @@ members = [
     "util/movement/*",
     "util/movement-aptos/*",
     "util/bcs-ext",
-    "util/types"
+    "util/types",
+    "util/util"
 
 ]
 
@@ -149,8 +150,8 @@ include-vendor = { git = "https://github.com/movementlabsxyz/kestrel.git", rev =
 ready-docker = { git = "https://github.com/movementlabsxyz/kestrel.git", rev = "3220d704df7e06d1dcc5266e15eaf05db86fdb07" }
 
 # orfile
-orfile = { git = "https://github.com/movementlabsxyz/orfile.git", rev = "f02851242af77791b905efc19aef6af21918fb1e" }
-slect = { git = "https://github.com/movementlabsxyz/orfile.git", rev = "f02851242af77791b905efc19aef6af21918fb1e" }
+orfile = { git = "https://github.com/movementlabsxyz/orfile.git", rev = "e3279f2fe4a9bafe1197e4d4f1c769d7255b94ac" }
+select = { git = "https://github.com/movementlabsxyz/orfile.git", rev = "e3279f2fe4a9bafe1197e4d4f1c769d7255b94ac" }
 
 # docs
 clap-markdown-ext = { git = "https://github.com/movementlabsxyz/clap-markdown-ext.git", rev = "a4c6c00193baa50eefa214bd74b4a1d50c3255a0" }
@@ -215,6 +216,7 @@ movement-aptos = { path = "util/movement-aptos/movement-aptos" }
 movement-aptos-core = { path = "util/movement-aptos/core" }
 movement-aptos-core-util = { path = "util/movement-aptos/core-util" }
 mtma-types = { path = "util/types" }
+mtma-util = { path = "util/util" }
 
 ## Environments
 ### util

checks/migrator/checks/pre-l1-merge/src/pre_l1_merge.rs

@@ -31,7 +31,7 @@ pub mod test {
 			// Create a Movement instance with the config
 			let movement = Movement::new(
 				movement_config.clone(),
-				movement_core::MovementWorkspace::try_temp()?,
+				movement_core::MovementWorkspace::try_temp()?.into(),
 				Overlays::default(),
 				true,
 				true,

environments/core/box/Cargo.toml

@@ -15,6 +15,8 @@ clap = { workspace = true }
 thiserror = { workspace = true }
 mtma-migrator-types = { workspace = true }
 mtma-environment-types = { workspace = true }
+kestrel = { workspace = true }
+tokio = { workspace = true }
 
 [lints]
 workspace = true

environments/core/box/src/environment/config.rs

@@ -2,6 +2,7 @@ use crate::BoxEnvironment;
 use clap::Parser;
 use serde::{Deserialize, Serialize};
 use std::fmt::Debug;
+use std::path::PathBuf;
 
 /// Errors thrown when working with the [Config].
 #[derive(Debug, thiserror::Error)]
@@ -16,13 +17,27 @@ pub enum EnvironmentConfigError {
 /// This is the frontend for the core API.
 #[derive(Parser, Debug, Serialize, Deserialize, Clone, Default)]
 #[clap(help_expected = true)]
-pub struct Config {}
+pub struct Config {
+	/// The rest api url of the box environment.
+	#[clap(long)]
+	pub rest_api_url: String,
+	/// The db dir of the box environment.
+	#[clap(long)]
+	pub db_dir: PathBuf,
+	/// Whether to isolate the box environment by snapshotting the movement runner and where to store the snapshot.
+	#[clap(long)]
+	pub snapshot_dir: Option<PathBuf>,
+}
 
 impl Config {
 	/// Builds the [BoxEnvironment] struct from the config.
 	///
 	/// Note: preserving faillibility here because we may add debug path configuration to the [BoxEnvironment] soon.
 	pub fn build(&self) -> Result<BoxEnvironment, EnvironmentConfigError> {
-		Ok(BoxEnvironment::new())
+		Ok(BoxEnvironment::new(
+			self.rest_api_url.clone(),
+			self.db_dir.clone(),
+			self.snapshot_dir.clone(),
+		))
 	}
 }

environments/core/box/src/environment/environment.rs

@@ -1,6 +1,11 @@
+use anyhow::Context;
 use mtma_environment_types::{EnvironmentError, Environmentish};
-use mtma_migrator_types::migrator::MovementMigrator;
+use mtma_migrator_types::migrator::{
+	movement_migrator::{LiveMigrator, Runner},
+	MovementMigrator,
+};
 use std::fmt::Debug;
+use std::path::PathBuf;
 
 /// Errors thrown when using the [BoxEnvironment].
 #[derive(Debug, thiserror::Error)]
@@ -15,16 +20,55 @@ impl From<BoxEnvironmentError> for EnvironmentError {
 	}
 }
 
-pub struct BoxEnvironment {}
+pub struct BoxEnvironment {
+	/// The rest api url of the box environment.
+	pub rest_api_url: String,
+	/// The db dir of the box environment.
+	pub db_dir: PathBuf,
+	/// Whether to isolate the box environment by snapshotting the movement runner and where to store the snapshot.
+	pub snapshot_dir: Option<PathBuf>,
+}
 
 impl BoxEnvironment {
-	pub fn new() -> Self {
-		Self {}
+	pub fn new(rest_api_url: String, db_dir: PathBuf, snapshot_dir: Option<PathBuf>) -> Self {
+		Self { rest_api_url, db_dir, snapshot_dir }
 	}
 }
 
 impl Environmentish for BoxEnvironment {
 	async fn build_movement_migrator(&self) -> Result<MovementMigrator, EnvironmentError> {
-		Err(BoxEnvironmentError::Internal(anyhow::anyhow!("not implemented").into()).into())
+		// construct the live migrator
+		let movement_migrator = MovementMigrator::new(Runner::Live(LiveMigrator::new(
+			self.rest_api_url.clone(),
+			self.db_dir.clone(),
+		)));
+
+		// make sure the live migrator is ready
+		let movement_migrator_for_task = movement_migrator.clone();
+		let migrator_task = kestrel::task(async move { movement_migrator_for_task.run().await });
+
+		// wait for the rest api to be ready
+		let _rest_api_url = movement_migrator
+			.wait_for_rest_api_url(tokio::time::Duration::from_secs(300))
+			.await
+			.context("failed to wait for the rest api to be ready")
+			.map_err(|e| BoxEnvironmentError::Internal(e.into()))?;
+
+		// end the migrator task
+		kestrel::end!(migrator_task)
+			.context("failed to end the migrator task")
+			.map_err(|e| BoxEnvironmentError::Internal(e.into()))?;
+
+		// if we're snapshotting the movement_migrator should be a snapshot of the live migrator
+		let movement_migrator = match &self.snapshot_dir {
+			Some(snapshot_dir) => movement_migrator
+				.snapshot(snapshot_dir.clone())
+				.await
+				.context("failed to snapshot the movement migrator")
+				.map_err(|e| BoxEnvironmentError::Internal(e.into()))?,
+			None => movement_migrator,
+		};
+
+		Ok(movement_migrator)
 	}
 }

migration/cli/migrate-node/Cargo.toml

@@ -21,7 +21,7 @@ jemallocator = { workspace = true }
 mtma-node-null-core = { workspace = true }
 tracing = { workspace = true }
 tracing-subscriber = { workspace = true }
-slect = { workspace = true }
+select = { workspace = true }
 mtma-testing-environment = { workspace = true }
 mtma-box-environment = { workspace = true }
 mtma-provisioner-environment = { workspace = true }

migration/cli/migrate-node/docs/cli/README.md

@@ -140,10 +140,14 @@ Options:
 **Selection (2/4):** `environment-box`
 The config for the [BoxEnvironment]
 
-Usage: --environment-box.*
+Usage: --environment-box.* [OPTIONS] --rest-api-url <REST_API_URL> --db-dir <DB_DIR>
 
 Options:
-  -h, --help  Print help (see more with '--help')
+      --rest-api-url <REST_API_URL>  The rest api url of the box environment
+      --db-dir <DB_DIR>              The db dir of the box environment
+      --snapshot-dir <SNAPSHOT_DIR>  Whether to isolate the box environment by snapshotting the movement runner and
+                                     where to store the snapshot
+  -h, --help                         Print help (see more with '--help')
 
 **Selection (3/4):** `environment-provisioner`
 The config for the [ProvisionerEnvironment]

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

@@ -10,7 +10,7 @@ pub enum Migrate {
 	/// Core migration over the node.
 	Core(core::Core),
 	/// Select migration over the node.
-	Select(select::select::Select),
+	Select(select::select_command::Select),
 }
 
 impl Migrate {