| "use strict"; |
| |
| Object.defineProperty(exports, "__esModule", { |
| value: true |
| }); |
| exports.arrowFunctionToExpression = arrowFunctionToExpression; |
| exports.arrowFunctionToShadowed = arrowFunctionToShadowed; |
| exports.ensureBlock = ensureBlock; |
| exports.toComputedKey = toComputedKey; |
| exports.unwrapFunctionEnvironment = unwrapFunctionEnvironment; |
| |
| var _t = require("@babel/types"); |
| |
| var _helperEnvironmentVisitor = require("@babel/helper-environment-visitor"); |
| |
| var _helperFunctionName = require("@babel/helper-function-name"); |
| |
| var _visitors = require("../visitors"); |
| |
| const { |
| arrowFunctionExpression, |
| assignmentExpression, |
| binaryExpression, |
| blockStatement, |
| callExpression, |
| conditionalExpression, |
| expressionStatement, |
| identifier, |
| isIdentifier, |
| jsxIdentifier, |
| memberExpression, |
| metaProperty, |
| numericLiteral, |
| objectExpression, |
| restElement, |
| returnStatement, |
| sequenceExpression, |
| spreadElement, |
| stringLiteral, |
| super: _super, |
| thisExpression, |
| toExpression, |
| unaryExpression |
| } = _t; |
| |
| function toComputedKey() { |
| let key; |
| |
| if (this.isMemberExpression()) { |
| key = this.node.property; |
| } else if (this.isProperty() || this.isMethod()) { |
| key = this.node.key; |
| } else { |
| throw new ReferenceError("todo"); |
| } |
| |
| if (!this.node.computed) { |
| if (isIdentifier(key)) key = stringLiteral(key.name); |
| } |
| |
| return key; |
| } |
| |
| function ensureBlock() { |
| const body = this.get("body"); |
| const bodyNode = body.node; |
| |
| if (Array.isArray(body)) { |
| throw new Error("Can't convert array path to a block statement"); |
| } |
| |
| if (!bodyNode) { |
| throw new Error("Can't convert node without a body"); |
| } |
| |
| if (body.isBlockStatement()) { |
| return bodyNode; |
| } |
| |
| const statements = []; |
| let stringPath = "body"; |
| let key; |
| let listKey; |
| |
| if (body.isStatement()) { |
| listKey = "body"; |
| key = 0; |
| statements.push(body.node); |
| } else { |
| stringPath += ".body.0"; |
| |
| if (this.isFunction()) { |
| key = "argument"; |
| statements.push(returnStatement(body.node)); |
| } else { |
| key = "expression"; |
| statements.push(expressionStatement(body.node)); |
| } |
| } |
| |
| this.node.body = blockStatement(statements); |
| const parentPath = this.get(stringPath); |
| body.setup(parentPath, listKey ? parentPath.node[listKey] : parentPath.node, listKey, key); |
| return this.node; |
| } |
| |
| function arrowFunctionToShadowed() { |
| if (!this.isArrowFunctionExpression()) return; |
| this.arrowFunctionToExpression(); |
| } |
| |
| function unwrapFunctionEnvironment() { |
| if (!this.isArrowFunctionExpression() && !this.isFunctionExpression() && !this.isFunctionDeclaration()) { |
| throw this.buildCodeFrameError("Can only unwrap the environment of a function."); |
| } |
| |
| hoistFunctionEnvironment(this); |
| } |
| |
| function arrowFunctionToExpression({ |
| allowInsertArrow = true, |
| specCompliant = false, |
| noNewArrows = !specCompliant |
| } = {}) { |
| if (!this.isArrowFunctionExpression()) { |
| throw this.buildCodeFrameError("Cannot convert non-arrow function to a function expression."); |
| } |
| |
| const { |
| thisBinding, |
| fnPath: fn |
| } = hoistFunctionEnvironment(this, noNewArrows, allowInsertArrow); |
| fn.ensureBlock(); |
| fn.node.type = "FunctionExpression"; |
| |
| if (!noNewArrows) { |
| const checkBinding = thisBinding ? null : fn.scope.generateUidIdentifier("arrowCheckId"); |
| |
| if (checkBinding) { |
| fn.parentPath.scope.push({ |
| id: checkBinding, |
| init: objectExpression([]) |
| }); |
| } |
| |
| fn.get("body").unshiftContainer("body", expressionStatement(callExpression(this.hub.addHelper("newArrowCheck"), [thisExpression(), checkBinding ? identifier(checkBinding.name) : identifier(thisBinding)]))); |
| fn.replaceWith(callExpression(memberExpression((0, _helperFunctionName.default)(this, true) || fn.node, identifier("bind")), [checkBinding ? identifier(checkBinding.name) : thisExpression()])); |
| } |
| } |
| |
| const getSuperCallsVisitor = (0, _visitors.merge)([{ |
| CallExpression(child, { |
| allSuperCalls |
| }) { |
| if (!child.get("callee").isSuper()) return; |
| allSuperCalls.push(child); |
| } |
| |
| }, _helperEnvironmentVisitor.default]); |
| |
| function hoistFunctionEnvironment(fnPath, noNewArrows = true, allowInsertArrow = true) { |
| let arrowParent; |
| let thisEnvFn = fnPath.findParent(p => { |
| if (p.isArrowFunctionExpression()) { |
| var _arrowParent; |
| |
| (_arrowParent = arrowParent) != null ? _arrowParent : arrowParent = p; |
| return false; |
| } |
| |
| return p.isFunction() || p.isProgram() || p.isClassProperty({ |
| static: false |
| }) || p.isClassPrivateProperty({ |
| static: false |
| }); |
| }); |
| const inConstructor = thisEnvFn.isClassMethod({ |
| kind: "constructor" |
| }); |
| |
| if (thisEnvFn.isClassProperty() || thisEnvFn.isClassPrivateProperty()) { |
| if (arrowParent) { |
| thisEnvFn = arrowParent; |
| } else if (allowInsertArrow) { |
| fnPath.replaceWith(callExpression(arrowFunctionExpression([], toExpression(fnPath.node)), [])); |
| thisEnvFn = fnPath.get("callee"); |
| fnPath = thisEnvFn.get("body"); |
| } else { |
| throw fnPath.buildCodeFrameError("Unable to transform arrow inside class property"); |
| } |
| } |
| |
| const { |
| thisPaths, |
| argumentsPaths, |
| newTargetPaths, |
| superProps, |
| superCalls |
| } = getScopeInformation(fnPath); |
| |
| if (inConstructor && superCalls.length > 0) { |
| if (!allowInsertArrow) { |
| throw superCalls[0].buildCodeFrameError("Unable to handle nested super() usage in arrow"); |
| } |
| |
| const allSuperCalls = []; |
| thisEnvFn.traverse(getSuperCallsVisitor, { |
| allSuperCalls |
| }); |
| const superBinding = getSuperBinding(thisEnvFn); |
| allSuperCalls.forEach(superCall => { |
| const callee = identifier(superBinding); |
| callee.loc = superCall.node.callee.loc; |
| superCall.get("callee").replaceWith(callee); |
| }); |
| } |
| |
| if (argumentsPaths.length > 0) { |
| const argumentsBinding = getBinding(thisEnvFn, "arguments", () => { |
| const args = () => identifier("arguments"); |
| |
| if (thisEnvFn.scope.path.isProgram()) { |
| return conditionalExpression(binaryExpression("===", unaryExpression("typeof", args()), stringLiteral("undefined")), thisEnvFn.scope.buildUndefinedNode(), args()); |
| } else { |
| return args(); |
| } |
| }); |
| argumentsPaths.forEach(argumentsChild => { |
| const argsRef = identifier(argumentsBinding); |
| argsRef.loc = argumentsChild.node.loc; |
| argumentsChild.replaceWith(argsRef); |
| }); |
| } |
| |
| if (newTargetPaths.length > 0) { |
| const newTargetBinding = getBinding(thisEnvFn, "newtarget", () => metaProperty(identifier("new"), identifier("target"))); |
| newTargetPaths.forEach(targetChild => { |
| const targetRef = identifier(newTargetBinding); |
| targetRef.loc = targetChild.node.loc; |
| targetChild.replaceWith(targetRef); |
| }); |
| } |
| |
| if (superProps.length > 0) { |
| if (!allowInsertArrow) { |
| throw superProps[0].buildCodeFrameError("Unable to handle nested super.prop usage"); |
| } |
| |
| const flatSuperProps = superProps.reduce((acc, superProp) => acc.concat(standardizeSuperProperty(superProp)), []); |
| flatSuperProps.forEach(superProp => { |
| const key = superProp.node.computed ? "" : superProp.get("property").node.name; |
| const isAssignment = superProp.parentPath.isAssignmentExpression({ |
| left: superProp.node |
| }); |
| const isCall = superProp.parentPath.isCallExpression({ |
| callee: superProp.node |
| }); |
| const superBinding = getSuperPropBinding(thisEnvFn, isAssignment, key); |
| const args = []; |
| |
| if (superProp.node.computed) { |
| args.push(superProp.get("property").node); |
| } |
| |
| if (isAssignment) { |
| const value = superProp.parentPath.node.right; |
| args.push(value); |
| } |
| |
| const call = callExpression(identifier(superBinding), args); |
| |
| if (isCall) { |
| superProp.parentPath.unshiftContainer("arguments", thisExpression()); |
| superProp.replaceWith(memberExpression(call, identifier("call"))); |
| thisPaths.push(superProp.parentPath.get("arguments.0")); |
| } else if (isAssignment) { |
| superProp.parentPath.replaceWith(call); |
| } else { |
| superProp.replaceWith(call); |
| } |
| }); |
| } |
| |
| let thisBinding; |
| |
| if (thisPaths.length > 0 || !noNewArrows) { |
| thisBinding = getThisBinding(thisEnvFn, inConstructor); |
| |
| if (noNewArrows || inConstructor && hasSuperClass(thisEnvFn)) { |
| thisPaths.forEach(thisChild => { |
| const thisRef = thisChild.isJSX() ? jsxIdentifier(thisBinding) : identifier(thisBinding); |
| thisRef.loc = thisChild.node.loc; |
| thisChild.replaceWith(thisRef); |
| }); |
| if (!noNewArrows) thisBinding = null; |
| } |
| } |
| |
| return { |
| thisBinding, |
| fnPath |
| }; |
| } |
| |
| function standardizeSuperProperty(superProp) { |
| if (superProp.parentPath.isAssignmentExpression() && superProp.parentPath.node.operator !== "=") { |
| const assignmentPath = superProp.parentPath; |
| const op = assignmentPath.node.operator.slice(0, -1); |
| const value = assignmentPath.node.right; |
| assignmentPath.node.operator = "="; |
| |
| if (superProp.node.computed) { |
| const tmp = superProp.scope.generateDeclaredUidIdentifier("tmp"); |
| assignmentPath.get("left").replaceWith(memberExpression(superProp.node.object, assignmentExpression("=", tmp, superProp.node.property), true)); |
| assignmentPath.get("right").replaceWith(binaryExpression(op, memberExpression(superProp.node.object, identifier(tmp.name), true), value)); |
| } else { |
| assignmentPath.get("left").replaceWith(memberExpression(superProp.node.object, superProp.node.property)); |
| assignmentPath.get("right").replaceWith(binaryExpression(op, memberExpression(superProp.node.object, identifier(superProp.node.property.name)), value)); |
| } |
| |
| return [assignmentPath.get("left"), assignmentPath.get("right").get("left")]; |
| } else if (superProp.parentPath.isUpdateExpression()) { |
| const updateExpr = superProp.parentPath; |
| const tmp = superProp.scope.generateDeclaredUidIdentifier("tmp"); |
| const computedKey = superProp.node.computed ? superProp.scope.generateDeclaredUidIdentifier("prop") : null; |
| const parts = [assignmentExpression("=", tmp, memberExpression(superProp.node.object, computedKey ? assignmentExpression("=", computedKey, superProp.node.property) : superProp.node.property, superProp.node.computed)), assignmentExpression("=", memberExpression(superProp.node.object, computedKey ? identifier(computedKey.name) : superProp.node.property, superProp.node.computed), binaryExpression("+", identifier(tmp.name), numericLiteral(1)))]; |
| |
| if (!superProp.parentPath.node.prefix) { |
| parts.push(identifier(tmp.name)); |
| } |
| |
| updateExpr.replaceWith(sequenceExpression(parts)); |
| const left = updateExpr.get("expressions.0.right"); |
| const right = updateExpr.get("expressions.1.left"); |
| return [left, right]; |
| } |
| |
| return [superProp]; |
| } |
| |
| function hasSuperClass(thisEnvFn) { |
| return thisEnvFn.isClassMethod() && !!thisEnvFn.parentPath.parentPath.node.superClass; |
| } |
| |
| const assignSuperThisVisitor = (0, _visitors.merge)([{ |
| CallExpression(child, { |
| supers, |
| thisBinding |
| }) { |
| if (!child.get("callee").isSuper()) return; |
| if (supers.has(child.node)) return; |
| supers.add(child.node); |
| child.replaceWithMultiple([child.node, assignmentExpression("=", identifier(thisBinding), identifier("this"))]); |
| } |
| |
| }, _helperEnvironmentVisitor.default]); |
| |
| function getThisBinding(thisEnvFn, inConstructor) { |
| return getBinding(thisEnvFn, "this", thisBinding => { |
| if (!inConstructor || !hasSuperClass(thisEnvFn)) return thisExpression(); |
| thisEnvFn.traverse(assignSuperThisVisitor, { |
| supers: new WeakSet(), |
| thisBinding |
| }); |
| }); |
| } |
| |
| function getSuperBinding(thisEnvFn) { |
| return getBinding(thisEnvFn, "supercall", () => { |
| const argsBinding = thisEnvFn.scope.generateUidIdentifier("args"); |
| return arrowFunctionExpression([restElement(argsBinding)], callExpression(_super(), [spreadElement(identifier(argsBinding.name))])); |
| }); |
| } |
| |
| function getSuperPropBinding(thisEnvFn, isAssignment, propName) { |
| const op = isAssignment ? "set" : "get"; |
| return getBinding(thisEnvFn, `superprop_${op}:${propName || ""}`, () => { |
| const argsList = []; |
| let fnBody; |
| |
| if (propName) { |
| fnBody = memberExpression(_super(), identifier(propName)); |
| } else { |
| const method = thisEnvFn.scope.generateUidIdentifier("prop"); |
| argsList.unshift(method); |
| fnBody = memberExpression(_super(), identifier(method.name), true); |
| } |
| |
| if (isAssignment) { |
| const valueIdent = thisEnvFn.scope.generateUidIdentifier("value"); |
| argsList.push(valueIdent); |
| fnBody = assignmentExpression("=", fnBody, identifier(valueIdent.name)); |
| } |
| |
| return arrowFunctionExpression(argsList, fnBody); |
| }); |
| } |
| |
| function getBinding(thisEnvFn, key, init) { |
| const cacheKey = "binding:" + key; |
| let data = thisEnvFn.getData(cacheKey); |
| |
| if (!data) { |
| const id = thisEnvFn.scope.generateUidIdentifier(key); |
| data = id.name; |
| thisEnvFn.setData(cacheKey, data); |
| thisEnvFn.scope.push({ |
| id: id, |
| init: init(data) |
| }); |
| } |
| |
| return data; |
| } |
| |
| const getScopeInformationVisitor = (0, _visitors.merge)([{ |
| ThisExpression(child, { |
| thisPaths |
| }) { |
| thisPaths.push(child); |
| }, |
| |
| JSXIdentifier(child, { |
| thisPaths |
| }) { |
| if (child.node.name !== "this") return; |
| |
| if (!child.parentPath.isJSXMemberExpression({ |
| object: child.node |
| }) && !child.parentPath.isJSXOpeningElement({ |
| name: child.node |
| })) { |
| return; |
| } |
| |
| thisPaths.push(child); |
| }, |
| |
| CallExpression(child, { |
| superCalls |
| }) { |
| if (child.get("callee").isSuper()) superCalls.push(child); |
| }, |
| |
| MemberExpression(child, { |
| superProps |
| }) { |
| if (child.get("object").isSuper()) superProps.push(child); |
| }, |
| |
| Identifier(child, { |
| argumentsPaths |
| }) { |
| if (!child.isReferencedIdentifier({ |
| name: "arguments" |
| })) return; |
| let curr = child.scope; |
| |
| do { |
| if (curr.hasOwnBinding("arguments")) { |
| curr.rename("arguments"); |
| return; |
| } |
| |
| if (curr.path.isFunction() && !curr.path.isArrowFunctionExpression()) { |
| break; |
| } |
| } while (curr = curr.parent); |
| |
| argumentsPaths.push(child); |
| }, |
| |
| MetaProperty(child, { |
| newTargetPaths |
| }) { |
| if (!child.get("meta").isIdentifier({ |
| name: "new" |
| })) return; |
| if (!child.get("property").isIdentifier({ |
| name: "target" |
| })) return; |
| newTargetPaths.push(child); |
| } |
| |
| }, _helperEnvironmentVisitor.default]); |
| |
| function getScopeInformation(fnPath) { |
| const thisPaths = []; |
| const argumentsPaths = []; |
| const newTargetPaths = []; |
| const superProps = []; |
| const superCalls = []; |
| fnPath.traverse(getScopeInformationVisitor, { |
| thisPaths, |
| argumentsPaths, |
| newTargetPaths, |
| superProps, |
| superCalls |
| }); |
| return { |
| thisPaths, |
| argumentsPaths, |
| newTargetPaths, |
| superProps, |
| superCalls |
| }; |
| } |