Replace loops with foreach

import-js · Jun 24, 2020 · c10e217 · c10e217
1 parent edbc93b
commit c10e217
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/utils/visit.js b/utils/visit.js
@@ -13,16 +13,11 @@ exports.default = function visit(node, keys, visitorSpec) {
   if (!childFields) {
     return
   }
-  for (const fieldName of childFields) {
-    const field = node[fieldName]
-    if (Array.isArray(field)) {
-      for (const item of field) {
-        visit(item, keys, visitorSpec)
-      }
-    } else {
-      visit(field, keys, visitorSpec)
-    }
-  }
+  childFields.forEach((fieldName) => {
+    [].concat(node[fieldName]).forEach((item) => {
+      visit(item, keys, visitorSpec)
+    })
+  })
   if (typeof visitorSpec[`${type}:Exit`] === 'function') {
     visitorSpec[`${type}:Exit`](node)
   }