@putout/plugin-promises/lib/apply-top-level-await/index.js

🐊Putout plugin improves Promise-related code

'use strict';

const fullstore = require('fullstore');
const {replaceWithMultiple} = require('putout').operator;

const add = ({push, isImports, isExports}) => (path) => {
    const calleePath = path.get('callee');
    
    if (!isImports() && !isExports())
        return;
    
    const {id} = calleePath.node;
    
    if (id)
        return;
    
    push(path);
};

module.exports.report = () => `Use top level 'await'`;

module.exports.fix = (path) => {
    const {body} = path.get('callee.body').node;
    replaceWithMultiple(path, body);
};

module.exports.traverse = ({push}) => {
    const isExports = fullstore();
    const isImports = fullstore();
    
    const addPath = add({
        push,
        isImports,
        isExports,
    });
    
    return {
        'import __ from "__"'() {
            isImports(true);
        },
        ExportNamedDeclaration() {
            isExports(true);
        },
        ExportDefaultDeclaration() {
            isExports(true);
        },
        '(async function __() {})()': (path) => {
            if (!isAsyncParent(path))
                return;
            
            addPath(path);
        },
        '(async () => __)()': (path) => {
            if (!isAsyncParent(path))
                return;
            
            addPath(path);
        },
    };
};

function isAsyncParent(path) {
    const {parentPath} = path.parentPath.parentPath;
    
    if (!parentPath)
        return true;
    
    return !parentPath.isFunction() || parentPath.node.async;
}