| "use strict"; |
| |
| Object.defineProperty(exports, "__esModule", { |
| value: true |
| }); |
| exports.default = _default; |
| |
| var _helperGetFunctionArity = require("@babel/helper-get-function-arity"); |
| |
| var _template = require("@babel/template"); |
| |
| var _t = require("@babel/types"); |
| |
| const { |
| NOT_LOCAL_BINDING, |
| cloneNode, |
| identifier, |
| isAssignmentExpression, |
| isFunction, |
| isIdentifier, |
| isLiteral, |
| isNullLiteral, |
| isObjectMethod, |
| isObjectProperty, |
| isRegExpLiteral, |
| isTemplateLiteral, |
| isVariableDeclarator, |
| toBindingIdentifierName |
| } = _t; |
| const buildPropertyMethodAssignmentWrapper = (0, _template.default)(` |
| (function (FUNCTION_KEY) { |
| function FUNCTION_ID() { |
| return FUNCTION_KEY.apply(this, arguments); |
| } |
| |
| FUNCTION_ID.toString = function () { |
| return FUNCTION_KEY.toString(); |
| } |
| |
| return FUNCTION_ID; |
| })(FUNCTION) |
| `); |
| const buildGeneratorPropertyMethodAssignmentWrapper = (0, _template.default)(` |
| (function (FUNCTION_KEY) { |
| function* FUNCTION_ID() { |
| return yield* FUNCTION_KEY.apply(this, arguments); |
| } |
| |
| FUNCTION_ID.toString = function () { |
| return FUNCTION_KEY.toString(); |
| }; |
| |
| return FUNCTION_ID; |
| })(FUNCTION) |
| `); |
| const visitor = { |
| "ReferencedIdentifier|BindingIdentifier"(path, state) { |
| if (path.node.name !== state.name) return; |
| const localDeclar = path.scope.getBindingIdentifier(state.name); |
| if (localDeclar !== state.outerDeclar) return; |
| state.selfReference = true; |
| path.stop(); |
| } |
| |
| }; |
| |
| function getNameFromLiteralId(id) { |
| if (isNullLiteral(id)) { |
| return "null"; |
| } |
| |
| if (isRegExpLiteral(id)) { |
| return `_${id.pattern}_${id.flags}`; |
| } |
| |
| if (isTemplateLiteral(id)) { |
| return id.quasis.map(quasi => quasi.value.raw).join(""); |
| } |
| |
| if (id.value !== undefined) { |
| return id.value + ""; |
| } |
| |
| return ""; |
| } |
| |
| function wrap(state, method, id, scope) { |
| if (state.selfReference) { |
| if (scope.hasBinding(id.name) && !scope.hasGlobal(id.name)) { |
| scope.rename(id.name); |
| } else { |
| if (!isFunction(method)) return; |
| let build = buildPropertyMethodAssignmentWrapper; |
| |
| if (method.generator) { |
| build = buildGeneratorPropertyMethodAssignmentWrapper; |
| } |
| |
| const template = build({ |
| FUNCTION: method, |
| FUNCTION_ID: id, |
| FUNCTION_KEY: scope.generateUidIdentifier(id.name) |
| }).expression; |
| const params = template.callee.body.body[0].params; |
| |
| for (let i = 0, len = (0, _helperGetFunctionArity.default)(method); i < len; i++) { |
| params.push(scope.generateUidIdentifier("x")); |
| } |
| |
| return template; |
| } |
| } |
| |
| method.id = id; |
| scope.getProgramParent().references[id.name] = true; |
| } |
| |
| function visit(node, name, scope) { |
| const state = { |
| selfAssignment: false, |
| selfReference: false, |
| outerDeclar: scope.getBindingIdentifier(name), |
| references: [], |
| name: name |
| }; |
| const binding = scope.getOwnBinding(name); |
| |
| if (binding) { |
| if (binding.kind === "param") { |
| state.selfReference = true; |
| } else {} |
| } else if (state.outerDeclar || scope.hasGlobal(name)) { |
| scope.traverse(node, visitor, state); |
| } |
| |
| return state; |
| } |
| |
| function _default({ |
| node, |
| parent, |
| scope, |
| id |
| }, localBinding = false) { |
| if (node.id) return; |
| |
| if ((isObjectProperty(parent) || isObjectMethod(parent, { |
| kind: "method" |
| })) && (!parent.computed || isLiteral(parent.key))) { |
| id = parent.key; |
| } else if (isVariableDeclarator(parent)) { |
| id = parent.id; |
| |
| if (isIdentifier(id) && !localBinding) { |
| const binding = scope.parent.getBinding(id.name); |
| |
| if (binding && binding.constant && scope.getBinding(id.name) === binding) { |
| node.id = cloneNode(id); |
| node.id[NOT_LOCAL_BINDING] = true; |
| return; |
| } |
| } |
| } else if (isAssignmentExpression(parent, { |
| operator: "=" |
| })) { |
| id = parent.left; |
| } else if (!id) { |
| return; |
| } |
| |
| let name; |
| |
| if (id && isLiteral(id)) { |
| name = getNameFromLiteralId(id); |
| } else if (id && isIdentifier(id)) { |
| name = id.name; |
| } |
| |
| if (name === undefined) { |
| return; |
| } |
| |
| name = toBindingIdentifierName(name); |
| id = identifier(name); |
| id[NOT_LOCAL_BINDING] = true; |
| const state = visit(node, name, scope); |
| return wrap(state, node, id, scope) || node; |
| } |