From 4d8d2a888333fc4b20dda166ad9e02db40dcee66 Mon Sep 17 00:00:00 2001
From: Johnson Chu <johnsoncodehk@gmail.com>
Date: Sat, 8 Mar 2025 05:08:00 +0800
Subject: [PATCH] fix(language-service): enhance provideInlayHints to support
 range and cancellationToken

---
 .../lib/plugins/vue-missing-props-hints.ts    | 46 +++++++++++--------
 1 file changed, 26 insertions(+), 20 deletions(-)

diff --git a/packages/language-service/lib/plugins/vue-missing-props-hints.ts b/packages/language-service/lib/plugins/vue-missing-props-hints.ts
index d8b6d6ca17..eb29a5d726 100644
--- a/packages/language-service/lib/plugins/vue-missing-props-hints.ts
+++ b/packages/language-service/lib/plugins/vue-missing-props-hints.ts
@@ -21,7 +21,7 @@ export function create(
 
 			return {
 
-				async provideInlayHints(document) {
+				async provideInlayHints(document, range, cancellationToken) {
 
 					if (!isSupportedDocument(document)) {
 						return;
@@ -68,29 +68,40 @@ export function create(
 					let current: {
 						unburnedRequiredProps: string[];
 						labelOffset: number;
-						insertOffset: number;
 					} | undefined;
 
 					while ((token = scanner.scan()) !== html.TokenType.EOS) {
 						if (token === html.TokenType.StartTag) {
-							const tagName = scanner.getTokenText();
-							if (intrinsicElementNames.has(tagName)) {
-								continue;
-							}
 
+							const tagName = scanner.getTokenText();
+							const tagOffset = scanner.getTokenOffset();
 							const checkTag = tagName.includes('.')
 								? tagName
 								: components.find(component => component === tagName || hyphenateTag(component) === tagName);
-							if (checkTag) {
-								componentProps[checkTag] ??= (await tsPluginClient?.getComponentProps(root.fileName, checkTag) ?? [])
+
+							if (intrinsicElementNames.has(tagName) || !checkTag) {
+								continue;
+							}
+							if (tagOffset < document.offsetAt(range.start)) {
+								continue;
+							}
+							if (tagOffset > document.offsetAt(range.end)) {
+								break;
+							}
+
+							if (!componentProps[checkTag]) {
+								if (cancellationToken.isCancellationRequested) {
+									break;
+								}
+								componentProps[checkTag] = (await tsPluginClient?.getComponentProps(root.fileName, checkTag) ?? [])
 									.filter(prop => prop.required)
 									.map(prop => prop.name);
-								current = {
-									unburnedRequiredProps: [...componentProps[checkTag]],
-									labelOffset: scanner.getTokenOffset() + scanner.getTokenLength(),
-									insertOffset: scanner.getTokenOffset() + scanner.getTokenLength(),
-								};
 							}
+
+							current = {
+								unburnedRequiredProps: [...componentProps[checkTag]],
+								labelOffset: scanner.getTokenOffset() + scanner.getTokenLength(),
+							};
 						}
 						else if (token === html.TokenType.AttributeName) {
 							if (current) {
@@ -141,8 +152,8 @@ export function create(
 										kind: 2 satisfies typeof vscode.InlayHintKind.Parameter,
 										textEdits: [{
 											range: {
-												start: document.positionAt(current.insertOffset),
-												end: document.positionAt(current.insertOffset),
+												start: document.positionAt(current.labelOffset),
+												end: document.positionAt(current.labelOffset),
 											},
 											newText: ` :${casing.attr === AttrNameCasing.Kebab ? hyphenateAttr(requiredProp) : requiredProp}=`,
 										}],
@@ -151,11 +162,6 @@ export function create(
 								current = undefined;
 							}
 						}
-						if (token === html.TokenType.AttributeName || token === html.TokenType.AttributeValue) {
-							if (current) {
-								current.insertOffset = scanner.getTokenOffset() + scanner.getTokenLength();
-							}
-						}
 					}
 
 					return result;