From 635898bcc9994d1a846e5e48a504439d3c206b86 Mon Sep 17 00:00:00 2001
From: johnsoncodehk <johnsoncodehk@gmail.com>
Date: Fri, 17 Jun 2022 18:56:49 +0800
Subject: [PATCH] fix: component auto import unreliable

close #1470
---
 .../src/plugins/vue-template.ts               | 23 +++++++++++++++----
 1 file changed, 18 insertions(+), 5 deletions(-)

diff --git a/packages/vue-language-service/src/plugins/vue-template.ts b/packages/vue-language-service/src/plugins/vue-template.ts
index 16e6dcc6e..85fc0561e 100644
--- a/packages/vue-language-service/src/plugins/vue-template.ts
+++ b/packages/vue-language-service/src/plugins/vue-template.ts
@@ -323,8 +323,8 @@ 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()!;
-		const insert = await getInsert();
-		if (insert) {
+		const insert = await getTypeScriptInsert() ?? getMonkeyInsert();
+		if (insert.description) {
 			item.detail = insert.description + '\n\n' + rPath;
 		}
 		if (descriptor.scriptSetup) {
@@ -335,7 +335,7 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 				item.additionalTextEdits = [
 					vscode.TextEdit.insert(
 						editPosition,
-						'\n' + insert?.insertText,
+						'\n' + insert.insertText,
 					),
 				];
 			}
@@ -348,7 +348,7 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 				item.additionalTextEdits = [
 					vscode.TextEdit.insert(
 						editPosition,
-						'\n' + insert?.insertText,
+						'\n' + insert.insertText,
 					),
 				];
 				const scriptRanges = parseScriptRanges(options.ts, scriptAst, !!descriptor.scriptSetup, true, true);
@@ -401,7 +401,7 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 		}
 		return item;
 
-		async function getInsert() {
+		async function getTypeScriptInsert() {
 			const embeddedScriptUri = shared.fsPathToUri(vueDocument.file.getScriptFileName());
 			const tsImportName = camelize(path.basename(importFile).replace(/\./g, '-'));
 			const [formatOptions, preferences] = await Promise.all([
@@ -424,6 +424,19 @@ export default function useVueTemplateLanguagePlugin<T extends ReturnType<typeof
 				}
 			}
 		}
+		function getMonkeyInsert() {
+			const anyImport = scriptSetupImport ?? scriptImport;
+			let withSemicolon = true;
+			let quote = '"';
+			if (anyImport) {
+				withSemicolon = anyImport.text.endsWith(';');
+				quote = anyImport.text.includes("'") ? "'" : '"';
+			}
+			return {
+				insertText: `import ${componentName} from ${quote}${importPath}${quote}${withSemicolon ? ';' : ''}`,
+				description: '',
+			};
+		}
 	}
 
 	async function provideHtmlData(vueDocument: VueDocument) {