@mui/codemod/util/getReturnExpression.js

Codemod scripts for Material UI.

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = getReturnExpression;
/**
 * @param {import('ast-types').namedTypes.ArrowFunctionExpression | import('ast-types').namedTypes.FunctionExpression} node
 */
function getReturnExpression(node) {
  let body = node.body;
  if (body === 'BlockStatement') {
    body = body.body;
  }
  if (Array.isArray(body)) {
    return body.find(statement => statement.type === 'ReturnStatement')?.argument;
  }
  return body;
}