feat: add support for loading sample data

Author: jeroenvervaeke

Cargo.lock

@@ -11,7 +11,7 @@ crate-type = ["cdylib"]
 
 [dependencies]
 anyhow = "1.0.99"
-atlas-local = { git = "https://github.com/mongodb/atlas-local-lib.git", rev = "9b582845ebf2e4ca4f6543fa5bb1262920e2fbe6" }
+atlas-local = { git = "https://github.com/mongodb/atlas-local-lib.git", rev = "493fb78eb2f38228d486c8537274ab752f1eac04" }
 bollard = "0.19.4"
 napi = { version = "^3.5.2", features = ["async", "anyhow"] }
 napi-derive = "^3.3.3"

Cargo.toml

@@ -33,6 +33,7 @@ test('smoke test', async (t) => {
   // Create deployment
   let createDeploymentOptions = {
     name: "test_deployment",
+    doNotTrack: true,
   }
   let deployment = await client.createDeployment(createDeploymentOptions)
   t.is(deployment.name, createDeploymentOptions.name)

__test__/index.spec.ts

@@ -22,6 +22,7 @@ export interface CreateDeploymentOptions {
   waitUntilHealthyTimeout?: number
   creationSource?: CreationSource
   localSeedLocation?: string
+  loadSampleData?: boolean
   mongodbInitdbDatabase?: string
   mongodbInitdbRootPasswordFile?: string
   mongodbInitdbRootPassword?: string
@@ -58,9 +59,10 @@ export interface Deployment {
   mongodbInitdbRootPassword?: string
   mongodbInitdbRootUsernameFile?: string
   mongodbInitdbRootUsername?: string
+  mongodbLoadSampleData?: boolean
   mongotLogFile?: string
   runnerLogFile?: string
-  doNotTrack?: string
+  doNotTrack: boolean
   telemetryBaseUrl?: string
 }
 

index.d.ts

@@ -3,9 +3,6 @@
 // @ts-nocheck
 /* auto-generated by NAPI-RS */
 
-const { createRequire } = require('node:module')
-require = createRequire(__filename)
-
 const { readFileSync } = require('node:fs')
 let nativeBinding = null
 const loadErrors = []
@@ -108,7 +105,7 @@ function requireNative() {
     }
   } else if (process.platform === 'win32') {
     if (process.arch === 'x64') {
-      if (process.report?.getReport?.()?.header?.osName?.startsWith?.('MINGW')) {
+      if (process.config?.variables?.shlib_suffix === 'dll.a' || process.config?.variables?.node_target_type === 'shared_library') {
         try {
         return require('./atlas-local.win32-x64-gnu.node')
       } catch (e) {

index.js

@@ -19,6 +19,7 @@ pub struct CreateDeploymentOptions {
 
   // Initial database configuration
   pub local_seed_location: Option<String>,
+  pub load_sample_data: Option<bool>,
   pub mongodb_initdb_database: Option<String>,
   pub mongodb_initdb_root_password_file: Option<String>,
   pub mongodb_initdb_root_password: Option<String>,
@@ -60,6 +61,7 @@ impl From<CreateDeploymentOptions> for atlas_local::models::CreateDeploymentOpti
         .creation_source
         .map(atlas_local::models::CreationSource::from),
       local_seed_location: source.local_seed_location,
+      load_sample_data: source.load_sample_data,
       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,
@@ -95,6 +97,7 @@ mod tests {
         source: "MCPSERVER".to_string(),
       }),
       local_seed_location: Some("/host/seed-data".to_string()),
+      load_sample_data: Some(true),
       mongodb_initdb_database: Some("testdb".to_string()),
       mongodb_initdb_root_password_file: Some("/run/secrets/password".to_string()),
       mongodb_initdb_root_password: Some("password123".to_string()),
@@ -137,6 +140,7 @@ mod tests {
       lib_create_deployment_options.local_seed_location,
       Some("/host/seed-data".to_string())
     );
+    assert_eq!(lib_create_deployment_options.load_sample_data, Some(true));
     assert_eq!(
       lib_create_deployment_options.mongodb_initdb_database,
       Some("testdb".to_string())

src/models/create_deployment.rs

@@ -26,13 +26,14 @@ pub struct Deployment {
   pub mongodb_initdb_root_password: Option<String>,
   pub mongodb_initdb_root_username_file: Option<String>,
   pub mongodb_initdb_root_username: Option<String>,
+  pub mongodb_load_sample_data: Option<bool>,
 
   // Logging
   pub mongot_log_file: Option<String>,
   pub runner_log_file: Option<String>,
 
   // Telemetry
-  pub do_not_track: Option<String>,
+  pub do_not_track: bool,
   pub telemetry_base_url: Option<String>,
 }
 
@@ -105,6 +106,7 @@ impl From<atlas_local::models::Deployment> for Deployment {
       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,
+      mongodb_load_sample_data: source.mongodb_load_sample_data,
       mongot_log_file: source.mongot_log_file,
       runner_log_file: source.runner_log_file,
       do_not_track: source.do_not_track,
@@ -242,9 +244,10 @@ mod tests {
       mongodb_initdb_root_password: Some("password123".to_string()),
       mongodb_initdb_root_username_file: Some("/run/secrets/username".to_string()),
       mongodb_initdb_root_username: Some("admin".to_string()),
+      mongodb_load_sample_data: Some(false),
       mongot_log_file: Some("/tmp/mongot.log".to_string()),
       runner_log_file: Some("/tmp/runner.log".to_string()),
-      do_not_track: Some("false".to_string()),
+      do_not_track: false,
       telemetry_base_url: Some("https://telemetry.example.com".to_string()),
     };
 
@@ -291,6 +294,7 @@ mod tests {
       deployment.mongodb_initdb_root_username,
       Some("admin".to_string())
     );
+    assert_eq!(deployment.mongodb_load_sample_data, Some(false));
     assert_eq!(
       deployment.mongot_log_file,
       Some("/tmp/mongot.log".to_string())
@@ -299,7 +303,7 @@ mod tests {
       deployment.runner_log_file,
       Some("/tmp/runner.log".to_string())
     );
-    assert_eq!(deployment.do_not_track, Some("false".to_string()));
+    assert_eq!(deployment.do_not_track, false);
     assert_eq!(
       deployment.telemetry_base_url,
       Some("https://telemetry.example.com".to_string())