Use new version of withVirtualDocUri() for statement range provider…

… as well
quarto-dev · Nov 9, 2024 · 3e6b431 · 3e6b431
1 parent 323a5ca
commit 3e6b431
Showing 1 changed file with 3 additions and 10 deletions.
diff --git a/apps/vscode/src/host/hooks.ts b/apps/vscode/src/host/hooks.ts
@@ -163,20 +163,13 @@ class EmbeddedStatementRangeProvider implements HostStatementRangeProvider {
     token: vscode.CancellationToken): Promise<hooks.StatementRange | undefined> {
     const vdoc = await virtualDoc(document, position, this._engine);
     if (vdoc) {
-      const vdocUri = await virtualDocUri(vdoc, document.uri, "statementRange");
-      try {
+      return await withVirtualDocUri(vdoc, document.uri, "statementRange", async (uri: vscode.Uri) => {
         return getStatementRange(
-          vdocUri.uri,
+          uri,
           adjustedPosition(vdoc.language, position),
           vdoc.language
         );
-      } catch (error) {
-        return undefined;
-      } finally {
-        if (vdocUri.cleanup) {
-          await vdocUri.cleanup();
-        }
-      }
+      });
     } else {
       return undefined;
     }