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.
		
		
		
		
		
			
		
			
				
					202 lines
				
				4.5 KiB
			
		
		
			
		
	
	
					202 lines
				
				4.5 KiB
			| 
											3 years ago
										 | "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; | ||
|  | } | ||
|  | 
 | ||
|  | //# sourceMappingURL=index.js.map
 |