@putout/plugin-try-catch/lib/apply-try-catch.js

🐊Putout plugin adds ability to apply try-catch

import {types, operator} from 'putout';

const {
    awaitExpression,
    arrayPattern,
    variableDeclarator,
    variableDeclaration,
    ifStatement,
    identifier,
    callExpression,
} = types;

const {replaceWithMultiple} = operator;

export const applyTryCatch = (tryName) => (path) => {
    const expression = parseExpression(path);
    
    const callNode = callExpression(identifier(tryName), [
        expression.callee,
        ...expression.arguments,
    ]);
    
    const {param} = path.node.handler;
    const {body} = path.get('handler').node;
    
    if (!param) {
        replaceWithMultiple(path, [
            maybeAwait(path, callNode),
            ...body.body,
        ]);
        return;
    }
    
    const ifNode = body.body.length ? [
        ifStatement(param, body),
    ] : body.body;
    
    const varNode = variableDeclaration('const', [
        variableDeclarator(arrayPattern([param]), maybeAwait(path, callNode)),
    ]);
    
    replaceWithMultiple(path, [
        varNode,
        ...ifNode,
    ]);
};

function parseExpression(path) {
    const expressionPath = path.get('block.body.0.expression');
    
    if (expressionPath.isAwaitExpression())
        return expressionPath.node.argument;
    
    return expressionPath.node;
}

function maybeAwait(path, node) {
    const expressionPath = path.get('block.body.0.expression');
    
    if (expressionPath.isAwaitExpression())
        return awaitExpression(node);
    
    return node;
}