updated fastapi support

Author: doppleware

src/services/languages/extractors.ts

@@ -45,7 +45,7 @@ export interface IMethodExtractor {
     extractMethods(document: vscode.TextDocument, docSymbols: DocumentSymbol[]) : SymbolInfo[];
 }
 export interface IEndpointExtractor {
-    extractEndpoints(document: vscode.TextDocument, symbolInfos: SymbolInfo[], tokens: Token[]): EndpointInfo[];
+    extractEndpoints(document: vscode.TextDocument, symbolInfos: SymbolInfo[], tokens: Token[]): Promise<EndpointInfo[]>;
 }
 export interface ISpanExtractor {
     extractSpans(

src/services/languages/python/fastapiEndpointExtractor.ts

@@ -6,7 +6,7 @@ import { EndpointInfo, IEndpointExtractor, SymbolInfo } from "../extractors";
 
 export class FastapiEndpointExtractor implements IEndpointExtractor
 {
-    extractEndpoints(document: vscode.TextDocument, symbolInfo: SymbolInfo[], tokens: Token[]): EndpointInfo[] 
+    async extractEndpoints(document: vscode.TextDocument, symbolInfo: SymbolInfo[], tokens: Token[]): Promise<EndpointInfo[]> 
     {
         // Ensure fastapi module was imported
         if(!tokens.any(t => t.text == 'fastapi' && t.type == TokenType.module))
@@ -34,6 +34,21 @@ export class FastapiEndpointExtractor implements IEndpointExtractor
                 if (match){
                     prefix = match[2];
                 }
+
+                //Future: use the references to extract the 'include' statement to the router if it contains a prefix
+                
+                // const position  =new vscode.Position(appToken.range.start.line,appToken.range.start.character );
+
+                // let references : vscode.Location[] = await vscode.commands.executeCommand("vscode.executeReferenceProvider", document.uri,position);
+                // references = references.filter(x=>x.uri.fsPath!==document.uri.fsPath);
+                // if (references.length>0){
+                //     let referencedDocument = vscode.workspace.openTextDocument(references[0].uri);
+                //     let text = (await referencedDocument).getText(references[0].range);
+                //     console.log(text);
+
+                // }
+
+
             }
 
 

src/services/languages/symbolProvider.ts

@@ -68,9 +68,10 @@ export class SymbolProvider
         if(!supportedLanguage)
             return [];
 
-        return supportedLanguage.endpointExtractors
-            .map(x => x.extractEndpoints(document, symbolInfos, tokens))
-            .flat();
+        let extractedInfos = await Promise.all( supportedLanguage.endpointExtractors
+            .map( x =>  x.extractEndpoints(document, symbolInfos, tokens)));
+
+        return extractedInfos.flat();
     }
 
     public async getSpans(