chore(export-to-language): replace usage of vm module COMPASS-7312 (#…

…4967)
mongodb-js · Oct 10, 2023 · 66099ac · 66099ac
1 parent 2d07d1d
commit 66099ac
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 45 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
@@ -80,6 +80,7 @@
     "chai": "^4.3.6",
     "compass-preferences-model": "^2.15.1",
     "depcheck": "^1.4.1",
+    "ejson-shell-parser": "^1.2.4",
     "enzyme": "^3.11.0",
     "eslint": "^7.25.0",
     "hadron-app-registry": "^9.0.11",

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;
+}