diff --git a/packages/babel-traverse/src/path/removal.js b/packages/babel-traverse/src/path/removal.js index 07144f2b6bc5..a8bd4523fdc6 100644 --- a/packages/babel-traverse/src/path/removal.js +++ b/packages/babel-traverse/src/path/removal.js @@ -1,6 +1,7 @@ // This file contains methods responsible for removing a node. import { hooks } from "./lib/removal-hooks"; +import { path as pathCache } from "../cache"; import { REMOVED, SHOULD_SKIP } from "./index"; export function remove() { @@ -44,6 +45,7 @@ export function _remove() { export function _markRemoved() { // this.shouldSkip = true; this.removed = true; this._traverseFlags |= SHOULD_SKIP | REMOVED; + if (this.parent) pathCache.get(this.parent).delete(this.node); this.node = null; } diff --git a/packages/babel-traverse/src/path/replacement.js b/packages/babel-traverse/src/path/replacement.js index 211940841367..c3add1418745 100644 --- a/packages/babel-traverse/src/path/replacement.js +++ b/packages/babel-traverse/src/path/replacement.js @@ -3,6 +3,7 @@ import { codeFrameColumns } from "@babel/code-frame"; import traverse from "../index"; import NodePath from "./index"; +import { path as pathCache } from "../cache"; import { parse } from "@babel/parser"; import * as t from "@babel/types"; @@ -49,6 +50,7 @@ export function replaceWithMultiple(nodes: Array) { nodes = this._verifyNodeList(nodes); t.inheritLeadingComments(nodes[0], this.node); t.inheritTrailingComments(nodes[nodes.length - 1], this.node); + pathCache.get(this.parent).delete(this.node); this.node = this.container[this.key] = null; const paths = this.insertAfter(nodes);