feat: implement textDocument/selectionRange request (#642)

README.md

@@ -628,6 +628,7 @@ interface DefinitionSymbol {
 - [x] textDocument/rangeFormatting
 - [x] textDocument/references
 - [x] textDocument/rename
+- [x] textDocument/selectionRange
 - [x] textDocument/signatureHelp
 - [x] workspace/symbol
 - [x] workspace/didChangeConfiguration

src/lsp-connection.ts

@@ -53,6 +53,7 @@ export function createLspConnection(options: LspConnectionOptions): lsp.Connecti
     connection.onReferences(server.references.bind(server));
     connection.onRenameRequest(server.rename.bind(server));
     connection.onPrepareRename(server.prepareRename.bind(server));
+    connection.onSelectionRanges(server.selectionRanges.bind(server));
     connection.onSignatureHelp(server.signatureHelp.bind(server));
     connection.onWorkspaceSymbol(server.workspaceSymbol.bind(server));
     connection.onFoldingRanges(server.foldingRanges.bind(server));

src/lsp-server.ts

@@ -16,7 +16,7 @@ import API from './utils/api.js';
 import { Logger, LogLevel, PrefixingLogger } from './utils/logger.js';
 import { TspClient } from './tsp-client.js';
 import { DiagnosticEventQueue } from './diagnostic-queue.js';
-import { toDocumentHighlight, uriToPath, toSymbolKind, toLocation, pathToUri, toTextEdit, normalizePath } from './protocol-translation.js';
+import { toDocumentHighlight, uriToPath, toSymbolKind, toLocation, toSelectionRange, pathToUri, toTextEdit, normalizePath } from './protocol-translation.js';
 import { LspDocuments, LspDocument } from './document.js';
 import { asCompletionItem, asResolvedCompletionItem, getCompletionTriggerCharacter } from './completion.js';
 import { asSignatureHelp, toTsTriggerReason } from './hover.js';
@@ -217,6 +217,7 @@ export class LspServer {
                 inlayHintProvider: true,
                 renameProvider: prepareSupport ? { prepareProvider: true } : true,
                 referencesProvider: true,
+                selectionRangeProvider: true,
                 signatureHelpProvider: {
                     triggerCharacters: ['(', ',', '<'],
                     retriggerCharacters: [')'],
@@ -791,6 +792,21 @@ export class LspServer {
         return [];
     }
 
+    async selectionRanges(params: lsp.SelectionRangeParams): Promise<lsp.SelectionRange[] | null> {
+        const file = uriToPath(params.textDocument.uri);
+        if (!file) {
+            return null;
+        }
+        const response = await this.tspClient.request(CommandTypes.SelectionRange, {
+            file,
+            locations: params.positions.map(Position.toLocation),
+        });
+        if (response.type !== 'response' || !response.body) {
+            return null;
+        }
+        return response.body.map(toSelectionRange);
+    }
+
     async signatureHelp(params: lsp.SignatureHelpParams): Promise<lsp.SignatureHelp | undefined> {
         const file = uriToPath(params.textDocument.uri);
         this.logger.log('signatureHelp', params, file);

src/protocol-translation.ts

@@ -9,7 +9,7 @@ import * as lsp from 'vscode-languageserver';
 import vscodeUri from 'vscode-uri';
 import { LspDocuments } from './document.js';
 import { tslib, tsp, SupportedFeatures } from './ts-protocol.js';
-import { Position } from './utils/typeConverters.js';
+import { Position, Range } from './utils/typeConverters.js';
 
 const RE_PATHSEP_WINDOWS = /\\/g;
 
@@ -160,6 +160,13 @@ function asRelatedInformation(info: tsp.DiagnosticRelatedInformation[] | undefin
     return result;
 }
 
+export function toSelectionRange(range: tsp.SelectionRange): lsp.SelectionRange {
+    return lsp.SelectionRange.create(
+        Range.fromTextSpan(range.textSpan),
+        range.parent ? toSelectionRange(range.parent) : undefined,
+    );
+}
+
 export function toTextEdit(edit: tsp.CodeEdit): lsp.TextEdit {
     return {
         range: {

src/tsServer/requests.ts

@@ -73,6 +73,7 @@ export interface TypeScriptRequestTypes {
     [CommandTypes.Quickinfo]: [tsp.FileLocationRequestArgs, tsp.QuickInfoResponse];
     [CommandTypes.References]: [tsp.FileLocationRequestArgs, tsp.ReferencesResponse];
     [CommandTypes.Rename]: [tsp.RenameRequestArgs, tsp.RenameResponse];
+    [CommandTypes.SelectionRange]: [tsp.SelectionRangeRequestArgs, tsp.SelectionRangeResponse];
     [CommandTypes.SignatureHelp]: [tsp.SignatureHelpRequestArgs, tsp.SignatureHelpResponse];
     [CommandTypes.TypeDefinition]: [tsp.FileLocationRequestArgs, tsp.TypeDefinitionResponse];
     [CommandTypes.UpdateOpen]: [tsp.UpdateOpenRequestArgs, tsp.Response];