@iov/cli/build/async.js

"use strict";
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (Object.hasOwnProperty.call(mod, k)) result[k] = mod[k];
    result["default"] = mod;
    return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
const babylon = require("babylon");
const recast = __importStar(require("recast"));
function wrapInAsyncFunction(code) {
    const codeInAsyncFunction = `(async () => {
    ${code}
  })()`;
    const ast = recast.parse(codeInAsyncFunction, { parser: babylon });
    const body = ast.program.body[0].expression.callee.body.body;
    if (body.length !== 0) {
        const last = body.pop();
        if (last.type === "ExpressionStatement") {
            body.push({
                type: "ReturnStatement",
                argument: last,
            });
        }
        else {
            body.push(last);
        }
    }
    // Remove var, let, const from variable declarations to make them available in context
    // tslint:disable-next-line:no-object-mutation
    ast.program.body[0].expression.callee.body.body = body.map((node) => {
        if (node.type === "VariableDeclaration") {
            return {
                type: "ExpressionStatement",
                expression: {
                    type: "SequenceExpression",
                    expressions: node.declarations.map((declaration) => ({
                        type: "AssignmentExpression",
                        operator: "=",
                        left: declaration.id,
                        right: declaration.init,
                    })),
                },
            };
        }
        else {
            return node;
        }
    });
    return recast.print(ast).code;
}
exports.wrapInAsyncFunction = wrapInAsyncFunction;
//# sourceMappingURL=async.js.map