@babel/traverse 7.0.0-beta.31

package/lib/path/context.js

@@ -0,0 +1,269 @@
+"use strict";
+
+exports.__esModule = true;
+exports.call = call;
+exports._call = _call;
+exports.isBlacklisted = isBlacklisted;
+exports.visit = visit;
+exports.skip = skip;
+exports.skipKey = skipKey;
+exports.stop = stop;
+exports.setScope = setScope;
+exports.setContext = setContext;
+exports.resync = resync;
+exports._resyncParent = _resyncParent;
+exports._resyncKey = _resyncKey;
+exports._resyncList = _resyncList;
+exports._resyncRemoved = _resyncRemoved;
+exports.popContext = popContext;
+exports.pushContext = pushContext;
+exports.setup = setup;
+exports.setKey = setKey;
+exports.requeue = requeue;
+exports._getQueueContexts = _getQueueContexts;
+
+var _index = _interopRequireDefault(require("../index"));
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+function call(key) {
+  var opts = this.opts;
+  this.debug(key);
+
+  if (this.node) {
+    if (this._call(opts[key])) return true;
+  }
+
+  if (this.node) {
+    return this._call(opts[this.node.type] && opts[this.node.type][key]);
+  }
+
+  return false;
+}
+
+function _call(fns) {
+  if (!fns) return false;
+
+  for (var _iterator = fns, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : _iterator[Symbol.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 _fn = _ref;
+    if (!_fn) continue;
+    var node = this.node;
+    if (!node) return true;
+
+    var ret = _fn.call(this.state, this, this.state);
+
+    if (ret) {
+      throw new Error("Unexpected return value from visitor method " + _fn);
+    }
+
+    if (this.node !== node) return true;
+    if (this.shouldStop || this.shouldSkip || this.removed) return true;
+  }
+
+  return false;
+}
+
+function isBlacklisted() {
+  var blacklist = this.opts.blacklist;
+  return blacklist && blacklist.indexOf(this.node.type) > -1;
+}
+
+function visit() {
+  if (!this.node) {
+    return false;
+  }
+
+  if (this.isBlacklisted()) {
+    return false;
+  }
+
+  if (this.opts.shouldSkip && this.opts.shouldSkip(this)) {
+    return false;
+  }
+
+  if (this.call("enter") || this.shouldSkip) {
+    this.debug("Skip...");
+    return this.shouldStop;
+  }
+
+  this.debug("Recursing into...");
+
+  _index.default.node(this.node, this.opts, this.scope, this.state, this, this.skipKeys);
+
+  this.call("exit");
+  return this.shouldStop;
+}
+
+function skip() {
+  this.shouldSkip = true;
+}
+
+function skipKey(key) {
+  this.skipKeys[key] = true;
+}
+
+function stop() {
+  this.shouldStop = true;
+  this.shouldSkip = true;
+}
+
+function setScope() {
+  if (this.opts && this.opts.noScope) return;
+  var path = this.parentPath;
+  var target;
+
+  while (path && !target) {
+    if (path.opts && path.opts.noScope) return;
+    target = path.scope;
+    path = path.parentPath;
+  }
+
+  this.scope = this.getScope(target);
+  if (this.scope) this.scope.init();
+}
+
+function setContext(context) {
+  this.shouldSkip = false;
+  this.shouldStop = false;
+  this.removed = false;
+  this.skipKeys = {};
+
+  if (context) {
+    this.context = context;
+    this.state = context.state;
+    this.opts = context.opts;
+  }
+
+  this.setScope();
+  return this;
+}
+
+function resync() {
+  if (this.removed) return;
+
+  this._resyncParent();
+
+  this._resyncList();
+
+  this._resyncKey();
+}
+
+function _resyncParent() {
+  if (this.parentPath) {
+    this.parent = this.parentPath.node;
+  }
+}
+
+function _resyncKey() {
+  if (!this.container) return;
+  if (this.node === this.container[this.key]) return;
+
+  if (Array.isArray(this.container)) {
+    for (var i = 0; i < this.container.length; i++) {
+      if (this.container[i] === this.node) {
+        return this.setKey(i);
+      }
+    }
+  } else {
+    for (var key in this.container) {
+      if (this.container[key] === this.node) {
+        return this.setKey(key);
+      }
+    }
+  }
+
+  this.key = null;
+}
+
+function _resyncList() {
+  if (!this.parent || !this.inList) return;
+  var newContainer = this.parent[this.listKey];
+  if (this.container === newContainer) return;
+  this.container = newContainer || null;
+}
+
+function _resyncRemoved() {
+  if (this.key == null || !this.container || this.container[this.key] !== this.node) {
+    this._markRemoved();
+  }
+}
+
+function popContext() {
+  this.contexts.pop();
+
+  if (this.contexts.length > 0) {
+    this.setContext(this.contexts[this.contexts.length - 1]);
+  } else {
+    this.setContext(undefined);
+  }
+}
+
+function pushContext(context) {
+  this.contexts.push(context);
+  this.setContext(context);
+}
+
+function setup(parentPath, container, listKey, key) {
+  this.inList = !!listKey;
+  this.listKey = listKey;
+  this.parentKey = listKey || key;
+  this.container = container;
+  this.parentPath = parentPath || this.parentPath;
+  this.setKey(key);
+}
+
+function setKey(key) {
+  this.key = key;
+  this.node = this.container[this.key];
+  this.type = this.node && this.node.type;
+}
+
+function requeue(pathToQueue) {
+  if (pathToQueue === void 0) {
+    pathToQueue = this;
+  }
+
+  if (pathToQueue.removed) return;
+  var contexts = this.contexts;
+
+  for (var _iterator2 = contexts, _isArray2 = Array.isArray(_iterator2), _i2 = 0, _iterator2 = _isArray2 ? _iterator2 : _iterator2[Symbol.iterator]();;) {
+    var _ref2;
+
+    if (_isArray2) {
+      if (_i2 >= _iterator2.length) break;
+      _ref2 = _iterator2[_i2++];
+    } else {
+      _i2 = _iterator2.next();
+      if (_i2.done) break;
+      _ref2 = _i2.value;
+    }
+
+    var _context = _ref2;
+
+    _context.maybeQueue(pathToQueue);
+  }
+}
+
+function _getQueueContexts() {
+  var path = this;
+  var contexts = this.contexts;
+
+  while (!contexts.length) {
+    path = path.parentPath;
+    if (!path) break;
+    contexts = path.contexts;
+  }
+
+  return contexts;
+}

package/lib/path/conversion.js

@@ -0,0 +1,446 @@
+"use strict";
+
+exports.__esModule = true;
+exports.toComputedKey = toComputedKey;
+exports.ensureBlock = ensureBlock;
+exports.arrowFunctionToShadowed = arrowFunctionToShadowed;
+exports.unwrapFunctionEnvironment = unwrapFunctionEnvironment;
+exports.arrowFunctionToExpression = arrowFunctionToExpression;
+
+var t = _interopRequireWildcard(require("@babel/types"));
+
+var _helperFunctionName = _interopRequireDefault(require("@babel/helper-function-name"));
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+
+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 toComputedKey() {
+  var node = this.node;
+  var key;
+
+  if (this.isMemberExpression()) {
+    key = node.property;
+  } else if (this.isProperty() || this.isMethod()) {
+    key = node.key;
+  } else {
+    throw new ReferenceError("todo");
+  }
+
+  if (!node.computed) {
+    if (t.isIdentifier(key)) key = t.stringLiteral(key.name);
+  }
+
+  return key;
+}
+
+function ensureBlock() {
+  var body = this.get("body");
+  var bodyNode = body.node;
+
+  if (Array.isArray(body)) {
+    throw new Error("Can't convert array path to a block statement");
+  }
+
+  if (!bodyNode) {
+    throw new Error("Can't convert node without a body");
+  }
+
+  if (body.isBlockStatement()) {
+    return bodyNode;
+  }
+
+  var statements = [];
+  var stringPath = "body";
+  var key;
+  var listKey;
+
+  if (body.isStatement()) {
+    listKey = "body";
+    key = 0;
+    statements.push(body.node);
+  } else {
+    stringPath += ".body.0";
+
+    if (this.isFunction()) {
+      key = "argument";
+      statements.push(t.returnStatement(body.node));
+    } else {
+      key = "expression";
+      statements.push(t.expressionStatement(body.node));
+    }
+  }
+
+  this.node.body = t.blockStatement(statements);
+  var parentPath = this.get(stringPath);
+  body.setup(parentPath, listKey ? parentPath.node[listKey] : parentPath.node, listKey, key);
+  return this.node;
+}
+
+function arrowFunctionToShadowed() {
+  if (!this.isArrowFunctionExpression()) return;
+  this.arrowFunctionToExpression();
+}
+
+function unwrapFunctionEnvironment() {
+  if (!this.isArrowFunctionExpression() && !this.isFunctionExpression() && !this.isFunctionDeclaration()) {
+    throw this.buildCodeFrameError("Can only unwrap the environment of a function.");
+  }
+
+  hoistFunctionEnvironment(this);
+}
+
+function arrowFunctionToExpression(_temp) {
+  var _ref = _temp === void 0 ? {} : _temp,
+      _ref$allowInsertArrow = _ref.allowInsertArrow,
+      allowInsertArrow = _ref$allowInsertArrow === void 0 ? true : _ref$allowInsertArrow,
+      _ref$specCompliant = _ref.specCompliant,
+      specCompliant = _ref$specCompliant === void 0 ? false : _ref$specCompliant;
+
+  if (!this.isArrowFunctionExpression()) {
+    throw this.buildCodeFrameError("Cannot convert non-arrow function to a function expression.");
+  }
+
+  var thisBinding = hoistFunctionEnvironment(this, specCompliant, allowInsertArrow);
+  this.ensureBlock();
+  this.node.type = "FunctionExpression";
+
+  if (specCompliant) {
+    var checkBinding = thisBinding ? null : this.parentPath.scope.generateUidIdentifier("arrowCheckId");
+
+    if (checkBinding) {
+      this.parentPath.scope.push({
+        id: checkBinding,
+        init: t.objectExpression([])
+      });
+    }
+
+    this.get("body").unshiftContainer("body", t.expressionStatement(t.callExpression(this.hub.file.addHelper("newArrowCheck"), [t.thisExpression(), checkBinding ? t.identifier(checkBinding.name) : t.identifier(thisBinding)])));
+    this.replaceWith(t.callExpression(t.memberExpression((0, _helperFunctionName.default)(this) || this.node, t.identifier("bind")), [checkBinding ? t.identifier(checkBinding.name) : t.thisExpression()]));
+  }
+}
+
+function hoistFunctionEnvironment(fnPath, specCompliant, allowInsertArrow) {
+  if (specCompliant === void 0) {
+    specCompliant = false;
+  }
+
+  if (allowInsertArrow === void 0) {
+    allowInsertArrow = true;
+  }
+
+  var thisEnvFn = fnPath.findParent(function (p) {
+    return p.isFunction() && !p.isArrowFunctionExpression() || p.isProgram() || p.isClassProperty({
+      static: false
+    });
+  });
+  var inConstructor = thisEnvFn && thisEnvFn.node.kind === "constructor";
+
+  if (thisEnvFn.isClassProperty()) {
+    throw fnPath.buildCodeFrameError("Unable to transform arrow inside class property");
+  }
+
+  var _getScopeInformation = getScopeInformation(fnPath),
+      thisPaths = _getScopeInformation.thisPaths,
+      argumentsPaths = _getScopeInformation.argumentsPaths,
+      newTargetPaths = _getScopeInformation.newTargetPaths,
+      superProps = _getScopeInformation.superProps,
+      superCalls = _getScopeInformation.superCalls;
+
+  if (inConstructor && superCalls.length > 0) {
+    if (!allowInsertArrow) {
+      throw superCalls[0].buildCodeFrameError("Unable to handle nested super() usage in arrow");
+    }
+
+    var allSuperCalls = [];
+    thisEnvFn.traverse({
+      Function: function Function(child) {
+        if (child.isArrowFunctionExpression()) return;
+        child.skip();
+      },
+      ClassProperty: function ClassProperty(child) {
+        if (child.node.static) return;
+        child.skip();
+      },
+      CallExpression: function CallExpression(child) {
+        if (!child.get("callee").isSuper()) return;
+        allSuperCalls.push(child);
+      }
+    });
+    var superBinding = getSuperBinding(thisEnvFn);
+    allSuperCalls.forEach(function (superCall) {
+      return superCall.get("callee").replaceWith(t.identifier(superBinding));
+    });
+  }
+
+  var thisBinding;
+
+  if (thisPaths.length > 0 || specCompliant) {
+    thisBinding = getThisBinding(thisEnvFn, inConstructor);
+
+    if (!specCompliant || inConstructor && hasSuperClass(thisEnvFn)) {
+      thisPaths.forEach(function (thisChild) {
+        thisChild.replaceWith(thisChild.isJSX() ? t.jSXIdentifier(thisBinding) : t.identifier(thisBinding));
+      });
+      if (specCompliant) thisBinding = null;
+    }
+  }
+
+  if (argumentsPaths.length > 0) {
+    var argumentsBinding = getBinding(thisEnvFn, "arguments", function () {
+      return t.identifier("arguments");
+    });
+    argumentsPaths.forEach(function (argumentsChild) {
+      argumentsChild.replaceWith(t.identifier(argumentsBinding));
+    });
+  }
+
+  if (newTargetPaths.length > 0) {
+    var newTargetBinding = getBinding(thisEnvFn, "newtarget", function () {
+      return t.metaProperty(t.identifier("new"), t.identifier("target"));
+    });
+    newTargetPaths.forEach(function (argumentsChild) {
+      argumentsChild.replaceWith(t.identifier(newTargetBinding));
+    });
+  }
+
+  if (superProps.length > 0) {
+    if (!allowInsertArrow) {
+      throw superProps[0].buildCodeFrameError("Unable to handle nested super.prop usage");
+    }
+
+    var flatSuperProps = superProps.reduce(function (acc, superProp) {
+      return acc.concat(standardizeSuperProperty(superProp));
+    }, []);
+    flatSuperProps.forEach(function (superProp) {
+      var key = superProp.node.computed ? "" : superProp.get("property").node.name;
+
+      if (superProp.parentPath.isCallExpression({
+        callee: superProp.node
+      })) {
+        var _superBinding = getSuperPropCallBinding(thisEnvFn, key);
+
+        if (superProp.node.computed) {
+          var prop = superProp.get("property").node;
+          superProp.replaceWith(t.identifier(_superBinding));
+          superProp.parentPath.node.arguments.unshift(prop);
+        } else {
+          superProp.replaceWith(t.identifier(_superBinding));
+        }
+      } else {
+        var isAssignment = superProp.parentPath.isAssignmentExpression({
+          left: superProp.node
+        });
+
+        var _superBinding2 = getSuperPropBinding(thisEnvFn, isAssignment, key);
+
+        var args = [];
+
+        if (superProp.node.computed) {
+          args.push(superProp.get("property").node);
+        }
+
+        if (isAssignment) {
+          var value = superProp.parentPath.node.right;
+          args.push(value);
+          superProp.parentPath.replaceWith(t.callExpression(t.identifier(_superBinding2), args));
+        } else {
+          superProp.replaceWith(t.callExpression(t.identifier(_superBinding2), args));
+        }
+      }
+    });
+  }
+
+  return thisBinding;
+}
+
+function standardizeSuperProperty(superProp) {
+  if (superProp.parentPath.isAssignmentExpression() && superProp.parentPath.node.operator !== "=") {
+    var assignmentPath = superProp.parentPath;
+    var op = assignmentPath.node.operator.slice(0, -1);
+    var value = assignmentPath.node.right;
+    assignmentPath.node.operator = "=";
+
+    if (superProp.node.computed) {
+      var tmp = superProp.scope.generateDeclaredUidIdentifier("tmp");
+      assignmentPath.get("left").replaceWith(t.memberExpression(superProp.node.object, t.assignmentExpression("=", tmp, superProp.node.property), true));
+      assignmentPath.get("right").replaceWith(t.binaryExpression(op, t.memberExpression(superProp.node.object, t.identifier(tmp.name), true), value));
+    } else {
+      assignmentPath.get("left").replaceWith(t.memberExpression(superProp.node.object, superProp.node.property));
+      assignmentPath.get("right").replaceWith(t.binaryExpression(op, t.memberExpression(superProp.node.object, t.identifier(superProp.node.property.name)), value));
+    }
+
+    return [assignmentPath.get("left"), assignmentPath.get("right").get("left")];
+  } else if (superProp.parentPath.isUpdateExpression()) {
+    var updateExpr = superProp.parentPath;
+
+    var _tmp = superProp.scope.generateDeclaredUidIdentifier("tmp");
+
+    var computedKey = superProp.node.computed ? superProp.scope.generateDeclaredUidIdentifier("prop") : null;
+    var parts = [t.assignmentExpression("=", _tmp, t.memberExpression(superProp.node.object, computedKey ? t.assignmentExpression("=", computedKey, superProp.node.property) : superProp.node.property, superProp.node.computed)), t.assignmentExpression("=", t.memberExpression(superProp.node.object, computedKey ? t.identifier(computedKey.name) : superProp.node.property, superProp.node.computed), t.binaryExpression("+", t.identifier(_tmp.name), t.numericLiteral(1)))];
+
+    if (!superProp.parentPath.node.prefix) {
+      parts.push(t.identifier(_tmp.name));
+    }
+
+    updateExpr.replaceWith(t.sequenceExpression(parts));
+    var left = updateExpr.get("expressions.0.right");
+    var right = updateExpr.get("expressions.1.left");
+    return [left, right];
+  }
+
+  return [superProp];
+}
+
+function hasSuperClass(thisEnvFn) {
+  return thisEnvFn.isClassMethod() && !!thisEnvFn.parentPath.parentPath.node.superClass;
+}
+
+function getThisBinding(thisEnvFn, inConstructor) {
+  return getBinding(thisEnvFn, "this", function (thisBinding) {
+    if (!inConstructor || !hasSuperClass(thisEnvFn)) return t.thisExpression();
+    var supers = new WeakSet();
+    thisEnvFn.traverse({
+      Function: function Function(child) {
+        if (child.isArrowFunctionExpression()) return;
+        child.skip();
+      },
+      ClassProperty: function ClassProperty(child) {
+        if (child.node.static) return;
+        child.skip();
+      },
+      CallExpression: function CallExpression(child) {
+        if (!child.get("callee").isSuper()) return;
+        if (supers.has(child.node)) return;
+        supers.add(child.node);
+        child.replaceWith(t.assignmentExpression("=", t.identifier(thisBinding), child.node));
+      }
+    });
+  });
+}
+
+function getSuperBinding(thisEnvFn) {
+  return getBinding(thisEnvFn, "supercall", function () {
+    var argsBinding = thisEnvFn.scope.generateUidIdentifier("args");
+    return t.arrowFunctionExpression([t.restElement(argsBinding)], t.callExpression(t.super(), [t.spreadElement(t.identifier(argsBinding.name))]));
+  });
+}
+
+function getSuperPropCallBinding(thisEnvFn, propName) {
+  return getBinding(thisEnvFn, "superprop_call:" + (propName || ""), function () {
+    var argsBinding = thisEnvFn.scope.generateUidIdentifier("args");
+    var argsList = [t.restElement(argsBinding)];
+    var fnBody;
+
+    if (propName) {
+      fnBody = t.callExpression(t.memberExpression(t.super(), t.identifier(propName)), [t.spreadElement(t.identifier(argsBinding.name))]);
+    } else {
+      var method = thisEnvFn.scope.generateUidIdentifier("prop");
+      argsList.unshift(method);
+      fnBody = t.callExpression(t.memberExpression(t.super(), t.identifier(method.name), true), [t.spreadElement(t.identifier(argsBinding.name))]);
+    }
+
+    return t.arrowFunctionExpression(argsList, fnBody);
+  });
+}
+
+function getSuperPropBinding(thisEnvFn, isAssignment, propName) {
+  var op = isAssignment ? "set" : "get";
+  return getBinding(thisEnvFn, "superprop_" + op + ":" + (propName || ""), function () {
+    var argsList = [];
+    var fnBody;
+
+    if (propName) {
+      fnBody = t.memberExpression(t.super(), t.identifier(propName));
+    } else {
+      var method = thisEnvFn.scope.generateUidIdentifier("prop");
+      argsList.unshift(method);
+      fnBody = t.memberExpression(t.super(), t.identifier(method.name), true);
+    }
+
+    if (isAssignment) {
+      var valueIdent = thisEnvFn.scope.generateUidIdentifier("value");
+      argsList.push(valueIdent);
+      fnBody = t.assignmentExpression("=", fnBody, t.identifier(valueIdent.name));
+    }
+
+    return t.arrowFunctionExpression(argsList, fnBody);
+  });
+}
+
+function getBinding(thisEnvFn, key, init) {
+  var cacheKey = "binding:" + key;
+  var data = thisEnvFn.getData(cacheKey);
+
+  if (!data) {
+    var id = thisEnvFn.scope.generateUidIdentifier(key);
+    data = id.name;
+    thisEnvFn.setData(cacheKey, data);
+    thisEnvFn.scope.push({
+      id: id,
+      init: init(data)
+    });
+  }
+
+  return data;
+}
+
+function getScopeInformation(fnPath) {
+  var thisPaths = [];
+  var argumentsPaths = [];
+  var newTargetPaths = [];
+  var superProps = [];
+  var superCalls = [];
+  fnPath.traverse({
+    ClassProperty: function ClassProperty(child) {
+      if (child.node.static) return;
+      child.skip();
+    },
+    Function: function Function(child) {
+      if (child.isArrowFunctionExpression()) return;
+      child.skip();
+    },
+    ThisExpression: function ThisExpression(child) {
+      thisPaths.push(child);
+    },
+    JSXIdentifier: function JSXIdentifier(child) {
+      if (child.node.name !== "this") return;
+
+      if (!child.parentPath.isJSXMemberExpression({
+        object: child.node
+      }) && !child.parentPath.isJSXOpeningElement({
+        name: child.node
+      })) {
+        return;
+      }
+
+      thisPaths.push(child);
+    },
+    CallExpression: function CallExpression(child) {
+      if (child.get("callee").isSuper()) superCalls.push(child);
+    },
+    MemberExpression: function MemberExpression(child) {
+      if (child.get("object").isSuper()) superProps.push(child);
+    },
+    ReferencedIdentifier: function ReferencedIdentifier(child) {
+      if (child.node.name !== "arguments") return;
+      argumentsPaths.push(child);
+    },
+    MetaProperty: function MetaProperty(child) {
+      if (!child.get("meta").isIdentifier({
+        name: "new"
+      })) return;
+      if (!child.get("property").isIdentifier({
+        name: "target"
+      })) return;
+      newTargetPaths.push(child);
+    }
+  });
+  return {
+    thisPaths: thisPaths,
+    argumentsPaths: argumentsPaths,
+    newTargetPaths: newTargetPaths,
+    superProps: superProps,
+    superCalls: superCalls
+  };
+}