Function renderParsedBody

packages/kernel/core/src/code-recovery.ts:87–120

(source: string)

};

const renderParsedBody = (source: string): string => {
  const program = parse(source, {
    sourceType: "module",
    allowAwaitOutsideFunction: true,
    allowReturnOutsideFunction: true,
    allowImportExportEverywhere: true,
    plugins: ["typescript"],
  }).program;

  if (program.body.length !== 1) return source;

  const [statement] = program.body;
  if (!statement) return source;

  switch (statement.type) {
    case "ExpressionStatement": {
      const expression = unwrapExpression(
        statement.expression as { type: string; expression?: unknown },
      ) as {
        type?: string;
      };
      return expression?.type === "ArrowFunctionExpression" ||
        expression?.type === "FunctionExpression"
        ? wrapCallableBody(source)
        : source;
    }
    case "FunctionDeclaration":
      return statement.id?.name ? wrapNamedFunctionBody(source, statement.id.name) : source;
    case "ExportDefaultDeclaration":
      return renderExportDefaultBody(source, statement.declaration);
    default:
      return source;
  }
};

const renderHeuristicBody = (source: string): string => {
  const withoutDefaultExport = source.replace(/^export\s+default\s+/, "").trim();