index.js 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = void 0;
  6. var _helperPluginUtils = require("@babel/helper-plugin-utils");
  7. var _helperRemapAsyncToGenerator = _interopRequireDefault(require("@babel/helper-remap-async-to-generator"));
  8. var _helperModuleImports = require("@babel/helper-module-imports");
  9. var _core = require("@babel/core");
  10. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  11. var _default = (0, _helperPluginUtils.declare)((api, options) => {
  12. api.assertVersion(7);
  13. const {
  14. method,
  15. module
  16. } = options;
  17. const noNewArrows = api.assumption("noNewArrows");
  18. if (method && module) {
  19. return {
  20. name: "transform-async-to-generator",
  21. visitor: {
  22. Function(path, state) {
  23. if (!path.node.async || path.node.generator) return;
  24. let wrapAsync = state.methodWrapper;
  25. if (wrapAsync) {
  26. wrapAsync = _core.types.cloneNode(wrapAsync);
  27. } else {
  28. wrapAsync = state.methodWrapper = (0, _helperModuleImports.addNamed)(path, method, module);
  29. }
  30. (0, _helperRemapAsyncToGenerator.default)(path, {
  31. wrapAsync
  32. }, noNewArrows);
  33. }
  34. }
  35. };
  36. }
  37. return {
  38. name: "transform-async-to-generator",
  39. visitor: {
  40. Function(path, state) {
  41. if (!path.node.async || path.node.generator) return;
  42. (0, _helperRemapAsyncToGenerator.default)(path, {
  43. wrapAsync: state.addHelper("asyncToGenerator")
  44. }, noNewArrows);
  45. }
  46. }
  47. };
  48. });
  49. exports.default = _default;