fix: expand selection broken

close #1085
vuejs · Mar 21, 2022 · ed75138 · ed75138
1 parent 9dd9a08
commit ed75138
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/packages/vue-language-service/src/documentFeatures/selectionRanges.ts b/packages/vue-language-service/src/documentFeatures/selectionRanges.ts
@@ -13,11 +13,11 @@ export function register(context: DocumentServiceRuntimeContext) {
 			context,
 			document,
 			positions,
-			sourceMap => true,
+			sourceMap => sourceMap.embeddedFile.capabilities.foldingRanges,
 			(positions, sourceMap) => [positions
-				.map(position => sourceMap.getMappedRange(position, position, data => !!data.capabilities.basic)?.[0].start)
+				.map(position => sourceMap.getMappedRange(position, position)?.[0].start)
 				.filter(shared.notEmpty)],
-			(plugin, document, positions) => plugin.getSelectionRanges?.(document, positions),
+			(plugin, document, positions) => positions.length ? plugin.getSelectionRanges?.(document, positions) : undefined,
 			(data, sourceMap) => transformSelectionRanges(
 				data,
 				range => {
@@ -28,7 +28,6 @@ export function register(context: DocumentServiceRuntimeContext) {
 					return sourceMap.getSourceRange(range.start, range.end)?.[0]
 				},
 			),
-			arr => arr.flat(),
 		);
 	}
 }