Add NAPI wrapper for create_deployment

Author: Luke-Sanderson

Cargo.toml

@@ -10,10 +10,11 @@ crate-type = ["cdylib"]
 
 [dependencies]
 anyhow = "1.0.99"
-atlas-local = { git = "https://github.com/mongodb/atlas-local-lib.git", rev = "36f56065e891bbe045beeb46489dd7d4142dbd41" }
+atlas-local = { git = "https://github.com/mongodb/atlas-local-lib.git", rev = "d471e14e1a4462cad95b42ce504f2dcd28b81ee5" }
 bollard = "0.19.2"
 napi = { version = "3.0.0", features = ["async", "anyhow"] }
 napi-derive = "3.0.0"
+semver = "1.0.26"
 
 [build-dependencies]
 napi-build = "2"

__test__/index.spec.ts

@@ -20,16 +20,23 @@ test('smoke test', async (t) => {
     return
   }
 
-  // TODO: Implement once createDeployment is added
-  // let deploymentName = "test_deployment"
-  // await client.createDeployment(...)
+  // Count initial deployments
+  let start_deployments_count = (await client.listDeployments()).length
 
-  // List deployments
-  // We don't care about the number, we're just testing that the method doesn't fail
-  await client.listDeployments()
+  // Create deployment
+  let createDeploymentOptions = {
+    name: "test_deployment",
+  }
+  await client.createDeployment(createDeploymentOptions)
+
+  // Count deployments after creation
+  let after_create_deployment_count = (await client.listDeployments()).length
+  t.assert(after_create_deployment_count - start_deployments_count === 1)
 
-  // TODO: Uncommment when createDeployment is added
-  // await client.deleteDeployment(deploymentName)
+  // Delete deployment
+  await client.deleteDeployment(createDeploymentOptions.name)
 
-  t.pass()
+  // Count deployments after deletion
+  let after_delete_deployment_count = (await client.listDeployments()).length
+  t.assert(start_deployments_count === after_delete_deployment_count)
 })

index.d.ts

@@ -2,6 +2,7 @@
 /* eslint-disable */
 export declare class Client {
   static connect(): Client
+  createDeployment(createDeplomentOptions: CreateDeploymentOptions): Promise<void>
   listDeployments(): Promise<Array<Deployment>>
   deleteDeployment(deploymentName: string): Promise<void>
 }
@@ -12,6 +13,24 @@ export declare const enum BindingType {
   Specific = 'Specific'
 }
 
+export interface CreateDeploymentOptions {
+  name?: string
+  image?: string
+  mongodbVersion?: string
+  creationSource?: CreationSource
+  localSeedLocation?: string
+  mongodbInitdbDatabase?: string
+  mongodbInitdbRootPasswordFile?: string
+  mongodbInitdbRootPassword?: string
+  mongodbInitdbRootUsernameFile?: string
+  mongodbInitdbRootUsername?: string
+  mongotLogFile?: string
+  runnerLogFile?: string
+  doNotTrack?: boolean
+  telemetryBaseUrl?: string
+  mongodbPortBinding?: MongoDBPortBinding
+}
+
 export interface CreationSource {
   type: CreationSourceType
   source: string
@@ -20,7 +39,7 @@ export interface CreationSource {
 export declare const enum CreationSourceType {
   AtlasCLI = 'AtlasCLI',
   Container = 'Container',
-  MCP = 'MCP',
+  MCPServer = 'MCPServer',
   Other = 'Other'
 }
 

src/lib.rs

@@ -27,6 +27,19 @@ impl Client {
     })
   }
 
+  #[napi]
+  pub async fn create_deployment(
+    &self,
+    create_deploment_options: crate::models::create_deployment::CreateDeploymentOptions,
+  ) -> Result<()> {
+    let options: atlas_local::models::CreateDeploymentOptions = create_deploment_options.into();
+    self
+      .client
+      .create_deployment(&options)
+      .await
+      .context("create deployment")
+  }
+
   #[napi]
   pub async fn list_deployments(&self) -> Result<Vec<Deployment>> {
     self

src/models/create_deployment.rs

@@ -0,0 +1,70 @@
+use crate::models::list_deployments::{CreationSource, MongoDBPortBinding};
+use napi_derive::napi;
+use semver::Version;
+
+#[napi(object)]
+pub struct CreateDeploymentOptions {
+  // Identifiers
+  pub name: Option<String>,
+
+  // Image details
+  pub image: Option<String>,
+  pub mongodb_version: Option<String>,
+
+  // Creation source
+  pub creation_source: Option<CreationSource>,
+
+  // Initial database configuration
+  pub local_seed_location: Option<String>,
+  pub mongodb_initdb_database: Option<String>,
+  pub mongodb_initdb_root_password_file: Option<String>,
+  pub mongodb_initdb_root_password: Option<String>,
+  pub mongodb_initdb_root_username_file: Option<String>,
+  pub mongodb_initdb_root_username: Option<String>,
+
+  // Logging
+  pub mongot_log_file: Option<String>,
+  pub runner_log_file: Option<String>,
+
+  // Telemetry
+  pub do_not_track: Option<bool>,
+  pub telemetry_base_url: Option<String>,
+
+  // Port configuration
+  pub mongodb_port_binding: Option<MongoDBPortBinding>,
+}
+
+impl From<CreateDeploymentOptions> for atlas_local::models::CreateDeploymentOptions {
+  fn from(source: CreateDeploymentOptions) -> Self {
+    let version: Option<Version> = match source.mongodb_version.as_deref() {
+      Some("latest") => None,
+      None => None,
+      Some(ver_string) => {
+        // If malformed Version if given, it will panic here
+        Some(Version::parse(ver_string).expect("Parse version string"))
+      }
+    };
+
+    Self {
+      name: source.name,
+      image: source.image,
+      mongodb_version: version,
+      creation_source: source
+        .creation_source
+        .map(atlas_local::models::CreationSource::from),
+      local_seed_location: source.local_seed_location,
+      mongodb_initdb_database: source.mongodb_initdb_database,
+      mongodb_initdb_root_password_file: source.mongodb_initdb_root_password_file,
+      mongodb_initdb_root_password: source.mongodb_initdb_root_password,
+      mongodb_initdb_root_username_file: source.mongodb_initdb_root_username_file,
+      mongodb_initdb_root_username: source.mongodb_initdb_root_username,
+      mongot_log_file: source.mongot_log_file,
+      runner_log_file: source.runner_log_file,
+      do_not_track: source.do_not_track,
+      telemetry_base_url: source.telemetry_base_url,
+      mongodb_port_binding: source
+        .mongodb_port_binding
+        .map(atlas_local::models::MongoDBPortBinding::from),
+    }
+  }
+}

src/models/list_deployments.rs

@@ -1,3 +1,5 @@
+use std::net::IpAddr;
+
 use napi_derive::napi;
 
 #[napi(object)]
@@ -77,7 +79,7 @@ pub struct CreationSource {
 pub enum CreationSourceType {
   AtlasCLI,
   Container,
-  MCP,
+  MCPServer,
   Other,
 }
 
@@ -143,6 +145,27 @@ impl From<atlas_local::models::MongoDBPortBinding> for MongoDBPortBinding {
   }
 }
 
+impl From<MongoDBPortBinding> for atlas_local::models::MongoDBPortBinding {
+  fn from(source: MongoDBPortBinding) -> Self {
+    match source.binding_type {
+      BindingType::Loopback => atlas_local::models::MongoDBPortBinding {
+        binding_type: atlas_local::models::BindingType::Loopback,
+        port: source.port,
+      },
+      BindingType::AnyInterface => atlas_local::models::MongoDBPortBinding {
+        binding_type: atlas_local::models::BindingType::AnyInterface,
+        port: source.port,
+      },
+      BindingType::Specific => atlas_local::models::MongoDBPortBinding {
+        binding_type: atlas_local::models::BindingType::Specific(
+          source.ip.parse::<IpAddr>().expect("Parse IP address"),
+        ),
+        port: source.port,
+      },
+    }
+  }
+}
+
 impl From<atlas_local::models::MongodbType> for MongodbType {
   fn from(source: atlas_local::models::MongodbType) -> Self {
     match source {
@@ -165,10 +188,25 @@ impl From<atlas_local::models::CreationSource> for CreationSource {
         source_type: CreationSourceType::Container,
         source: "CONTAINER".to_string(),
       },
+      CreationSourceSource::MCPServer => CreationSource {
+        source_type: CreationSourceType::MCPServer,
+        source: "MCPSERVER".to_string(),
+      },
       CreationSourceSource::Unknown(source) => CreationSource {
         source_type: CreationSourceType::Other,
         source,
       },
     }
   }
 }
+
+impl From<CreationSource> for atlas_local::models::CreationSource {
+  fn from(source: CreationSource) -> Self {
+    match source.source_type {
+      CreationSourceType::AtlasCLI => atlas_local::models::CreationSource::AtlasCLI,
+      CreationSourceType::Container => atlas_local::models::CreationSource::Container,
+      CreationSourceType::MCPServer => atlas_local::models::CreationSource::MCPServer,
+      CreationSourceType::Other => atlas_local::models::CreationSource::Unknown(source.source),
+    }
+  }
+}

src/models/mod.rs

@@ -1 +1,2 @@
+pub mod create_deployment;
 pub mod list_deployments;