Separately walk over ClassBody

acornjs · Feb 7, 2018 · 24147c6 · 24147c6
1 parent 7d3267d
commit 24147c6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/walk/index.js b/src/walk/index.js
@@ -361,8 +361,11 @@ base.ClassDeclaration = base.ClassExpression = (node, st, c) => c(node, st, "Cla
 base.Class = (node, st, c) => {
   if (node.id) c(node.id, st, "Pattern")
   if (node.superClass) c(node.superClass, st, "Expression")
-  for (let item of node.body.body)
-    c(item, st)
+  c(node.body, st)
+}
+base.ClassBody = (node, st, c) => {
+  for (let elt of node.body)
+    c(elt, st)
 }
 base.MethodDefinition = base.Property = (node, st, c) => {
   if (node.computed) c(node.key, st, "Expression")