index.js 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. "use strict";
  2. exports.__esModule = true;
  3. exports.definitions = undefined;
  4. exports.default = function (_ref) {
  5. var t = _ref.types;
  6. var RUNTIME_MODULE_NAME = "babel-runtime";
  7. function has(obj, key) {
  8. return Object.prototype.hasOwnProperty.call(obj, key);
  9. }
  10. var HELPER_BLACKLIST = ["interopRequireWildcard", "interopRequireDefault"];
  11. return {
  12. pre: function pre(file) {
  13. if (this.opts.helpers !== false) {
  14. file.set("helperGenerator", function (name) {
  15. if (HELPER_BLACKLIST.indexOf(name) < 0) {
  16. return file.addImport(RUNTIME_MODULE_NAME + "/helpers/" + name, "default", name);
  17. }
  18. });
  19. }
  20. this.setDynamic("regeneratorIdentifier", function () {
  21. return file.addImport(RUNTIME_MODULE_NAME + "/regenerator", "default", "regeneratorRuntime");
  22. });
  23. },
  24. visitor: {
  25. ReferencedIdentifier: function ReferencedIdentifier(path, state) {
  26. var node = path.node;
  27. var parent = path.parent;
  28. var scope = path.scope;
  29. if (node.name === "regeneratorRuntime" && state.opts.regenerator !== false) {
  30. path.replaceWith(state.get("regeneratorIdentifier"));
  31. return;
  32. }
  33. if (state.opts.polyfill === false) return;
  34. if (t.isMemberExpression(parent)) return;
  35. if (!has(_definitions2.default.builtins, node.name)) return;
  36. if (scope.getBindingIdentifier(node.name)) return;
  37. // Symbol() -> _core.Symbol(); new Promise -> new _core.Promise
  38. path.replaceWith(state.addImport(RUNTIME_MODULE_NAME + "/core-js/" + _definitions2.default.builtins[node.name], "default", node.name));
  39. },
  40. // arr[Symbol.iterator]() -> _core.$for.getIterator(arr)
  41. CallExpression: function CallExpression(path, state) {
  42. if (state.opts.polyfill === false) return;
  43. // we can't compile this
  44. if (path.node.arguments.length) return;
  45. var callee = path.node.callee;
  46. if (!t.isMemberExpression(callee)) return;
  47. if (!callee.computed) return;
  48. if (!path.get("callee.property").matchesPattern("Symbol.iterator")) return;
  49. path.replaceWith(t.callExpression(state.addImport(RUNTIME_MODULE_NAME + "/core-js/get-iterator", "default", "getIterator"), [callee.object]));
  50. },
  51. // Symbol.iterator in arr -> core.$for.isIterable(arr)
  52. BinaryExpression: function BinaryExpression(path, state) {
  53. if (state.opts.polyfill === false) return;
  54. if (path.node.operator !== "in") return;
  55. if (!path.get("left").matchesPattern("Symbol.iterator")) return;
  56. path.replaceWith(t.callExpression(state.addImport(RUNTIME_MODULE_NAME + "/core-js/is-iterable", "default", "isIterable"), [path.node.right]));
  57. },
  58. // Array.from -> _core.Array.from
  59. MemberExpression: {
  60. enter: function enter(path, state) {
  61. if (state.opts.polyfill === false) return;
  62. if (!path.isReferenced()) return;
  63. var node = path.node;
  64. var obj = node.object;
  65. var prop = node.property;
  66. if (!t.isReferenced(obj, node)) return;
  67. if (node.computed) return;
  68. if (!has(_definitions2.default.methods, obj.name)) return;
  69. var methods = _definitions2.default.methods[obj.name];
  70. if (!has(methods, prop.name)) return;
  71. // doesn't reference the global
  72. if (path.scope.getBindingIdentifier(obj.name)) return;
  73. // special case Object.defineProperty to not use core-js when using string keys
  74. if (obj.name === "Object" && prop.name === "defineProperty" && path.parentPath.isCallExpression()) {
  75. var call = path.parentPath.node;
  76. if (call.arguments.length === 3 && t.isLiteral(call.arguments[1])) return;
  77. }
  78. path.replaceWith(state.addImport(RUNTIME_MODULE_NAME + "/core-js/" + methods[prop.name], "default", obj.name + "$" + prop.name));
  79. },
  80. exit: function exit(path, state) {
  81. if (state.opts.polyfill === false) return;
  82. if (!path.isReferenced()) return;
  83. var node = path.node;
  84. var obj = node.object;
  85. if (!has(_definitions2.default.builtins, obj.name)) return;
  86. if (path.scope.getBindingIdentifier(obj.name)) return;
  87. path.replaceWith(t.memberExpression(state.addImport(RUNTIME_MODULE_NAME + "/core-js/" + _definitions2.default.builtins[obj.name], "default", obj.name), node.property, node.computed));
  88. }
  89. }
  90. }
  91. };
  92. };
  93. var _definitions = require("./definitions");
  94. var _definitions2 = _interopRequireDefault(_definitions);
  95. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  96. exports.definitions = _definitions2.default;