123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899 |
- "use strict";
- Object.defineProperty(exports, "__esModule", {
- value: true
- });
- exports.default = void 0;
- var _helperPluginUtils = require("@babel/helper-plugin-utils");
- var _helperRemapAsyncToGenerator = require("@babel/helper-remap-async-to-generator");
- var _pluginSyntaxAsyncGenerators = require("@babel/plugin-syntax-async-generators");
- var _core = require("@babel/core");
- var _forAwait = require("./for-await");
- var _default = (0, _helperPluginUtils.declare)(api => {
- api.assertVersion(7);
- const yieldStarVisitor = {
- Function(path) {
- path.skip();
- },
- YieldExpression({
- node
- }, state) {
- if (!node.delegate) return;
- const callee = state.addHelper("asyncGeneratorDelegate");
- node.argument = _core.types.callExpression(callee, [_core.types.callExpression(state.addHelper("asyncIterator"), [node.argument]), state.addHelper("awaitAsyncGenerator")]);
- }
- };
- const forAwaitVisitor = {
- Function(path) {
- path.skip();
- },
- ForOfStatement(path, {
- file
- }) {
- const {
- node
- } = path;
- if (!node.await) return;
- const build = (0, _forAwait.default)(path, {
- getAsyncIterator: file.addHelper("asyncIterator")
- });
- const {
- declar,
- loop
- } = build;
- const block = loop.body;
- path.ensureBlock();
- if (declar) {
- block.body.push(declar);
- }
- block.body.push(...node.body.body);
- _core.types.inherits(loop, node);
- _core.types.inherits(loop.body, node.body);
- if (build.replaceParent) {
- path.parentPath.replaceWithMultiple(build.node);
- } else {
- path.replaceWithMultiple(build.node);
- }
- }
- };
- const visitor = {
- Function(path, state) {
- if (!path.node.async) return;
- path.traverse(forAwaitVisitor, state);
- if (!path.node.generator) return;
- path.traverse(yieldStarVisitor, state);
- (0, _helperRemapAsyncToGenerator.default)(path, {
- wrapAsync: state.addHelper("wrapAsyncGenerator"),
- wrapAwait: state.addHelper("awaitAsyncGenerator")
- });
- }
- };
- return {
- name: "proposal-async-generator-functions",
- inherits: _pluginSyntaxAsyncGenerators.default,
- visitor: {
- Program(path, state) {
- path.traverse(visitor, state);
- }
- }
- };
- });
- exports.default = _default;
|