Adopt pylance new api from prerelease

microsoft · Mar 17, 2024 · 54007ac · 54007ac
1 parent 7e44bfd
commit 54007ac
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/src/standalone/executionAnalysis/pylance.ts b/src/standalone/executionAnalysis/pylance.ts
@@ -29,7 +29,7 @@ export interface INotebookLanguageClient {
         },
         token: vscode.CancellationToken
     ): Promise<LocationWithReferenceKind[] | null | undefined>;
-    getDocumentSymbols(
+    getDocumentSymbols?(
         document: vscode.TextDocument,
         token: vscode.CancellationToken
     ): Promise<vscode.DocumentSymbol[] | undefined>;

diff --git a/src/standalone/executionAnalysis/symbols.ts b/src/standalone/executionAnalysis/symbols.ts
@@ -461,8 +461,15 @@ export class NotebookDocumentSymbolTracker {
     }
 
     private async _getDocumentSymbols(cell: vscode.NotebookCell) {
-        const tokenSource = new vscode.CancellationTokenSource();
-        return this._client.getDocumentSymbols(cell.document, tokenSource.token);
+        if (this._client.getDocumentSymbols) {
+            const tokenSource = new vscode.CancellationTokenSource();
+            return this._client.getDocumentSymbols(cell.document, tokenSource.token);
+        } else {
+            return vscode.commands.executeCommand<(vscode.SymbolInformation & vscode.DocumentSymbol)[] | undefined>(
+                'vscode.executeDocumentSymbolProvider',
+                cell.document.uri
+            );
+        }
     }
 
     private async _doRequestCellSymbols(cell: vscode.NotebookCell, token: vscode.CancellationToken) {