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.
551 lines
17 KiB
551 lines
17 KiB
"use strict"; |
|
|
|
exports.__esModule = true; |
|
|
|
var _classCallCheck2 = require("babel-runtime/helpers/classCallCheck"); |
|
|
|
var _classCallCheck3 = _interopRequireDefault(_classCallCheck2); |
|
|
|
var _getIterator2 = require("babel-runtime/core-js/get-iterator"); |
|
|
|
var _getIterator3 = _interopRequireDefault(_getIterator2); |
|
|
|
exports.default = function (_ref) { |
|
var t = _ref.types; |
|
|
|
|
|
function variableDeclarationHasPattern(node) { |
|
for (var _iterator = node.declarations, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : (0, _getIterator3.default)(_iterator);;) { |
|
var _ref2; |
|
|
|
if (_isArray) { |
|
if (_i >= _iterator.length) break; |
|
_ref2 = _iterator[_i++]; |
|
} else { |
|
_i = _iterator.next(); |
|
if (_i.done) break; |
|
_ref2 = _i.value; |
|
} |
|
|
|
var declar = _ref2; |
|
|
|
if (t.isPattern(declar.id)) { |
|
return true; |
|
} |
|
} |
|
return false; |
|
} |
|
|
|
function hasRest(pattern) { |
|
for (var _iterator2 = pattern.elements, _isArray2 = Array.isArray(_iterator2), _i2 = 0, _iterator2 = _isArray2 ? _iterator2 : (0, _getIterator3.default)(_iterator2);;) { |
|
var _ref3; |
|
|
|
if (_isArray2) { |
|
if (_i2 >= _iterator2.length) break; |
|
_ref3 = _iterator2[_i2++]; |
|
} else { |
|
_i2 = _iterator2.next(); |
|
if (_i2.done) break; |
|
_ref3 = _i2.value; |
|
} |
|
|
|
var elem = _ref3; |
|
|
|
if (t.isRestElement(elem)) { |
|
return true; |
|
} |
|
} |
|
return false; |
|
} |
|
|
|
var arrayUnpackVisitor = { |
|
ReferencedIdentifier: function ReferencedIdentifier(path, state) { |
|
if (state.bindings[path.node.name]) { |
|
state.deopt = true; |
|
path.stop(); |
|
} |
|
} |
|
}; |
|
|
|
var DestructuringTransformer = function () { |
|
function DestructuringTransformer(opts) { |
|
(0, _classCallCheck3.default)(this, DestructuringTransformer); |
|
|
|
this.blockHoist = opts.blockHoist; |
|
this.operator = opts.operator; |
|
this.arrays = {}; |
|
this.nodes = opts.nodes || []; |
|
this.scope = opts.scope; |
|
this.file = opts.file; |
|
this.kind = opts.kind; |
|
} |
|
|
|
DestructuringTransformer.prototype.buildVariableAssignment = function buildVariableAssignment(id, init) { |
|
var op = this.operator; |
|
if (t.isMemberExpression(id)) op = "="; |
|
|
|
var node = void 0; |
|
|
|
if (op) { |
|
node = t.expressionStatement(t.assignmentExpression(op, id, init)); |
|
} else { |
|
node = t.variableDeclaration(this.kind, [t.variableDeclarator(id, init)]); |
|
} |
|
|
|
node._blockHoist = this.blockHoist; |
|
|
|
return node; |
|
}; |
|
|
|
DestructuringTransformer.prototype.buildVariableDeclaration = function buildVariableDeclaration(id, init) { |
|
var declar = t.variableDeclaration("var", [t.variableDeclarator(id, init)]); |
|
declar._blockHoist = this.blockHoist; |
|
return declar; |
|
}; |
|
|
|
DestructuringTransformer.prototype.push = function push(id, init) { |
|
if (t.isObjectPattern(id)) { |
|
this.pushObjectPattern(id, init); |
|
} else if (t.isArrayPattern(id)) { |
|
this.pushArrayPattern(id, init); |
|
} else if (t.isAssignmentPattern(id)) { |
|
this.pushAssignmentPattern(id, init); |
|
} else { |
|
this.nodes.push(this.buildVariableAssignment(id, init)); |
|
} |
|
}; |
|
|
|
DestructuringTransformer.prototype.toArray = function toArray(node, count) { |
|
if (this.file.opts.loose || t.isIdentifier(node) && this.arrays[node.name]) { |
|
return node; |
|
} else { |
|
return this.scope.toArray(node, count); |
|
} |
|
}; |
|
|
|
DestructuringTransformer.prototype.pushAssignmentPattern = function pushAssignmentPattern(pattern, valueRef) { |
|
|
|
var tempValueRef = this.scope.generateUidIdentifierBasedOnNode(valueRef); |
|
|
|
var declar = t.variableDeclaration("var", [t.variableDeclarator(tempValueRef, valueRef)]); |
|
declar._blockHoist = this.blockHoist; |
|
this.nodes.push(declar); |
|
|
|
var tempConditional = t.conditionalExpression(t.binaryExpression("===", tempValueRef, t.identifier("undefined")), pattern.right, tempValueRef); |
|
|
|
var left = pattern.left; |
|
if (t.isPattern(left)) { |
|
var tempValueDefault = t.expressionStatement(t.assignmentExpression("=", tempValueRef, tempConditional)); |
|
tempValueDefault._blockHoist = this.blockHoist; |
|
|
|
this.nodes.push(tempValueDefault); |
|
this.push(left, tempValueRef); |
|
} else { |
|
this.nodes.push(this.buildVariableAssignment(left, tempConditional)); |
|
} |
|
}; |
|
|
|
DestructuringTransformer.prototype.pushObjectRest = function pushObjectRest(pattern, objRef, spreadProp, spreadPropIndex) { |
|
|
|
var keys = []; |
|
|
|
for (var i = 0; i < pattern.properties.length; i++) { |
|
var prop = pattern.properties[i]; |
|
|
|
if (i >= spreadPropIndex) break; |
|
|
|
if (t.isRestProperty(prop)) continue; |
|
|
|
var key = prop.key; |
|
if (t.isIdentifier(key) && !prop.computed) key = t.stringLiteral(prop.key.name); |
|
keys.push(key); |
|
} |
|
|
|
keys = t.arrayExpression(keys); |
|
|
|
var value = t.callExpression(this.file.addHelper("objectWithoutProperties"), [objRef, keys]); |
|
this.nodes.push(this.buildVariableAssignment(spreadProp.argument, value)); |
|
}; |
|
|
|
DestructuringTransformer.prototype.pushObjectProperty = function pushObjectProperty(prop, propRef) { |
|
if (t.isLiteral(prop.key)) prop.computed = true; |
|
|
|
var pattern = prop.value; |
|
var objRef = t.memberExpression(propRef, prop.key, prop.computed); |
|
|
|
if (t.isPattern(pattern)) { |
|
this.push(pattern, objRef); |
|
} else { |
|
this.nodes.push(this.buildVariableAssignment(pattern, objRef)); |
|
} |
|
}; |
|
|
|
DestructuringTransformer.prototype.pushObjectPattern = function pushObjectPattern(pattern, objRef) { |
|
|
|
if (!pattern.properties.length) { |
|
this.nodes.push(t.expressionStatement(t.callExpression(this.file.addHelper("objectDestructuringEmpty"), [objRef]))); |
|
} |
|
|
|
if (pattern.properties.length > 1 && !this.scope.isStatic(objRef)) { |
|
var temp = this.scope.generateUidIdentifierBasedOnNode(objRef); |
|
this.nodes.push(this.buildVariableDeclaration(temp, objRef)); |
|
objRef = temp; |
|
} |
|
|
|
for (var i = 0; i < pattern.properties.length; i++) { |
|
var prop = pattern.properties[i]; |
|
if (t.isRestProperty(prop)) { |
|
this.pushObjectRest(pattern, objRef, prop, i); |
|
} else { |
|
this.pushObjectProperty(prop, objRef); |
|
} |
|
} |
|
}; |
|
|
|
DestructuringTransformer.prototype.canUnpackArrayPattern = function canUnpackArrayPattern(pattern, arr) { |
|
if (!t.isArrayExpression(arr)) return false; |
|
|
|
if (pattern.elements.length > arr.elements.length) return; |
|
if (pattern.elements.length < arr.elements.length && !hasRest(pattern)) return false; |
|
|
|
for (var _iterator3 = pattern.elements, _isArray3 = Array.isArray(_iterator3), _i3 = 0, _iterator3 = _isArray3 ? _iterator3 : (0, _getIterator3.default)(_iterator3);;) { |
|
var _ref4; |
|
|
|
if (_isArray3) { |
|
if (_i3 >= _iterator3.length) break; |
|
_ref4 = _iterator3[_i3++]; |
|
} else { |
|
_i3 = _iterator3.next(); |
|
if (_i3.done) break; |
|
_ref4 = _i3.value; |
|
} |
|
|
|
var elem = _ref4; |
|
|
|
if (!elem) return false; |
|
|
|
if (t.isMemberExpression(elem)) return false; |
|
} |
|
|
|
for (var _iterator4 = arr.elements, _isArray4 = Array.isArray(_iterator4), _i4 = 0, _iterator4 = _isArray4 ? _iterator4 : (0, _getIterator3.default)(_iterator4);;) { |
|
var _ref5; |
|
|
|
if (_isArray4) { |
|
if (_i4 >= _iterator4.length) break; |
|
_ref5 = _iterator4[_i4++]; |
|
} else { |
|
_i4 = _iterator4.next(); |
|
if (_i4.done) break; |
|
_ref5 = _i4.value; |
|
} |
|
|
|
var _elem = _ref5; |
|
|
|
if (t.isSpreadElement(_elem)) return false; |
|
|
|
if (t.isCallExpression(_elem)) return false; |
|
|
|
if (t.isMemberExpression(_elem)) return false; |
|
} |
|
|
|
var bindings = t.getBindingIdentifiers(pattern); |
|
var state = { deopt: false, bindings: bindings }; |
|
this.scope.traverse(arr, arrayUnpackVisitor, state); |
|
return !state.deopt; |
|
}; |
|
|
|
DestructuringTransformer.prototype.pushUnpackedArrayPattern = function pushUnpackedArrayPattern(pattern, arr) { |
|
for (var i = 0; i < pattern.elements.length; i++) { |
|
var elem = pattern.elements[i]; |
|
if (t.isRestElement(elem)) { |
|
this.push(elem.argument, t.arrayExpression(arr.elements.slice(i))); |
|
} else { |
|
this.push(elem, arr.elements[i]); |
|
} |
|
} |
|
}; |
|
|
|
DestructuringTransformer.prototype.pushArrayPattern = function pushArrayPattern(pattern, arrayRef) { |
|
if (!pattern.elements) return; |
|
|
|
if (this.canUnpackArrayPattern(pattern, arrayRef)) { |
|
return this.pushUnpackedArrayPattern(pattern, arrayRef); |
|
} |
|
|
|
var count = !hasRest(pattern) && pattern.elements.length; |
|
|
|
var toArray = this.toArray(arrayRef, count); |
|
|
|
if (t.isIdentifier(toArray)) { |
|
arrayRef = toArray; |
|
} else { |
|
arrayRef = this.scope.generateUidIdentifierBasedOnNode(arrayRef); |
|
this.arrays[arrayRef.name] = true; |
|
this.nodes.push(this.buildVariableDeclaration(arrayRef, toArray)); |
|
} |
|
|
|
for (var i = 0; i < pattern.elements.length; i++) { |
|
var elem = pattern.elements[i]; |
|
|
|
if (!elem) continue; |
|
|
|
var elemRef = void 0; |
|
|
|
if (t.isRestElement(elem)) { |
|
elemRef = this.toArray(arrayRef); |
|
elemRef = t.callExpression(t.memberExpression(elemRef, t.identifier("slice")), [t.numericLiteral(i)]); |
|
|
|
elem = elem.argument; |
|
} else { |
|
elemRef = t.memberExpression(arrayRef, t.numericLiteral(i), true); |
|
} |
|
|
|
this.push(elem, elemRef); |
|
} |
|
}; |
|
|
|
DestructuringTransformer.prototype.init = function init(pattern, ref) { |
|
|
|
if (!t.isArrayExpression(ref) && !t.isMemberExpression(ref)) { |
|
var memo = this.scope.maybeGenerateMemoised(ref, true); |
|
if (memo) { |
|
this.nodes.push(this.buildVariableDeclaration(memo, ref)); |
|
ref = memo; |
|
} |
|
} |
|
|
|
this.push(pattern, ref); |
|
|
|
return this.nodes; |
|
}; |
|
|
|
return DestructuringTransformer; |
|
}(); |
|
|
|
return { |
|
visitor: { |
|
ExportNamedDeclaration: function ExportNamedDeclaration(path) { |
|
var declaration = path.get("declaration"); |
|
if (!declaration.isVariableDeclaration()) return; |
|
if (!variableDeclarationHasPattern(declaration.node)) return; |
|
|
|
var specifiers = []; |
|
|
|
for (var name in path.getOuterBindingIdentifiers(path)) { |
|
var id = t.identifier(name); |
|
specifiers.push(t.exportSpecifier(id, id)); |
|
} |
|
|
|
path.replaceWith(declaration.node); |
|
path.insertAfter(t.exportNamedDeclaration(null, specifiers)); |
|
}, |
|
ForXStatement: function ForXStatement(path, file) { |
|
var node = path.node, |
|
scope = path.scope; |
|
|
|
var left = node.left; |
|
|
|
if (t.isPattern(left)) { |
|
|
|
var temp = scope.generateUidIdentifier("ref"); |
|
|
|
node.left = t.variableDeclaration("var", [t.variableDeclarator(temp)]); |
|
|
|
path.ensureBlock(); |
|
|
|
node.body.body.unshift(t.variableDeclaration("var", [t.variableDeclarator(left, temp)])); |
|
|
|
return; |
|
} |
|
|
|
if (!t.isVariableDeclaration(left)) return; |
|
|
|
var pattern = left.declarations[0].id; |
|
if (!t.isPattern(pattern)) return; |
|
|
|
var key = scope.generateUidIdentifier("ref"); |
|
node.left = t.variableDeclaration(left.kind, [t.variableDeclarator(key, null)]); |
|
|
|
var nodes = []; |
|
|
|
var destructuring = new DestructuringTransformer({ |
|
kind: left.kind, |
|
file: file, |
|
scope: scope, |
|
nodes: nodes |
|
}); |
|
|
|
destructuring.init(pattern, key); |
|
|
|
path.ensureBlock(); |
|
|
|
var block = node.body; |
|
block.body = nodes.concat(block.body); |
|
}, |
|
CatchClause: function CatchClause(_ref6, file) { |
|
var node = _ref6.node, |
|
scope = _ref6.scope; |
|
|
|
var pattern = node.param; |
|
if (!t.isPattern(pattern)) return; |
|
|
|
var ref = scope.generateUidIdentifier("ref"); |
|
node.param = ref; |
|
|
|
var nodes = []; |
|
|
|
var destructuring = new DestructuringTransformer({ |
|
kind: "let", |
|
file: file, |
|
scope: scope, |
|
nodes: nodes |
|
}); |
|
destructuring.init(pattern, ref); |
|
|
|
node.body.body = nodes.concat(node.body.body); |
|
}, |
|
AssignmentExpression: function AssignmentExpression(path, file) { |
|
var node = path.node, |
|
scope = path.scope; |
|
|
|
if (!t.isPattern(node.left)) return; |
|
|
|
var nodes = []; |
|
|
|
var destructuring = new DestructuringTransformer({ |
|
operator: node.operator, |
|
file: file, |
|
scope: scope, |
|
nodes: nodes |
|
}); |
|
|
|
var ref = void 0; |
|
if (path.isCompletionRecord() || !path.parentPath.isExpressionStatement()) { |
|
ref = scope.generateUidIdentifierBasedOnNode(node.right, "ref"); |
|
|
|
nodes.push(t.variableDeclaration("var", [t.variableDeclarator(ref, node.right)])); |
|
|
|
if (t.isArrayExpression(node.right)) { |
|
destructuring.arrays[ref.name] = true; |
|
} |
|
} |
|
|
|
destructuring.init(node.left, ref || node.right); |
|
|
|
if (ref) { |
|
nodes.push(t.expressionStatement(ref)); |
|
} |
|
|
|
path.replaceWithMultiple(nodes); |
|
}, |
|
VariableDeclaration: function VariableDeclaration(path, file) { |
|
var node = path.node, |
|
scope = path.scope, |
|
parent = path.parent; |
|
|
|
if (t.isForXStatement(parent)) return; |
|
if (!parent || !path.container) return; |
|
if (!variableDeclarationHasPattern(node)) return; |
|
|
|
var nodes = []; |
|
var declar = void 0; |
|
|
|
for (var i = 0; i < node.declarations.length; i++) { |
|
declar = node.declarations[i]; |
|
|
|
var patternId = declar.init; |
|
var pattern = declar.id; |
|
|
|
var destructuring = new DestructuringTransformer({ |
|
blockHoist: node._blockHoist, |
|
nodes: nodes, |
|
scope: scope, |
|
kind: node.kind, |
|
file: file |
|
}); |
|
|
|
if (t.isPattern(pattern)) { |
|
destructuring.init(pattern, patternId); |
|
|
|
if (+i !== node.declarations.length - 1) { |
|
t.inherits(nodes[nodes.length - 1], declar); |
|
} |
|
} else { |
|
nodes.push(t.inherits(destructuring.buildVariableAssignment(declar.id, declar.init), declar)); |
|
} |
|
} |
|
|
|
var nodesOut = []; |
|
for (var _iterator5 = nodes, _isArray5 = Array.isArray(_iterator5), _i5 = 0, _iterator5 = _isArray5 ? _iterator5 : (0, _getIterator3.default)(_iterator5);;) { |
|
var _ref7; |
|
|
|
if (_isArray5) { |
|
if (_i5 >= _iterator5.length) break; |
|
_ref7 = _iterator5[_i5++]; |
|
} else { |
|
_i5 = _iterator5.next(); |
|
if (_i5.done) break; |
|
_ref7 = _i5.value; |
|
} |
|
|
|
var _node = _ref7; |
|
|
|
var tail = nodesOut[nodesOut.length - 1]; |
|
if (tail && t.isVariableDeclaration(tail) && t.isVariableDeclaration(_node) && tail.kind === _node.kind) { |
|
var _tail$declarations; |
|
|
|
(_tail$declarations = tail.declarations).push.apply(_tail$declarations, _node.declarations); |
|
} else { |
|
nodesOut.push(_node); |
|
} |
|
} |
|
|
|
for (var _iterator6 = nodesOut, _isArray6 = Array.isArray(_iterator6), _i6 = 0, _iterator6 = _isArray6 ? _iterator6 : (0, _getIterator3.default)(_iterator6);;) { |
|
var _ref8; |
|
|
|
if (_isArray6) { |
|
if (_i6 >= _iterator6.length) break; |
|
_ref8 = _iterator6[_i6++]; |
|
} else { |
|
_i6 = _iterator6.next(); |
|
if (_i6.done) break; |
|
_ref8 = _i6.value; |
|
} |
|
|
|
var nodeOut = _ref8; |
|
|
|
if (!nodeOut.declarations) continue; |
|
for (var _iterator7 = nodeOut.declarations, _isArray7 = Array.isArray(_iterator7), _i7 = 0, _iterator7 = _isArray7 ? _iterator7 : (0, _getIterator3.default)(_iterator7);;) { |
|
var _ref9; |
|
|
|
if (_isArray7) { |
|
if (_i7 >= _iterator7.length) break; |
|
_ref9 = _iterator7[_i7++]; |
|
} else { |
|
_i7 = _iterator7.next(); |
|
if (_i7.done) break; |
|
_ref9 = _i7.value; |
|
} |
|
|
|
var declaration = _ref9; |
|
var name = declaration.id.name; |
|
|
|
if (scope.bindings[name]) { |
|
scope.bindings[name].kind = nodeOut.kind; |
|
} |
|
} |
|
} |
|
|
|
if (nodesOut.length === 1) { |
|
path.replaceWith(nodesOut[0]); |
|
} else { |
|
path.replaceWithMultiple(nodesOut); |
|
} |
|
} |
|
} |
|
}; |
|
}; |
|
|
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } |
|
|
|
module.exports = exports["default"]; |