chore(export-to-language): reokace usage of vm module COMPASS-7312

mongodb-js · Oct 9, 2023 · 8f60807 · 8f60807
1 parent c27cf54
commit 8f60807
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 48 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/packages/compass-export-to-language/package.json b/packages/compass-export-to-language/package.json
@@ -61,14 +61,16 @@
     "@mongodb-js/compass-maybe-protect-connection-string": "^0.13.0",
     "@mongodb-js/mongodb-redux-common": "*",
     "bson-transpilers": "*",
+    "ejson-shell-parser": "^1.2.4",
     "react": "^17.0.2"
   },
   "dependencies": {
     "@mongodb-js/compass-components": "^1.15.0",
     "@mongodb-js/compass-logging": "^1.2.1",
     "@mongodb-js/compass-maybe-protect-connection-string": "^0.13.0",
     "@mongodb-js/mongodb-redux-common": "^2.0.12",
-    "bson-transpilers": "^2.0.4"
+    "bson-transpilers": "^2.0.4",
+    "ejson-shell-parser": "^1.2.4"
   },
   "devDependencies": {
     "@mongodb-js/eslint-config-compass": "^1.0.9",

diff --git a/packages/compass-export-to-language/src/modules/count-aggregation-stages-in-string.js b/packages/compass-export-to-language/src/modules/count-aggregation-stages-in-string.js
diff --git a/packages/compass-export-to-language/src/modules/count-aggregation-stages-in-string.ts b/packages/compass-export-to-language/src/modules/count-aggregation-stages-in-string.ts
@@ -0,0 +1,9 @@
+import parseShellBSON, { ParseMode } from 'ejson-shell-parser';
+
+export function countAggregationStagesInString(source: string): number {
+  const parsed = parseShellBSON(source, { mode: ParseMode.Loose });
+  if (!Array.isArray(parsed)) {
+    throw new Error('Source expression is not an aggregation stage array');
+  }
+  return parsed.length;
+}