From 9f50c3dc2a67e8f54e88d3190c6266aebbe8ed17 Mon Sep 17 00:00:00 2001
From: Justin Chen <54879025+justschen@users.noreply.github.com>
Date: Mon, 5 Feb 2024 14:13:18 -0800
Subject: [PATCH] groundwork for code action ranges for quick fix (#204191)

* added in config

* fix balance between diagnostic and range

* cleanup
---
 extensions/typescript-language-features/package.json         | 3 ++-
 .../src/languageFeatures/quickFix.ts                         | 5 +++++
 extensions/typescript-language-features/tsconfig.json        | 1 +
 3 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/extensions/typescript-language-features/package.json b/extensions/typescript-language-features/package.json
index 9480d471455..9a5f8ff1f86 100644
--- a/extensions/typescript-language-features/package.json
+++ b/extensions/typescript-language-features/package.json
@@ -11,7 +11,8 @@
     "workspaceTrust",
     "multiDocumentHighlightProvider",
     "mappedEditsProvider",
-    "codeActionAI"
+    "codeActionAI",
+    "codeActionRanges"
   ],
   "capabilities": {
     "virtualWorkspaces": {
diff --git a/extensions/typescript-language-features/src/languageFeatures/quickFix.ts b/extensions/typescript-language-features/src/languageFeatures/quickFix.ts
index 26a658d9116..afc489c0521 100644
--- a/extensions/typescript-language-features/src/languageFeatures/quickFix.ts
+++ b/extensions/typescript-language-features/src/languageFeatures/quickFix.ts
@@ -331,6 +331,7 @@ class TypeScriptQuickFixProvider implements vscode.CodeActionProvider<VsCodeCode
 		const codeAction = new VsCodeCodeAction(action, action.description, vscode.CodeActionKind.QuickFix);
 		codeAction.edit = getEditForCodeAction(this.client, action);
 		codeAction.diagnostics = [diagnostic];
+		codeAction.ranges = [diagnostic.range];
 		codeAction.command = {
 			command: ApplyCodeActionCommand.ID,
 			arguments: [{ action, diagnostic, document } satisfies ApplyCodeActionCommand_args],
@@ -362,6 +363,8 @@ class TypeScriptQuickFixProvider implements vscode.CodeActionProvider<VsCodeCode
 				const inferFromBody = new VsCodeCodeAction(action, 'Infer types using Copilot', vscode.CodeActionKind.QuickFix);
 				inferFromBody.edit = new vscode.WorkspaceEdit();
 				inferFromBody.diagnostics = [diagnostic];
+				console.log(diagnostic, diagnostic.range);
+				inferFromBody.ranges = [diagnostic.range];
 				inferFromBody.isAI = true;
 				inferFromBody.command = {
 					command: EditorChatFollowUp.ID,
@@ -389,6 +392,7 @@ class TypeScriptQuickFixProvider implements vscode.CodeActionProvider<VsCodeCode
 				aiCodeAction.edit = getEditForCodeAction(this.client, action);
 				aiCodeAction.edit?.insert(document.uri, diagnostic.range.start, '');
 				aiCodeAction.diagnostics = [diagnostic];
+				aiCodeAction.ranges = [diagnostic.range];
 				aiCodeAction.isAI = true;
 				aiCodeAction.command = {
 					command: CompositeCommand.ID,
@@ -443,6 +447,7 @@ class TypeScriptQuickFixProvider implements vscode.CodeActionProvider<VsCodeCode
 			vscode.CodeActionKind.QuickFix);
 
 		action.diagnostics = [diagnostic];
+		action.ranges = [diagnostic.range];
 		action.command = {
 			command: ApplyFixAllCodeAction.ID,
 			arguments: [{ action } satisfies ApplyFixAllCodeAction_args],
diff --git a/extensions/typescript-language-features/tsconfig.json b/extensions/typescript-language-features/tsconfig.json
index e0c935ff517..d40c03a591d 100644
--- a/extensions/typescript-language-features/tsconfig.json
+++ b/extensions/typescript-language-features/tsconfig.json
@@ -12,6 +12,7 @@
 		"src/**/*",
 		"../../src/vscode-dts/vscode.d.ts",
 		"../../src/vscode-dts/vscode.proposed.codeActionAI.d.ts",
+		"../../src/vscode-dts/vscode.proposed.codeActionRanges.d.ts",
 		"../../src/vscode-dts/vscode.proposed.mappedEditsProvider.d.ts",
 		"../../src/vscode-dts/vscode.proposed.multiDocumentHighlightProvider.d.ts",
 		"../../src/vscode-dts/vscode.proposed.workspaceTrust.d.ts",