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.
162 lines
4.6 KiB
162 lines
4.6 KiB
"use strict"; |
|
|
|
exports.__esModule = true; |
|
exports.visitor = undefined; |
|
|
|
var _getIterator2 = require("babel-runtime/core-js/get-iterator"); |
|
|
|
var _getIterator3 = _interopRequireDefault(_getIterator2); |
|
|
|
var _babelHelperGetFunctionArity = require("babel-helper-get-function-arity"); |
|
|
|
var _babelHelperGetFunctionArity2 = _interopRequireDefault(_babelHelperGetFunctionArity); |
|
|
|
var _babelHelperCallDelegate = require("babel-helper-call-delegate"); |
|
|
|
var _babelHelperCallDelegate2 = _interopRequireDefault(_babelHelperCallDelegate); |
|
|
|
var _babelTemplate = require("babel-template"); |
|
|
|
var _babelTemplate2 = _interopRequireDefault(_babelTemplate); |
|
|
|
var _babelTypes = require("babel-types"); |
|
|
|
var t = _interopRequireWildcard(_babelTypes); |
|
|
|
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)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } } |
|
|
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } |
|
|
|
var buildDefaultParam = (0, _babelTemplate2.default)("\n let VARIABLE_NAME =\n ARGUMENTS.length > ARGUMENT_KEY && ARGUMENTS[ARGUMENT_KEY] !== undefined ?\n ARGUMENTS[ARGUMENT_KEY]\n :\n DEFAULT_VALUE;\n"); |
|
|
|
var buildCutOff = (0, _babelTemplate2.default)("\n let $0 = $1[$2];\n"); |
|
|
|
function hasDefaults(node) { |
|
for (var _iterator = node.params, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : (0, _getIterator3.default)(_iterator);;) { |
|
var _ref; |
|
|
|
if (_isArray) { |
|
if (_i >= _iterator.length) break; |
|
_ref = _iterator[_i++]; |
|
} else { |
|
_i = _iterator.next(); |
|
if (_i.done) break; |
|
_ref = _i.value; |
|
} |
|
|
|
var param = _ref; |
|
|
|
if (!t.isIdentifier(param)) return true; |
|
} |
|
return false; |
|
} |
|
|
|
function isSafeBinding(scope, node) { |
|
if (!scope.hasOwnBinding(node.name)) return true; |
|
|
|
var _scope$getOwnBinding = scope.getOwnBinding(node.name), |
|
kind = _scope$getOwnBinding.kind; |
|
|
|
return kind === "param" || kind === "local"; |
|
} |
|
|
|
var iifeVisitor = { |
|
ReferencedIdentifier: function ReferencedIdentifier(path, state) { |
|
var scope = path.scope, |
|
node = path.node; |
|
|
|
if (node.name === "eval" || !isSafeBinding(scope, node)) { |
|
state.iife = true; |
|
path.stop(); |
|
} |
|
}, |
|
Scope: function Scope(path) { |
|
path.skip(); |
|
} |
|
}; |
|
|
|
var visitor = exports.visitor = { |
|
Function: function Function(path) { |
|
var node = path.node, |
|
scope = path.scope; |
|
|
|
if (!hasDefaults(node)) return; |
|
|
|
path.ensureBlock(); |
|
|
|
var state = { |
|
iife: false, |
|
scope: scope |
|
}; |
|
|
|
var body = []; |
|
|
|
var argsIdentifier = t.identifier("arguments"); |
|
argsIdentifier._shadowedFunctionLiteral = path; |
|
|
|
function pushDefNode(left, right, i) { |
|
var defNode = buildDefaultParam({ |
|
VARIABLE_NAME: left, |
|
DEFAULT_VALUE: right, |
|
ARGUMENT_KEY: t.numericLiteral(i), |
|
ARGUMENTS: argsIdentifier |
|
}); |
|
defNode._blockHoist = node.params.length - i; |
|
body.push(defNode); |
|
} |
|
|
|
var lastNonDefaultParam = (0, _babelHelperGetFunctionArity2.default)(node); |
|
|
|
var params = path.get("params"); |
|
for (var i = 0; i < params.length; i++) { |
|
var param = params[i]; |
|
|
|
if (!param.isAssignmentPattern()) { |
|
if (!state.iife && !param.isIdentifier()) { |
|
param.traverse(iifeVisitor, state); |
|
} |
|
|
|
continue; |
|
} |
|
|
|
var left = param.get("left"); |
|
var right = param.get("right"); |
|
|
|
if (i >= lastNonDefaultParam || left.isPattern()) { |
|
var placeholder = scope.generateUidIdentifier("x"); |
|
placeholder._isDefaultPlaceholder = true; |
|
node.params[i] = placeholder; |
|
} else { |
|
node.params[i] = left.node; |
|
} |
|
|
|
if (!state.iife) { |
|
if (right.isIdentifier() && !isSafeBinding(scope, right.node)) { |
|
state.iife = true; |
|
} else { |
|
right.traverse(iifeVisitor, state); |
|
} |
|
} |
|
|
|
pushDefNode(left.node, right.node, i); |
|
} |
|
|
|
for (var _i2 = lastNonDefaultParam + 1; _i2 < node.params.length; _i2++) { |
|
var _param = node.params[_i2]; |
|
if (_param._isDefaultPlaceholder) continue; |
|
|
|
var declar = buildCutOff(_param, argsIdentifier, t.numericLiteral(_i2)); |
|
declar._blockHoist = node.params.length - _i2; |
|
body.push(declar); |
|
} |
|
|
|
node.params = node.params.slice(0, lastNonDefaultParam); |
|
|
|
if (state.iife) { |
|
body.push((0, _babelHelperCallDelegate2.default)(path, scope)); |
|
path.set("body", t.blockStatement(body)); |
|
} else { |
|
path.get("body").unshiftContainer("body", body); |
|
} |
|
} |
|
}; |