chore: wip

src/extract.ts

@@ -25,20 +25,27 @@ export async function extractTypeFromSource(filePath: string): Promise<string> {
   }
 
   // Handle exported functions with comments
-  const exportedFunctionRegex = /(\/\*\*[\s\S]*?\*\/\s*)?(export\s+(async\s+)?function\s+(\w+)\s*\(([^)]*)\)\s*:\s*([^{]+))/g
+  const exportedFunctionRegex = /(\/\*\*[\s\S]*?\*\/\s*)?(export\s+)(async\s+)?(function\s+(\w+)\s*\(([^)]*)\)\s*:\s*([^{]+))/g
   let match
   while ((match = exportedFunctionRegex.exec(fileContent)) !== null) {
-    const [, comment, , isAsync, name, params, returnType] = match
+    const [, comment, exportKeyword, isAsync, funcDeclaration, name, params, returnType] = match
     const cleanParams = params.replace(/\s*=\s*[^,)]+/g, '')
-    const declaration = `${comment || ''}export declare ${isAsync || ''}function ${name}(${cleanParams}): ${returnType.trim()}`
+    let cleanReturnType = returnType.trim()
+
+    // If the function is async, wrap the return type in Promise
+    if (isAsync) {
+      cleanReturnType = `Promise<${cleanReturnType}>`
+    }
+
+    const declaration = `${comment || ''}${exportKeyword}declare function ${name}(${cleanParams}): ${cleanReturnType}`
     declarations += `${declaration}\n\n`
 
     // Check for types used in parameters
     const paramTypes = params.match(/:\s*([^,)=]+)/g) || []
     paramTypes.forEach(type => addUsedType(type.slice(1).trim()))
 
     // Check for return type
-    addUsedType(returnType.trim())
+    addUsedType(cleanReturnType)
   }
 
   // Handle other exports (interface, type, const)