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.
		
		
		
		
		
			
		
			
				
					
					
						
							188 lines
						
					
					
						
							4.2 KiB
						
					
					
				
			
		
		
	
	
							188 lines
						
					
					
						
							4.2 KiB
						
					
					
				"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; | 
						|
} |