198 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			198 lines
		
	
	
		
			4.9 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| "use strict";
 | |
| 
 | |
| Object.defineProperty(exports, "__esModule", {
 | |
|   value: true
 | |
| });
 | |
| exports.default = _default;
 | |
| 
 | |
| function _helperGetFunctionArity() {
 | |
|   const data = _interopRequireDefault(require("@babel/helper-get-function-arity"));
 | |
| 
 | |
|   _helperGetFunctionArity = function () {
 | |
|     return data;
 | |
|   };
 | |
| 
 | |
|   return data;
 | |
| }
 | |
| 
 | |
| function _template() {
 | |
|   const data = _interopRequireDefault(require("@babel/template"));
 | |
| 
 | |
|   _template = function () {
 | |
|     return data;
 | |
|   };
 | |
| 
 | |
|   return data;
 | |
| }
 | |
| 
 | |
| function t() {
 | |
|   const data = _interopRequireWildcard(require("@babel/types"));
 | |
| 
 | |
|   t = function () {
 | |
|     return data;
 | |
|   };
 | |
| 
 | |
|   return data;
 | |
| }
 | |
| 
 | |
| function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
 | |
| 
 | |
| function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 | |
| 
 | |
| 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 (t().isNullLiteral(id)) {
 | |
|     return "null";
 | |
|   }
 | |
| 
 | |
|   if (t().isRegExpLiteral(id)) {
 | |
|     return `_${id.pattern}_${id.flags}`;
 | |
|   }
 | |
| 
 | |
|   if (t().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 (!t().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 ((t().isObjectProperty(parent) || t().isObjectMethod(parent, {
 | |
|     kind: "method"
 | |
|   })) && (!parent.computed || t().isLiteral(parent.key))) {
 | |
|     id = parent.key;
 | |
|   } else if (t().isVariableDeclarator(parent)) {
 | |
|     id = parent.id;
 | |
| 
 | |
|     if (t().isIdentifier(id) && !localBinding) {
 | |
|       const binding = scope.parent.getBinding(id.name);
 | |
| 
 | |
|       if (binding && binding.constant && scope.getBinding(id.name) === binding) {
 | |
|         node.id = t().cloneNode(id);
 | |
|         node.id[t().NOT_LOCAL_BINDING] = true;
 | |
|         return;
 | |
|       }
 | |
|     }
 | |
|   } else if (t().isAssignmentExpression(parent)) {
 | |
|     id = parent.left;
 | |
|   } else if (!id) {
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   let name;
 | |
| 
 | |
|   if (id && t().isLiteral(id)) {
 | |
|     name = getNameFromLiteralId(id);
 | |
|   } else if (id && t().isIdentifier(id)) {
 | |
|     name = id.name;
 | |
|   }
 | |
| 
 | |
|   if (name === undefined) {
 | |
|     return;
 | |
|   }
 | |
| 
 | |
|   name = t().toBindingIdentifierName(name);
 | |
|   id = t().identifier(name);
 | |
|   id[t().NOT_LOCAL_BINDING] = true;
 | |
|   const state = visit(node, name, scope);
 | |
|   return wrap(state, node, id, scope) || node;
 | |
| } |