You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
259 lines
7.0 KiB
259 lines
7.0 KiB
"use strict"; |
|
|
|
Object.defineProperty(exports, "__esModule", { |
|
value: true |
|
}); |
|
exports._replaceWith = _replaceWith; |
|
exports.replaceExpressionWithStatements = replaceExpressionWithStatements; |
|
exports.replaceInline = replaceInline; |
|
exports.replaceWith = replaceWith; |
|
exports.replaceWithMultiple = replaceWithMultiple; |
|
exports.replaceWithSourceString = replaceWithSourceString; |
|
|
|
var _codeFrame = require("@babel/code-frame"); |
|
|
|
var _index = require("../index"); |
|
|
|
var _index2 = require("./index"); |
|
|
|
var _cache = require("../cache"); |
|
|
|
var _parser = require("@babel/parser"); |
|
|
|
var _t = require("@babel/types"); |
|
|
|
var _helperHoistVariables = require("@babel/helper-hoist-variables"); |
|
|
|
const { |
|
FUNCTION_TYPES, |
|
arrowFunctionExpression, |
|
assignmentExpression, |
|
awaitExpression, |
|
blockStatement, |
|
callExpression, |
|
cloneNode, |
|
expressionStatement, |
|
identifier, |
|
inheritLeadingComments, |
|
inheritTrailingComments, |
|
inheritsComments, |
|
isExpression, |
|
isProgram, |
|
isStatement, |
|
removeComments, |
|
returnStatement, |
|
toSequenceExpression, |
|
validate, |
|
yieldExpression |
|
} = _t; |
|
|
|
function replaceWithMultiple(nodes) { |
|
var _pathCache$get; |
|
|
|
this.resync(); |
|
nodes = this._verifyNodeList(nodes); |
|
inheritLeadingComments(nodes[0], this.node); |
|
inheritTrailingComments(nodes[nodes.length - 1], this.node); |
|
(_pathCache$get = _cache.path.get(this.parent)) == null ? void 0 : _pathCache$get.delete(this.node); |
|
this.node = this.container[this.key] = null; |
|
const paths = this.insertAfter(nodes); |
|
|
|
if (this.node) { |
|
this.requeue(); |
|
} else { |
|
this.remove(); |
|
} |
|
|
|
return paths; |
|
} |
|
|
|
function replaceWithSourceString(replacement) { |
|
this.resync(); |
|
let ast; |
|
|
|
try { |
|
replacement = `(${replacement})`; |
|
ast = (0, _parser.parse)(replacement); |
|
} catch (err) { |
|
const loc = err.loc; |
|
|
|
if (loc) { |
|
err.message += " - make sure this is an expression.\n" + (0, _codeFrame.codeFrameColumns)(replacement, { |
|
start: { |
|
line: loc.line, |
|
column: loc.column + 1 |
|
} |
|
}); |
|
err.code = "BABEL_REPLACE_SOURCE_ERROR"; |
|
} |
|
|
|
throw err; |
|
} |
|
|
|
const expressionAST = ast.program.body[0].expression; |
|
|
|
_index.default.removeProperties(expressionAST); |
|
|
|
return this.replaceWith(expressionAST); |
|
} |
|
|
|
function replaceWith(replacementPath) { |
|
this.resync(); |
|
|
|
if (this.removed) { |
|
throw new Error("You can't replace this node, we've already removed it"); |
|
} |
|
|
|
let replacement = replacementPath instanceof _index2.default ? replacementPath.node : replacementPath; |
|
|
|
if (!replacement) { |
|
throw new Error("You passed `path.replaceWith()` a falsy node, use `path.remove()` instead"); |
|
} |
|
|
|
if (this.node === replacement) { |
|
return [this]; |
|
} |
|
|
|
if (this.isProgram() && !isProgram(replacement)) { |
|
throw new Error("You can only replace a Program root node with another Program node"); |
|
} |
|
|
|
if (Array.isArray(replacement)) { |
|
throw new Error("Don't use `path.replaceWith()` with an array of nodes, use `path.replaceWithMultiple()`"); |
|
} |
|
|
|
if (typeof replacement === "string") { |
|
throw new Error("Don't use `path.replaceWith()` with a source string, use `path.replaceWithSourceString()`"); |
|
} |
|
|
|
let nodePath = ""; |
|
|
|
if (this.isNodeType("Statement") && isExpression(replacement)) { |
|
if (!this.canHaveVariableDeclarationOrExpression() && !this.canSwapBetweenExpressionAndStatement(replacement) && !this.parentPath.isExportDefaultDeclaration()) { |
|
replacement = expressionStatement(replacement); |
|
nodePath = "expression"; |
|
} |
|
} |
|
|
|
if (this.isNodeType("Expression") && isStatement(replacement)) { |
|
if (!this.canHaveVariableDeclarationOrExpression() && !this.canSwapBetweenExpressionAndStatement(replacement)) { |
|
return this.replaceExpressionWithStatements([replacement]); |
|
} |
|
} |
|
|
|
const oldNode = this.node; |
|
|
|
if (oldNode) { |
|
inheritsComments(replacement, oldNode); |
|
removeComments(oldNode); |
|
} |
|
|
|
this._replaceWith(replacement); |
|
|
|
this.type = replacement.type; |
|
this.setScope(); |
|
this.requeue(); |
|
return [nodePath ? this.get(nodePath) : this]; |
|
} |
|
|
|
function _replaceWith(node) { |
|
var _pathCache$get2; |
|
|
|
if (!this.container) { |
|
throw new ReferenceError("Container is falsy"); |
|
} |
|
|
|
if (this.inList) { |
|
validate(this.parent, this.key, [node]); |
|
} else { |
|
validate(this.parent, this.key, node); |
|
} |
|
|
|
this.debug(`Replace with ${node == null ? void 0 : node.type}`); |
|
(_pathCache$get2 = _cache.path.get(this.parent)) == null ? void 0 : _pathCache$get2.set(node, this).delete(this.node); |
|
this.node = this.container[this.key] = node; |
|
} |
|
|
|
function replaceExpressionWithStatements(nodes) { |
|
this.resync(); |
|
const nodesAsSequenceExpression = toSequenceExpression(nodes, this.scope); |
|
|
|
if (nodesAsSequenceExpression) { |
|
return this.replaceWith(nodesAsSequenceExpression)[0].get("expressions"); |
|
} |
|
|
|
const functionParent = this.getFunctionParent(); |
|
const isParentAsync = functionParent == null ? void 0 : functionParent.is("async"); |
|
const isParentGenerator = functionParent == null ? void 0 : functionParent.is("generator"); |
|
const container = arrowFunctionExpression([], blockStatement(nodes)); |
|
this.replaceWith(callExpression(container, [])); |
|
const callee = this.get("callee"); |
|
(0, _helperHoistVariables.default)(callee.get("body"), id => { |
|
this.scope.push({ |
|
id |
|
}); |
|
}, "var"); |
|
const completionRecords = this.get("callee").getCompletionRecords(); |
|
|
|
for (const path of completionRecords) { |
|
if (!path.isExpressionStatement()) continue; |
|
const loop = path.findParent(path => path.isLoop()); |
|
|
|
if (loop) { |
|
let uid = loop.getData("expressionReplacementReturnUid"); |
|
|
|
if (!uid) { |
|
uid = callee.scope.generateDeclaredUidIdentifier("ret"); |
|
callee.get("body").pushContainer("body", returnStatement(cloneNode(uid))); |
|
loop.setData("expressionReplacementReturnUid", uid); |
|
} else { |
|
uid = identifier(uid.name); |
|
} |
|
|
|
path.get("expression").replaceWith(assignmentExpression("=", cloneNode(uid), path.node.expression)); |
|
} else { |
|
path.replaceWith(returnStatement(path.node.expression)); |
|
} |
|
} |
|
|
|
callee.arrowFunctionToExpression(); |
|
const newCallee = callee; |
|
|
|
const needToAwaitFunction = isParentAsync && _index.default.hasType(this.get("callee.body").node, "AwaitExpression", FUNCTION_TYPES); |
|
|
|
const needToYieldFunction = isParentGenerator && _index.default.hasType(this.get("callee.body").node, "YieldExpression", FUNCTION_TYPES); |
|
|
|
if (needToAwaitFunction) { |
|
newCallee.set("async", true); |
|
|
|
if (!needToYieldFunction) { |
|
this.replaceWith(awaitExpression(this.node)); |
|
} |
|
} |
|
|
|
if (needToYieldFunction) { |
|
newCallee.set("generator", true); |
|
this.replaceWith(yieldExpression(this.node, true)); |
|
} |
|
|
|
return newCallee.get("body.body"); |
|
} |
|
|
|
function replaceInline(nodes) { |
|
this.resync(); |
|
|
|
if (Array.isArray(nodes)) { |
|
if (Array.isArray(this.container)) { |
|
nodes = this._verifyNodeList(nodes); |
|
|
|
const paths = this._containerInsertAfter(nodes); |
|
|
|
this.remove(); |
|
return paths; |
|
} else { |
|
return this.replaceWithMultiple(nodes); |
|
} |
|
} else { |
|
return this.replaceWith(nodes); |
|
} |
|
} |