| "use strict"; |
| |
| Object.defineProperty(exports, "__esModule", { |
| value: true |
| }); |
| exports.toComputedKey = toComputedKey; |
| exports.ensureBlock = ensureBlock; |
| exports.arrowFunctionToShadowed = arrowFunctionToShadowed; |
| exports.unwrapFunctionEnvironment = unwrapFunctionEnvironment; |
| exports.arrowFunctionToExpression = arrowFunctionToExpression; |
| |
| var t = _interopRequireWildcard(require("@babel/types")); |
| |
| var _helperFunctionName = _interopRequireDefault(require("@babel/helper-function-name")); |
| |
| function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } |
| |
| function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function () { return cache; }; return cache; } |
| |
| function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; if (obj != null) { var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; } |
| |
| function toComputedKey() { |
| const node = this.node; |
| let key; |
| |
| if (this.isMemberExpression()) { |
| key = node.property; |
| } else if (this.isProperty() || this.isMethod()) { |
| key = node.key; |
| } else { |
| throw new ReferenceError("todo"); |
| } |
| |
| if (!node.computed) { |
| if (t.isIdentifier(key)) key = t.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(t.returnStatement(body.node)); |
| } else { |
| key = "expression"; |
| statements.push(t.expressionStatement(body.node)); |
| } |
| } |
| |
| this.node.body = t.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 |
| } = {}) { |
| if (!this.isArrowFunctionExpression()) { |
| throw this.buildCodeFrameError("Cannot convert non-arrow function to a function expression."); |
| } |
| |
| const thisBinding = hoistFunctionEnvironment(this, specCompliant, allowInsertArrow); |
| this.ensureBlock(); |
| this.node.type = "FunctionExpression"; |
| |
| if (specCompliant) { |
| const checkBinding = thisBinding ? null : this.parentPath.scope.generateUidIdentifier("arrowCheckId"); |
| |
| if (checkBinding) { |
| this.parentPath.scope.push({ |
| id: checkBinding, |
| init: t.objectExpression([]) |
| }); |
| } |
| |
| this.get("body").unshiftContainer("body", t.expressionStatement(t.callExpression(this.hub.addHelper("newArrowCheck"), [t.thisExpression(), checkBinding ? t.identifier(checkBinding.name) : t.identifier(thisBinding)]))); |
| this.replaceWith(t.callExpression(t.memberExpression((0, _helperFunctionName.default)(this, true) || this.node, t.identifier("bind")), [checkBinding ? t.identifier(checkBinding.name) : t.thisExpression()])); |
| } |
| } |
| |
| function hoistFunctionEnvironment(fnPath, specCompliant = false, allowInsertArrow = true) { |
| const thisEnvFn = fnPath.findParent(p => { |
| return p.isFunction() && !p.isArrowFunctionExpression() || p.isProgram() || p.isClassProperty({ |
| static: false |
| }); |
| }); |
| const inConstructor = thisEnvFn && thisEnvFn.node.kind === "constructor"; |
| |
| if (thisEnvFn.isClassProperty()) { |
| 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({ |
| Function(child) { |
| if (child.isArrowFunctionExpression()) return; |
| child.skip(); |
| }, |
| |
| ClassProperty(child) { |
| child.skip(); |
| }, |
| |
| CallExpression(child) { |
| if (!child.get("callee").isSuper()) return; |
| allSuperCalls.push(child); |
| } |
| |
| }); |
| const superBinding = getSuperBinding(thisEnvFn); |
| allSuperCalls.forEach(superCall => { |
| const callee = t.identifier(superBinding); |
| callee.loc = superCall.node.callee.loc; |
| superCall.get("callee").replaceWith(callee); |
| }); |
| } |
| |
| let thisBinding; |
| |
| if (thisPaths.length > 0 || specCompliant) { |
| thisBinding = getThisBinding(thisEnvFn, inConstructor); |
| |
| if (!specCompliant || inConstructor && hasSuperClass(thisEnvFn)) { |
| thisPaths.forEach(thisChild => { |
| const thisRef = thisChild.isJSX() ? t.jsxIdentifier(thisBinding) : t.identifier(thisBinding); |
| thisRef.loc = thisChild.node.loc; |
| thisChild.replaceWith(thisRef); |
| }); |
| if (specCompliant) thisBinding = null; |
| } |
| } |
| |
| if (argumentsPaths.length > 0) { |
| const argumentsBinding = getBinding(thisEnvFn, "arguments", () => t.identifier("arguments")); |
| argumentsPaths.forEach(argumentsChild => { |
| const argsRef = t.identifier(argumentsBinding); |
| argsRef.loc = argumentsChild.node.loc; |
| argumentsChild.replaceWith(argsRef); |
| }); |
| } |
| |
| if (newTargetPaths.length > 0) { |
| const newTargetBinding = getBinding(thisEnvFn, "newtarget", () => t.metaProperty(t.identifier("new"), t.identifier("target"))); |
| newTargetPaths.forEach(targetChild => { |
| const targetRef = t.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; |
| |
| if (superProp.parentPath.isCallExpression({ |
| callee: superProp.node |
| })) { |
| const superBinding = getSuperPropCallBinding(thisEnvFn, key); |
| |
| if (superProp.node.computed) { |
| const prop = superProp.get("property").node; |
| superProp.replaceWith(t.identifier(superBinding)); |
| superProp.parentPath.node.arguments.unshift(prop); |
| } else { |
| superProp.replaceWith(t.identifier(superBinding)); |
| } |
| } else { |
| const isAssignment = superProp.parentPath.isAssignmentExpression({ |
| left: 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); |
| superProp.parentPath.replaceWith(t.callExpression(t.identifier(superBinding), args)); |
| } else { |
| superProp.replaceWith(t.callExpression(t.identifier(superBinding), args)); |
| } |
| } |
| }); |
| } |
| |
| return thisBinding; |
| } |
| |
| 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(t.memberExpression(superProp.node.object, t.assignmentExpression("=", tmp, superProp.node.property), true)); |
| assignmentPath.get("right").replaceWith(t.binaryExpression(op, t.memberExpression(superProp.node.object, t.identifier(tmp.name), true), value)); |
| } else { |
| assignmentPath.get("left").replaceWith(t.memberExpression(superProp.node.object, superProp.node.property)); |
| assignmentPath.get("right").replaceWith(t.binaryExpression(op, t.memberExpression(superProp.node.object, t.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 = [t.assignmentExpression("=", tmp, t.memberExpression(superProp.node.object, computedKey ? t.assignmentExpression("=", computedKey, superProp.node.property) : superProp.node.property, superProp.node.computed)), t.assignmentExpression("=", t.memberExpression(superProp.node.object, computedKey ? t.identifier(computedKey.name) : superProp.node.property, superProp.node.computed), t.binaryExpression("+", t.identifier(tmp.name), t.numericLiteral(1)))]; |
| |
| if (!superProp.parentPath.node.prefix) { |
| parts.push(t.identifier(tmp.name)); |
| } |
| |
| updateExpr.replaceWith(t.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; |
| } |
| |
| function getThisBinding(thisEnvFn, inConstructor) { |
| return getBinding(thisEnvFn, "this", thisBinding => { |
| if (!inConstructor || !hasSuperClass(thisEnvFn)) return t.thisExpression(); |
| const supers = new WeakSet(); |
| thisEnvFn.traverse({ |
| Function(child) { |
| if (child.isArrowFunctionExpression()) return; |
| child.skip(); |
| }, |
| |
| ClassProperty(child) { |
| child.skip(); |
| }, |
| |
| CallExpression(child) { |
| if (!child.get("callee").isSuper()) return; |
| if (supers.has(child.node)) return; |
| supers.add(child.node); |
| child.replaceWithMultiple([child.node, t.assignmentExpression("=", t.identifier(thisBinding), t.identifier("this"))]); |
| } |
| |
| }); |
| }); |
| } |
| |
| function getSuperBinding(thisEnvFn) { |
| return getBinding(thisEnvFn, "supercall", () => { |
| const argsBinding = thisEnvFn.scope.generateUidIdentifier("args"); |
| return t.arrowFunctionExpression([t.restElement(argsBinding)], t.callExpression(t.super(), [t.spreadElement(t.identifier(argsBinding.name))])); |
| }); |
| } |
| |
| function getSuperPropCallBinding(thisEnvFn, propName) { |
| return getBinding(thisEnvFn, `superprop_call:${propName || ""}`, () => { |
| const argsBinding = thisEnvFn.scope.generateUidIdentifier("args"); |
| const argsList = [t.restElement(argsBinding)]; |
| let fnBody; |
| |
| if (propName) { |
| fnBody = t.callExpression(t.memberExpression(t.super(), t.identifier(propName)), [t.spreadElement(t.identifier(argsBinding.name))]); |
| } else { |
| const method = thisEnvFn.scope.generateUidIdentifier("prop"); |
| argsList.unshift(method); |
| fnBody = t.callExpression(t.memberExpression(t.super(), t.identifier(method.name), true), [t.spreadElement(t.identifier(argsBinding.name))]); |
| } |
| |
| return t.arrowFunctionExpression(argsList, fnBody); |
| }); |
| } |
| |
| function getSuperPropBinding(thisEnvFn, isAssignment, propName) { |
| const op = isAssignment ? "set" : "get"; |
| return getBinding(thisEnvFn, `superprop_${op}:${propName || ""}`, () => { |
| const argsList = []; |
| let fnBody; |
| |
| if (propName) { |
| fnBody = t.memberExpression(t.super(), t.identifier(propName)); |
| } else { |
| const method = thisEnvFn.scope.generateUidIdentifier("prop"); |
| argsList.unshift(method); |
| fnBody = t.memberExpression(t.super(), t.identifier(method.name), true); |
| } |
| |
| if (isAssignment) { |
| const valueIdent = thisEnvFn.scope.generateUidIdentifier("value"); |
| argsList.push(valueIdent); |
| fnBody = t.assignmentExpression("=", fnBody, t.identifier(valueIdent.name)); |
| } |
| |
| return t.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; |
| } |
| |
| function getScopeInformation(fnPath) { |
| const thisPaths = []; |
| const argumentsPaths = []; |
| const newTargetPaths = []; |
| const superProps = []; |
| const superCalls = []; |
| fnPath.traverse({ |
| ClassProperty(child) { |
| child.skip(); |
| }, |
| |
| Function(child) { |
| if (child.isArrowFunctionExpression()) return; |
| child.skip(); |
| }, |
| |
| ThisExpression(child) { |
| thisPaths.push(child); |
| }, |
| |
| JSXIdentifier(child) { |
| 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) { |
| if (child.get("callee").isSuper()) superCalls.push(child); |
| }, |
| |
| MemberExpression(child) { |
| if (child.get("object").isSuper()) superProps.push(child); |
| }, |
| |
| ReferencedIdentifier(child) { |
| if (child.node.name !== "arguments") return; |
| argumentsPaths.push(child); |
| }, |
| |
| MetaProperty(child) { |
| if (!child.get("meta").isIdentifier({ |
| name: "new" |
| })) return; |
| if (!child.get("property").isIdentifier({ |
| name: "target" |
| })) return; |
| newTargetPaths.push(child); |
| } |
| |
| }); |
| return { |
| thisPaths, |
| argumentsPaths, |
| newTargetPaths, |
| superProps, |
| superCalls |
| }; |
| } |