/
selectionRanges.ts
33 lines (28 loc) · 1.09 KB
/
selectionRanges.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import type { DocumentServiceRuntimeContext } from '../types';
import { documentArgFeatureWorker } from '../utils/featureWorkers';
import type { TextDocument } from 'vscode-languageserver-textdocument';
import { transformSelectionRanges } from '@volar/transforms';
import * as vscode from 'vscode-languageserver-protocol';
import * as shared from '@volar/shared';
export function register(context: DocumentServiceRuntimeContext) {
return (document: TextDocument, positions: vscode.Position[]) => {
return documentArgFeatureWorker(
context,
document,
positions,
sourceMap => sourceMap.embeddedFile.capabilities.foldingRanges,
(positions, sourceMap) => [positions
.map(position => sourceMap.getMappedRange(position, position)?.[0].start)
.filter(shared.notEmpty)],
(plugin, document, positions) => positions.length ? plugin.getSelectionRanges?.(document, positions) : undefined,
(data, sourceMap) => transformSelectionRanges(
data,
range => {
if (!sourceMap)
return range;
return sourceMap.getSourceRange(range.start, range.end)?.[0]
},
),
);
}
}