refactor(probe): allow array of validateNode functions (#191)

NodeSecure · Dec 21, 2023 · 1002d8c · 1002d8c
1 parent 64b8350
commit 1002d8c
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/probes/index.js b/src/probes/index.js
@@ -39,17 +39,23 @@ const kListOfProbes = [
 const kSymBreak = Symbol.for("breakWalk");
 const kSymSkip = Symbol.for("skipWalk");
 
-export function runOnProbes(node, analysis) {
+export function runOnProbes(node, sourceFile) {
   const breakedGroups = new Set();
 
   for (const probe of kListOfProbes) {
     if (breakedGroups.has(probe.breakGroup)) {
       continue;
     }
 
-    const [isMatching, data = null] = probe.validateNode(node, analysis);
-    if (isMatching) {
-      const result = probe.main(node, { analysis, data });
+    const validationFns = Array.isArray(probe.validateNode) ?
+      probe.validateNode : [probe.validateNode];
+    for (const validateNode of validationFns) {
+      const [isMatching, data = null] = validateNode(node, sourceFile);
+      if (!isMatching) {
+        continue;
+      }
+
+      const result = probe.main(node, { analysis: sourceFile, data });
 
       if (result === kSymSkip) {
         return "skip";