From 4c3c8c4242d789c5219bb4c59669ca4097b7168a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=B1=B1=E5=90=B9=E8=89=B2=E5=BE=A1=E5=AE=88?=
 <85992002+KazariEX@users.noreply.github.com>
Date: Sat, 2 Nov 2024 16:33:44 +0800
Subject: [PATCH] feat(language-core): support `@vue-generic` (#4971)

---
 .../lib/codegen/template/context.ts           |  5 ++++
 .../lib/codegen/template/element.ts           | 26 ++++++++++++++++++-
 .../lib/codegen/template/elementDirectives.ts |  1 +
 .../lib/codegen/template/templateChild.ts     | 16 +++++++++---
 .../lib/plugins/vue-template-inline-ts.ts     | 14 +++++++++-
 .../lib/plugins/vue-directive-comments.ts     | 17 +++++++-----
 .../tsc/passedFixtures/vue2/tsconfig.json     |  1 +
 .../passedFixtures/vue3/v-generic/comp.vue    |  5 ++++
 .../passedFixtures/vue3/v-generic/main.vue    | 17 ++++++++++++
 9 files changed, 90 insertions(+), 12 deletions(-)
 create mode 100644 test-workspace/tsc/passedFixtures/vue3/v-generic/comp.vue
 create mode 100644 test-workspace/tsc/passedFixtures/vue3/v-generic/main.vue

diff --git a/packages/language-core/lib/codegen/template/context.ts b/packages/language-core/lib/codegen/template/context.ts
index c5abab54e1..7334ae85a3 100644
--- a/packages/language-core/lib/codegen/template/context.ts
+++ b/packages/language-core/lib/codegen/template/context.ts
@@ -63,6 +63,10 @@ export function createTemplateCodegenContext(options: Pick<TemplateCodegenOption
 		errors: number;
 		node: CompilerDOM.CommentNode;
 	} | undefined;
+	let lastGenericComment: {
+		content: string;
+		offset: number;
+	} | undefined;
 	let variableId = 0;
 
 	const codeFeatures = new Proxy(_codeFeatures, {
@@ -123,6 +127,7 @@ export function createTemplateCodegenContext(options: Pick<TemplateCodegenOption
 		dynamicSlots,
 		codeFeatures,
 		accessExternalVariables,
+		lastGenericComment,
 		hasSlotElements,
 		blockConditions,
 		usedComponentCtxVars,
diff --git a/packages/language-core/lib/codegen/template/element.ts b/packages/language-core/lib/codegen/template/element.ts
index 258f317f59..6fe5abfe31 100644
--- a/packages/language-core/lib/codegen/template/element.ts
+++ b/packages/language-core/lib/codegen/template/element.ts
@@ -205,7 +205,9 @@ export function* generateComponent(
 	yield* generateElementProps(options, ctx, node, props, false);
 	yield `}))${endOfLine}`;
 
-	yield `const ${var_componentInstance} = ${var_functionalComponent}(`;
+	yield `const ${var_componentInstance} = ${var_functionalComponent}`;
+	yield* generateComponentGeneric(ctx);
+	yield `(`;
 	yield* wrapWith(
 		startTagOffset,
 		startTagOffset + node.tag.length,
@@ -452,6 +454,28 @@ function* generateCanonicalComponentName(tagText: string, offset: number, featur
 	}
 }
 
+function* generateComponentGeneric(
+	ctx: TemplateCodegenContext
+): Generator<Code> {
+	if (ctx.lastGenericComment) {
+		const { content, offset } = ctx.lastGenericComment;
+		yield* wrapWith(
+			offset,
+			offset + content.length,
+			ctx.codeFeatures.verification,
+			`<`,
+			[
+				content,
+				'template',
+				offset,
+				ctx.codeFeatures.all
+			],
+			`>`
+		);
+	}
+	ctx.lastGenericComment = undefined;
+}
+
 function* generateComponentSlot(
 	options: TemplateCodegenOptions,
 	ctx: TemplateCodegenContext,
diff --git a/packages/language-core/lib/codegen/template/elementDirectives.ts b/packages/language-core/lib/codegen/template/elementDirectives.ts
index 3c67bce054..c8ebf4758a 100644
--- a/packages/language-core/lib/codegen/template/elementDirectives.ts
+++ b/packages/language-core/lib/codegen/template/elementDirectives.ts
@@ -24,6 +24,7 @@ export function* generateElementDirectives(
 			|| prop.name === 'bind'
 			|| prop.name === 'scope'
 			|| prop.name === 'data'
+			|| prop.name === 'generic'
 		) {
 			continue;
 		}
diff --git a/packages/language-core/lib/codegen/template/templateChild.ts b/packages/language-core/lib/codegen/template/templateChild.ts
index c7df01b522..69be101239 100644
--- a/packages/language-core/lib/codegen/template/templateChild.ts
+++ b/packages/language-core/lib/codegen/template/templateChild.ts
@@ -36,16 +36,26 @@ export function* generateTemplateChild(
 ): Generator<Code> {
 	if (prevNode?.type === CompilerDOM.NodeTypes.COMMENT) {
 		const commentText = prevNode.content.trim().split(' ')[0];
-		if (commentText.match(/^@vue-skip\b[\s\S]*/)) {
+		if (/^@vue-skip\b[\s\S]*/.test(commentText)) {
 			yield `// @vue-skip${newLine}`;
 			return;
 		}
-		else if (commentText.match(/^@vue-ignore\b[\s\S]*/)) {
+		else if (/^@vue-ignore\b[\s\S]*/.test(commentText)) {
 			yield* ctx.ignoreError();
 		}
-		else if (commentText.match(/^@vue-expect-error\b[\s\S]*/)) {
+		else if (/^@vue-expect-error\b[\s\S]*/.test(commentText)) {
 			yield* ctx.expectError(prevNode);
 		}
+		else {
+			const match = prevNode.loc.source.match(/^<!--\s*@vue-generic\b\s*\{(?<content>[^}]*)\}/);
+			if (match) {
+				const { content } = match.groups ?? {};
+				ctx.lastGenericComment = {
+					content,
+					offset: prevNode.loc.start.offset + match[0].indexOf(content)
+				};
+			}
+		}
 	}
 
 	const shouldInheritRootNodeAttrs = options.inheritAttrs;
diff --git a/packages/language-core/lib/plugins/vue-template-inline-ts.ts b/packages/language-core/lib/plugins/vue-template-inline-ts.ts
index ad38f5e013..f23bc015ff 100644
--- a/packages/language-core/lib/plugins/vue-template-inline-ts.ts
+++ b/packages/language-core/lib/plugins/vue-template-inline-ts.ts
@@ -19,6 +19,7 @@ const formatBrackets = {
 	// fix https://github.com/vuejs/language-tools/issues/2305
 	curly: ['0 +', '+ 0;'] as [string, string],
 	event: ['() => ', ';'] as [string, string],
+	generic: ['<', '>() => {};'] as [string, string],
 };
 
 const plugin: VueLanguagePlugin = ctx => {
@@ -71,7 +72,18 @@ const plugin: VueLanguagePlugin = ctx => {
 		return data;
 
 		function visit(node: CompilerDOM.TemplateChildNode | CompilerDOM.SimpleExpressionNode) {
-			if (node.type === CompilerDOM.NodeTypes.ELEMENT) {
+			if (node.type === CompilerDOM.NodeTypes.COMMENT) {
+				const match = node.loc.source.match(/^<!--\s*@vue-generic\b\s*\{(?<content>[^}]*)\}/);
+				if (match) {
+					const { content } = match.groups ?? {};
+					addFormatCodes(
+						content,
+						node.loc.start.offset + match[0].indexOf(content),
+						formatBrackets.generic
+					);
+				}
+			}
+			else if (node.type === CompilerDOM.NodeTypes.ELEMENT) {
 				for (const prop of node.props) {
 					if (prop.type !== CompilerDOM.NodeTypes.DIRECTIVE) {
 						continue;
diff --git a/packages/language-service/lib/plugins/vue-directive-comments.ts b/packages/language-service/lib/plugins/vue-directive-comments.ts
index 7b2504e59e..17b70b1077 100644
--- a/packages/language-service/lib/plugins/vue-directive-comments.ts
+++ b/packages/language-service/lib/plugins/vue-directive-comments.ts
@@ -1,9 +1,11 @@
 import type { CompletionItem, LanguageServicePlugin, LanguageServicePluginInstance } from '@volar/language-service';
+import type * as vscode from 'vscode-languageserver-protocol';
 
 const cmds = [
-	'vue-ignore',
-	'vue-skip',
-	'vue-expect-error',
+	['vue-ignore'],
+	['vue-skip'],
+	['vue-expect-error'],
+	['vue-generic', 'vue-generic {$1}'],
 ];
 
 const directiveCommentReg = /<!--\s*@/;
@@ -34,17 +36,17 @@ export function create(): LanguageServicePlugin {
 					const remainText = line.substring(startIndex);
 					const result: CompletionItem[] = [];
 
-					for (const cmd of cmds) {
+					for (const [label, text = label] of cmds) {
 						let match = true;
 						for (let i = 0; i < remainText.length; i++) {
-							if (remainText[i] !== cmd[i]) {
+							if (remainText[i] !== label[i]) {
 								match = false;
 								break;
 							}
 						}
 						if (match) {
 							result.push({
-								label: '@' + cmd,
+								label: '@' + label,
 								textEdit: {
 									range: {
 										start: {
@@ -53,8 +55,9 @@ export function create(): LanguageServicePlugin {
 										},
 										end: position,
 									},
-									newText: '@' + cmd,
+									newText: '@' + text,
 								},
+								insertTextFormat: 2 satisfies typeof vscode.InsertTextFormat.Snippet
 							});
 						}
 					}
diff --git a/test-workspace/tsc/passedFixtures/vue2/tsconfig.json b/test-workspace/tsc/passedFixtures/vue2/tsconfig.json
index c7626d3065..50361e864d 100644
--- a/test-workspace/tsc/passedFixtures/vue2/tsconfig.json
+++ b/test-workspace/tsc/passedFixtures/vue2/tsconfig.json
@@ -41,5 +41,6 @@
 		"../vue3/slots",
 		"../vue3/templateRef",
 		"../vue3/templateRef_native",
+		"../vue3/v-generic"
 	]
 }
diff --git a/test-workspace/tsc/passedFixtures/vue3/v-generic/comp.vue b/test-workspace/tsc/passedFixtures/vue3/v-generic/comp.vue
new file mode 100644
index 0000000000..80a6229b95
--- /dev/null
+++ b/test-workspace/tsc/passedFixtures/vue3/v-generic/comp.vue
@@ -0,0 +1,5 @@
+<script setup lang="ts" generic="T extends string | number, K = any">
+defineEmits<{
+    foo: [bar: T, baz: K]
+}>();
+</script>
\ No newline at end of file
diff --git a/test-workspace/tsc/passedFixtures/vue3/v-generic/main.vue b/test-workspace/tsc/passedFixtures/vue3/v-generic/main.vue
new file mode 100644
index 0000000000..9491d28486
--- /dev/null
+++ b/test-workspace/tsc/passedFixtures/vue3/v-generic/main.vue
@@ -0,0 +1,17 @@
+<script setup lang="ts">
+import { exactType } from '../../shared';
+import Comp from './comp.vue';
+</script>
+
+<template>
+	<!-- @vue-generic {string} -->
+	<Comp @foo="(bar) => exactType(bar, {} as string)" />
+	<!-- @vue-generic {number, boolean} -->
+	<Comp @foo="(bar, baz) => (
+		exactType(bar, {} as number),
+		exactType(baz, {} as boolean)
+	)" />
+	<!-- @vue-expect-error -->
+	<!-- @vue-generic {boolean} -->
+	<Comp />
+</template>