123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551 |
- "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"];
|