feat: handle default export

stacksjs · Oct 20, 2024 · f7a9abd · f7a9abd
1 parent bc5d418
commit f7a9abd
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/extract.ts b/src/extract.ts
@@ -83,6 +83,13 @@ export async function extractTypeFromSource(filePath: string): Promise<string> {
     }
   }
 
+  // Handle default export
+  const defaultExportRegex = /export\s+default\s+(\w+)/
+  const defaultExportMatch = fileContent.match(defaultExportRegex)
+  if (defaultExportMatch) {
+    declarations += `export default ${defaultExportMatch[1]}\n`
+  }
+
   // Generate import statements for used types
   let importDeclarations = ''
   importMap.forEach((types, path) => {