fix(transform): transform should still apply even when node is removed

2019-09-22 17:02:32 -04:00
parent 3a177a18d2
commit 9c01e0bf0d
2 changed files with 14 additions and 9 deletions
--- a/packages/compiler-core/src/transform.ts
+++ b/packages/compiler-core/src/transform.ts
@@ -132,12 +132,13 @@ function traverseNode(
  for (let i = 0; i < nodeTransforms.length; i++) {
    const plugin = nodeTransforms[i]
    plugin(node, context)
-    if (!context.currentNode) {
-      return
-    } else {
-      // node may have been replaced
-      node = context.currentNode
-    }
+    // node may have been replaced
+    node = context.currentNode || node
+  }
+
+  if (!context.currentNode) {
+    // node was removed
+    return
  }

  // further traverse downwards