fix: handle arrow functions with literal return type

unjs · Mar 27, 2023 · 4d8453b · 4d8453b
1 parent cc693ff
commit 4d8453b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/loader/babel.ts b/src/loader/babel.ts
@@ -27,7 +27,9 @@ const babelPluginUntyped: PluginItem = function (api: ConfigAPI) {
           const newDeclaration = t.functionDeclaration(
             declaration.id,
             declaration.init.params,
-            declaration.init.body as t.BlockStatement
+            t.isLiteral(declaration.init.body)
+              ? t.blockStatement([t.returnStatement(declaration.init.body)])
+              : (declaration.init.body as t.BlockStatement)
           );
           newDeclaration.returnType = declaration.init.returnType;
           p.replaceWith(newDeclaration);