fix: moving components doesn't trigger import updates

close #1884
vuejs · Sep 19, 2022 · a13eaf1 · a13eaf1
1 parent 3dfd407
commit a13eaf1
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/packages/language-service/src/languageFeatures/fileRename.ts b/packages/language-service/src/languageFeatures/fileRename.ts
@@ -2,6 +2,7 @@ import type { LanguageServiceRuntimeContext } from '../types';
 import { embeddedEditToSourceEdit } from './rename';
 import type * as _ from 'vscode-languageserver-protocol';
 import * as dedupe from '../utils/dedupe';
+import { forEachEmbeddeds } from '@volar/language-core';
 
 export function register(context: LanguageServiceRuntimeContext) {
 
@@ -10,8 +11,21 @@ export function register(context: LanguageServiceRuntimeContext) {
 		const vueDocument = context.documents.get(oldUri);
 
 		if (vueDocument) {
-			oldUri += '.ts';
-			newUri += '.ts';
+
+			let tsExt: string | undefined;
+
+			forEachEmbeddeds(vueDocument.file.embeddeds, embedded => {
+				if (embedded.isTsHostFile && embedded.fileName.replace(vueDocument.file.fileName, '').match(/^\.(js|ts)x?$/)) {
+					tsExt = embedded.fileName.substring(embedded.fileName.lastIndexOf('.'));
+				}
+			});
+
+			if (!tsExt) {
+				return;
+			}
+
+			oldUri += tsExt;
+			newUri += tsExt;
 		}
 
 		for (const plugin of context.plugins) {