Merge pull request #47 from PCOffline/feature/validate-db-uri

feat: add db uri validation

Author: Eldar B

CHANGELOG.md

@@ -2,6 +2,18 @@
 
 All notable changes to this project will be documented in this file. See [standard-version](https://github.com/conventional-changelog/standard-version) for commit guidelines.
 
+### [0.4.1](https://github.com/pcoffline/reset-db/compare/v0.4.0...v0.4.1) (2022-10-08)
+
+
+### Features
+
+* add db uri validation ([fb81cd5](https://github.com/pcoffline/reset-db/commit/fb81cd59bf81d088db6d0e257fe4b59d3a1d6b03))
+
+
+### Bug Fixes
+
+* package.json & package-lock.json to reduce vulnerabilities ([8570b4d](https://github.com/pcoffline/reset-db/commit/8570b4d89f6fb2147a4f876c7635032433ad8825))
+
 ## [0.4.0](https://github.com/PCOffline/reset-db/compare/v0.3.0...v0.4.0) (2022-01-24)
 
 

index.js

@@ -162,6 +162,35 @@ async function evaluateCollectionValues({
 }
 
 async function initialise() {
+  // Validate Mongo URI values
+  const mongoUriValues = new Set(
+    ['MONGO_URI', 'MONGODB_URI', 'mongoUri', 'DB_URI', 'DATABASE_URI'].filter(
+      (name) => name in process.env,
+    ),
+  );
+
+  if (!config.mongoUri && !mongoUriValues.size()) {
+    logError('No MongoDB URI provided');
+    throw new Error('No MongoDB URI provided');
+  }
+
+  if (
+    mongoUriValues.size() > 1 ||
+    (mongoUriValues.size() &&
+      config.mongoUri !== mongoUriValues.values().next())
+  ) {
+    logError(
+      'Mismatch between MongoDB URIs:',
+      [config.mongoUri, ...Array.from(mongoUriValues)]
+        .filter((uri) => uri)
+        .join(', '),
+    );
+    throw new Error('Mismatch between MongoDB URIs');
+  }
+
+  // If config.mongoUri is unset, that means that the value is in the process.env
+  config.mongoUri ??= mongoUriValues.values().next();
+
   // Connect to mongoose
   await loadPromise(
     mongoose.connect(
@@ -235,18 +264,18 @@ async function initialise() {
       }),
     );
 
-  return loadPromise(modelsPromise, {
+  await loadPromise(modelsPromise, {
     text: 'Creating models',
     successText: 'Created models',
     failText: 'Failed to create models',
-  })
-    .then(logDebugData)
-    .then(() => {
-      if (standardCollections.length === 0) {
-        logError('No valid collections found!');
-        throw new Error('No valid collections found!');
-      }
-    });
+  });
+  
+  logDebugData();
+
+  if (standardCollections.length === 0) {
+    logError('No valid collections found!');
+    throw new Error('No valid collections found!');
+  }
 }
 
 async function dryRun() {

package-lock.json

@@ -1,7 +1,7 @@
 {
   "name": "reset-db",
   "type": "module",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "A fast and easy way to reset your collections",
   "main": "index.js",
   "scripts": {