From 714fd49d61f381ba2a592b3eec224860bdca10f5 Mon Sep 17 00:00:00 2001
From: johnsoncodehk <johnsoncodehk@gmail.com>
Date: Thu, 16 Jun 2022 10:44:49 +0800
Subject: [PATCH] fix: auto import component tag name including "Vue" on import

close #1461
---
 .../vue-code-gen/src/generators/template.ts   |  2 +-
 .../src/plugins/vue-template.ts               | 30 +++++--------------
 2 files changed, 8 insertions(+), 24 deletions(-)

diff --git a/packages/vue-code-gen/src/generators/template.ts b/packages/vue-code-gen/src/generators/template.ts
index 7852e35d4..53951f776 100644
--- a/packages/vue-code-gen/src/generators/template.ts
+++ b/packages/vue-code-gen/src/generators/template.ts
@@ -9,7 +9,7 @@ import { colletVars, walkInterpolationFragment } from '../transform';
 const capabilitiesSet = {
 	all: { basic: true, diagnostic: true, references: true, definitions: true, rename: true, completion: true, semanticTokens: true },
 	noDiagnostic: { basic: true, references: true, definitions: true, rename: true, completion: true, semanticTokens: true },
-	diagnosticOnly: { diagnostic: true, completion: true },
+	diagnosticOnly: { diagnostic: true },
 	tagHover: { basic: true },
 	event: { basic: true, diagnostic: true },
 	tagReference: { references: true, definitions: true, rename: { in: false, out: true } },
diff --git a/packages/vue-language-service/src/plugins/vue-template.ts b/packages/vue-language-service/src/plugins/vue-template.ts
index dbc431685..16e6dcc6e 100644
--- a/packages/vue-language-service/src/plugins/vue-template.ts
+++ b/packages/vue-language-service/src/plugins/vue-template.ts
@@ -323,15 +323,9 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 		const componentName = capitalize(camelize(item.label.replace(/\./g, '-')));
 		const textDoc = vueDocument.getDocument();
 		const compiledVue = vueDocument.file.getCompiledVue()!;
-		let insertText = '';
-		const planAResult = await planAInsertText();
-		if (planAResult) {
-			insertText = planAResult.insertText;
-			item.detail = planAResult.description + '\n\n' + rPath;
-		}
-		else {
-			insertText = planBInsertText();
-			item.detail = `Auto import from '${importPath}'\n\n${rPath}`;
+		const insert = await getInsert();
+		if (insert) {
+			item.detail = insert.description + '\n\n' + rPath;
 		}
 		if (descriptor.scriptSetup) {
 			const startTagEnd = compiledVue.getSourceRange(descriptor.scriptSetup.startTagEnd)?.[0].start;
@@ -341,7 +335,7 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 				item.additionalTextEdits = [
 					vscode.TextEdit.insert(
 						editPosition,
-						'\n' + insertText,
+						'\n' + insert?.insertText,
 					),
 				];
 			}
@@ -354,7 +348,7 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 				item.additionalTextEdits = [
 					vscode.TextEdit.insert(
 						editPosition,
-						'\n' + insertText,
+						'\n' + insert?.insertText,
 					),
 				];
 				const scriptRanges = parseScriptRanges(options.ts, scriptAst, !!descriptor.scriptSetup, true, true);
@@ -407,7 +401,7 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 		}
 		return item;
 
-		async function planAInsertText() {
+		async function getInsert() {
 			const embeddedScriptUri = shared.fsPathToUri(vueDocument.file.getScriptFileName());
 			const tsImportName = camelize(path.basename(importFile).replace(/\./g, '-'));
 			const [formatOptions, preferences] = await Promise.all([
@@ -430,16 +424,6 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 				}
 			}
 		}
-		function planBInsertText() {
-			const anyImport = scriptSetupImport ?? scriptImport;
-			let withSemicolon = true;
-			let quote = '"';
-			if (anyImport) {
-				withSemicolon = anyImport.text.endsWith(';');
-				quote = anyImport.text.includes("'") ? "'" : '"';
-			}
-			return `import ${componentName} from ${quote}${importPath}${quote}${withSemicolon ? ';' : ''}`;
-		}
 	}
 
 	async function provideHtmlData(vueDocument: VueDocument) {
@@ -546,7 +530,7 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 
 		if (enabledComponentAutoImport && (descriptor.script || descriptor.scriptSetup)) {
 			for (const vueDocument of options.vueDocuments.getAll()) {
-				let baseName = path.basename(vueDocument.uri);
+				let baseName = path.removeExt(path.basename(vueDocument.uri), '.vue');
 				if (baseName.toLowerCase() === 'index') {
 					baseName = path.basename(path.dirname(vueDocument.uri));
 				}