better workspace folder selection

howsoai · Jul 16, 2024 · 6295a40 · 6295a40
1 parent 1f1dbf2
commit 6295a40
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/language/definitions.ts b/src/language/definitions.ts
@@ -30,8 +30,12 @@ export class AmalgamDefinitionProvider implements vscode.DefinitionProvider {
       return undefined;
     }
 
-    const rootPath = workspaceFolders[0].uri.fsPath;
-    const files = await this.getAllAmlgFiles(rootPath);
+
+    const workspaceFolder = vscode.workspace.getWorkspaceFolder(document.uri);
+    if (!workspaceFolder) {
+      return undefined;
+    }
+    const files = await this.getAllAmlgFiles(workspaceFolder.uri.fsPath);
 
     for (const file of files) {
       if (token.isCancellationRequested) {