fix(babel): partially handle functions as default export

unjs · Mar 28, 2023 · a35258b · a35258b
1 parent 55f1ffb
commit a35258b
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/src/loader/babel.ts b/src/loader/babel.ts
@@ -101,7 +101,10 @@ const babelPluginUntyped: PluginItem = function (
         }
 
         // Do not add meta to internal functions
-        if (p.parent.type !== "ExportNamedDeclaration") {
+        if (
+          p.parent.type !== "ExportNamedDeclaration" &&
+          p.parent.type !== "ExportDefaultDeclaration"
+        ) {
           return;
         }
 
@@ -188,14 +191,18 @@ const babelPluginUntyped: PluginItem = function (
         });
 
         // Replace function with it's meta
-        p.replaceWith(
-          t.variableDeclaration("const", [
-            t.variableDeclarator(
-              t.identifier(p.node.id.name),
-              astify({ $schema: schema })
-            ),
-          ])
-        );
+        if (p.parent.type === "ExportDefaultDeclaration") {
+          p.replaceWith(astify({ $schema: schema }));
+        } else {
+          p.replaceWith(
+            t.variableDeclaration("const", [
+              t.variableDeclarator(
+                t.identifier(p.node.id.name),
+                astify({ $schema: schema })
+              ),
+            ])
+          );
+        }
       },
     },
   };