fix(babel): partially handle functions as default export

pi0 · pi0 · commit a35258bdae73 · 2023-03-28T12:28:10.000+02:00
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 })
+              ),
+            ])
+          );
+        }
       },
     },
   };
