diff --git a/src/babel/transformation/modules/common.js b/src/babel/transformation/modules/common.js index 3f43eda36f..90261a7b3b 100644 --- a/src/babel/transformation/modules/common.js +++ b/src/babel/transformation/modules/common.js @@ -20,7 +20,7 @@ export default class CommonJSFormatter extends DefaultFormatter { if (this.file.isLoose("es6.modules")) templateName += "-loose"; var declar = util.template(templateName, true); declar._blockHoist = 3; - file.ast.program.body.unshift(declar); + file.path.unshiftContainer("body", [declar]); } } diff --git a/src/babel/transformation/transformers/internal/strict.js b/src/babel/transformation/transformers/internal/strict.js index 6ffe5f7467..d3f6a4ca00 100644 --- a/src/babel/transformation/transformers/internal/strict.js +++ b/src/babel/transformation/transformers/internal/strict.js @@ -13,6 +13,6 @@ export function Program(program, parent, scope, file) { } } - program.body.unshift(directive); + this.unshiftContainer("body", [directive]); } } diff --git a/src/babel/traversal/path/index.js b/src/babel/traversal/path/index.js index 3a89790f48..197e9a66bd 100644 --- a/src/babel/traversal/path/index.js +++ b/src/babel/traversal/path/index.js @@ -425,7 +425,7 @@ export default class TraversalPath { * Description */ - insertOntoContainerStart(containerKey, nodes) { + unshiftContainer(containerKey, nodes) { nodes = this._verifyNodeList(nodes); // get the first path and insert our nodes before it, if it doesn't exist then it @@ -441,7 +441,7 @@ export default class TraversalPath { * Description */ - insertOntoContainerEnd(containerKey, nodes) { + pushContainer(containerKey, nodes) { nodes = this._verifyNodeList(nodes); // get an invisible path that represents the last node + 1 and replace it with our