unwrap-jsx-expressions.ts

import * as babel from "@babel/core";
import * as t from "@babel/types";
import { DEFAULT_UNWRAPPED_JSX_EXPRESSION_VARIABLE_NAME } from "./constants";
import { isInTheSameFunctionScope } from "./is-in-the-same-function-scope";
import { unwrapGenericExpression } from "./unwrap-generic-expression";

export function unwrapJsxExpressions(fn: babel.NodePath<t.Function>) {
  fn.traverse({
    JSXExpressionContainer(path) {
      const expressionPath = path.get("expression");
      const expression = expressionPath.node;

      if (
        t.isJSXEmptyExpression(expression) ||
        t.isIdentifier(expression) ||
        t.isLiteral(expression)
      ) {
        return;
      }
      if (!isInTheSameFunctionScope(path, fn)) {
        return;
      }

      unwrapGenericExpression(
        fn,
        expressionPath as babel.NodePath<t.Expression>,
        DEFAULT_UNWRAPPED_JSX_EXPRESSION_VARIABLE_NAME
      );
    },
  });
}