fix(language-service): typescript-semantic renaming first in style blocks (vuejs#4685)

Author: KazariEX

packages/language-service/lib/plugins/css.ts

@@ -1,12 +1,21 @@
-import type { LanguageServicePlugin } from '@volar/language-service';
-import { create as baseCreate } from 'volar-service-css';
+import type { LanguageServicePlugin, VirtualCode } from '@volar/language-service';
+import { VueVirtualCode } from '@vue/language-core';
+import { create as baseCreate, type Provide } from 'volar-service-css';
+import * as css from 'vscode-css-languageservice';
+import type { TextDocument } from 'vscode-languageserver-textdocument';
+import { URI } from 'vscode-uri';
 
 export function create(): LanguageServicePlugin {
 	const base = baseCreate({ scssDocumentSelector: ['scss', 'postcss'] });
 	return {
 		...base,
 		create(context) {
 			const baseInstance = base.create(context);
+			const {
+				'css/languageService': getCssLs,
+				'css/stylesheet': getStylesheet,
+			} = baseInstance.provide as Provide;
+
 			return {
 				...baseInstance,
 				async provideDiagnostics(document, token) {
@@ -18,7 +27,73 @@ export function create(): LanguageServicePlugin {
 					}
 					return diagnostics;
 				},
+				/**
+				 * If the editing position is within the virtual code and navigation is enabled,
+				 * skip the CSS renaming feature.
+				 */
+				provideRenameRange(document, position) {
+					do {
+						const uri = URI.parse(document.uri);
+						const decoded = context.decodeEmbeddedDocumentUri(uri);
+						const sourceScript = decoded && context.language.scripts.get(decoded[0]);
+						const virtualCode = decoded && sourceScript?.generated?.embeddedCodes.get(decoded[1]);
+						if (!sourceScript?.generated || !virtualCode?.id.startsWith('style_')) {
+							break;
+						}
+
+						const root = sourceScript.generated.root;
+						if (!(root instanceof VueVirtualCode)) {
+							break;
+						}
+
+						const block = root.sfc.styles.find(style => style.name === decoded![1]);
+						if (!block) {
+							break;
+						}
+
+						let script: VirtualCode | undefined;
+						for (const [key, value] of sourceScript.generated.embeddedCodes) {
+							if (key.startsWith('script_')) {
+								script = value;
+								break;
+							}
+						}
+						if (!script) {
+							break;
+						}
+
+						const offset = document.offsetAt(position) + block.startTagEnd;
+						for (const { sourceOffsets, lengths, data } of script.mappings) {
+							if (
+								!sourceOffsets.length
+								|| !data.navigation
+								|| typeof data.navigation === 'object' && !data.navigation.shouldRename
+							) {
+								continue;
+							}
+
+							const start = sourceOffsets[0];
+							const end = sourceOffsets.at(-1)! + lengths.at(-1)!;
+
+							if (offset >= start && offset <= end) {
+								return;
+							}
+						}
+					} while (0);
+
+					return worker(document, (stylesheet, cssLs) => {
+						return cssLs.prepareRename(document, position, stylesheet);
+					});
+				}
 			};
+
+			function worker<T>(document: TextDocument, callback: (stylesheet: css.Stylesheet, cssLs: css.LanguageService) => T) {
+				const cssLs = getCssLs(document);
+				if (!cssLs) {
+					return;
+				}
+				return callback(getStylesheet(document, cssLs), cssLs);
+			}
 		},
 	};
 }

packages/language-service/package.json

@@ -34,6 +34,7 @@
 		"volar-service-pug-beautify": "0.0.62",
 		"volar-service-typescript": "0.0.62",
 		"volar-service-typescript-twoslash-queries": "0.0.62",
+		"vscode-css-languageservice": "^6.3.1",
 		"vscode-html-languageservice": "^5.2.0",
 		"vscode-languageserver-textdocument": "^1.0.11",
 		"vscode-uri": "^3.0.8"