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.
199 lines
4.4 KiB
199 lines
4.4 KiB
"use strict"; |
|
|
|
Object.defineProperty(exports, "__esModule", { |
|
value: true |
|
}); |
|
exports.default = _default; |
|
|
|
var _template = require("@babel/template"); |
|
|
|
var _t = require("@babel/types"); |
|
|
|
const { |
|
NOT_LOCAL_BINDING, |
|
cloneNode, |
|
identifier, |
|
isAssignmentExpression, |
|
isAssignmentPattern, |
|
isFunction, |
|
isIdentifier, |
|
isLiteral, |
|
isNullLiteral, |
|
isObjectMethod, |
|
isObjectProperty, |
|
isRegExpLiteral, |
|
isRestElement, |
|
isTemplateLiteral, |
|
isVariableDeclarator, |
|
toBindingIdentifierName |
|
} = _t; |
|
|
|
function getFunctionArity(node) { |
|
const count = node.params.findIndex(param => isAssignmentPattern(param) || isRestElement(param)); |
|
return count === -1 ? node.params.length : count; |
|
} |
|
|
|
const buildPropertyMethodAssignmentWrapper = _template.default.statement(` |
|
(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 = _template.default.statement(` |
|
(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 = getFunctionArity(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), |
|
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, supportUnicodeId = 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; |
|
} |
|
|
|
if (!supportUnicodeId && isFunction(node) && /[\uD800-\uDFFF]/.test(name)) { |
|
return; |
|
} |
|
|
|
name = toBindingIdentifierName(name); |
|
const newId = identifier(name); |
|
newId[NOT_LOCAL_BINDING] = true; |
|
const state = visit(node, name, scope); |
|
return wrap(state, node, newId, scope) || node; |
|
} |