feat: Implement get_connection_string

Author: cveticm

Cargo.toml

@@ -10,7 +10,7 @@ crate-type = ["cdylib"]
 
 [dependencies]
 anyhow = "1.0.99"
-atlas-local = { git = "https://github.com/mongodb/atlas-local-lib.git", rev = "7d8de1e2b9b865fafeb4120c765b1223d500bbc1" }
+atlas-local = { git = "https://github.com/mongodb/atlas-local-lib.git", rev = "c5ac7e298dc7a654028784f94737b7100a2ad7ff" }
 bollard = "0.19.2"
 napi = { version = "3.0.0", features = ["async", "anyhow"] }
 napi-derive = "3.0.0"

__test__/index.spec.ts

@@ -41,6 +41,12 @@ test('smoke test', async (t) => {
   let getDeployment = await client.getDeployment(createDeploymentOptions.name)
   t.is(getDeployment.name,createDeploymentOptions.name)
 
+  let getConnectionStringOptions = {
+    containerIdOrName: createDeploymentOptions.name,
+  }
+  let connString = await client.getConnectionString(getConnectionStringOptions)
+  t.assert(connString === `mongodb://127.0.0.1:${getDeployment.portBindings.port}/?directConnection=true`)
+
   // Count deployments after creation
   let after_create_deployment_count = (await client.listDeployments()).length
   t.assert(after_create_deployment_count - start_deployments_count === 1)

index.d.ts

@@ -6,6 +6,7 @@ export declare class Client {
   listDeployments(): Promise<Array<Deployment>>
   deleteDeployment(deploymentName: string): Promise<void>
   getDeployment(deploymentName: string): Promise<Deployment>
+  getConnectionString(options: GetConnectionStringOptions): Promise<string>
 }
 
 export declare const enum BindingType {
@@ -66,6 +67,13 @@ export interface Deployment {
   telemetryBaseUrl?: string
 }
 
+export interface GetConnectionStringOptions {
+  containerIdOrName: string
+  dbUsername?: string
+  dbPassword?: string
+  verify?: boolean
+}
+
 export interface MongoDbPortBinding {
   type: BindingType
   ip: string

index.js

@@ -69,4 +69,17 @@ impl Client {
       .context("get deployment")
       .map(|d| d.into())
   }
+
+  #[napi]
+  pub async fn get_connection_string(
+    &self,
+    options: crate::models::get_connection_string::GetConnectionStringOptions,
+  ) -> Result<String> {
+    let options: atlas_local::models::GetConnectionStringOptions = options.into();
+    self
+      .client
+      .get_connection_string(options)
+      .await
+      .context("get connection string")
+  }
 }

src/lib.rs

@@ -0,0 +1,43 @@
+use napi_derive::napi;
+
+#[napi(object)]
+pub struct GetConnectionStringOptions {
+    pub container_id_or_name: String,
+    pub db_username: Option<String>,
+    pub db_password: Option<String>,
+    pub verify: Option<bool>,
+}
+
+impl From<GetConnectionStringOptions> for atlas_local::models::GetConnectionStringOptions {
+    fn from(source: GetConnectionStringOptions) -> Self {
+        Self {
+            container_id_or_name: source.container_id_or_name,
+            db_username: source.db_username,
+            db_password: source.db_password,
+            verify: source.verify,
+        }
+    }
+}
+
+// test
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_get_connection_string_options() {
+        let options = GetConnectionStringOptions {
+            container_id_or_name: "test_container".into(),
+            db_username: Some("test_user".into()),
+            db_password: Some("test_pass".into()),
+            verify: Some(true),
+        };
+
+        let get_connection_string_options: atlas_local::models::GetConnectionStringOptions = options.into();
+
+        assert_eq!(get_connection_string_options.container_id_or_name, "test_container");
+        assert_eq!(get_connection_string_options.db_username, Some("test_user".into()));
+        assert_eq!(get_connection_string_options.db_password, Some("test_pass".into()));
+        assert_eq!(get_connection_string_options.verify, Some(true));
+    }
+}

src/models/get_connection_string.rs

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