index.js 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. "use strict";
  2. exports.__esModule = true;
  3. var _getIterator2 = require("babel-runtime/core-js/get-iterator");
  4. var _getIterator3 = _interopRequireDefault(_getIterator2);
  5. exports.default = function (path, emit) {
  6. var kind = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : "var";
  7. path.traverse(visitor, { kind: kind, emit: emit });
  8. };
  9. var _babelTypes = require("babel-types");
  10. var t = _interopRequireWildcard(_babelTypes);
  11. 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; } }
  12. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  13. var visitor = {
  14. Scope: function Scope(path, state) {
  15. if (state.kind === "let") path.skip();
  16. },
  17. Function: function Function(path) {
  18. path.skip();
  19. },
  20. VariableDeclaration: function VariableDeclaration(path, state) {
  21. if (state.kind && path.node.kind !== state.kind) return;
  22. var nodes = [];
  23. var declarations = path.get("declarations");
  24. var firstId = void 0;
  25. for (var _iterator = declarations, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : (0, _getIterator3.default)(_iterator);;) {
  26. var _ref;
  27. if (_isArray) {
  28. if (_i >= _iterator.length) break;
  29. _ref = _iterator[_i++];
  30. } else {
  31. _i = _iterator.next();
  32. if (_i.done) break;
  33. _ref = _i.value;
  34. }
  35. var declar = _ref;
  36. firstId = declar.node.id;
  37. if (declar.node.init) {
  38. nodes.push(t.expressionStatement(t.assignmentExpression("=", declar.node.id, declar.node.init)));
  39. }
  40. for (var name in declar.getBindingIdentifiers()) {
  41. state.emit(t.identifier(name), name);
  42. }
  43. }
  44. if (path.parentPath.isFor({ left: path.node })) {
  45. path.replaceWith(firstId);
  46. } else {
  47. path.replaceWithMultiple(nodes);
  48. }
  49. }
  50. };
  51. module.exports = exports["default"];