Bring "Improve Code" back

cte · cte · commit cbb912dd40be · 2025-04-29T00:03:54.000-07:00
diff --git a/e2e/src/suite/extension.test.ts b/e2e/src/suite/extension.test.ts
@@ -12,6 +12,7 @@ suite("Roo Code Extension", () => {
 			"roo-cline.openInNewTab",
 			"roo-cline.explainCode",
 			"roo-cline.fixCode",
+			"roo-cline.improveCode",
 		]
 
 		const commands = await vscode.commands.getCommands(true)
diff --git a/package.json b/package.json
@@ -130,6 +130,11 @@
 				"title": "%command.fixCode.title%",
 				"category": "%configuration.title%"
 			},
+			{
+				"command": "roo-cline.improveCode",
+				"title": "%command.improveCode.title%",
+				"category": "%configuration.title%"
+			},
 			{
 				"command": "roo-cline.addToContext",
 				"title": "%command.addToContext.title%",
@@ -186,6 +191,10 @@
 				{
 					"command": "roo-cline.explainCode",
 					"group": "1_actions@2"
+				},
+				{
+					"command": "roo-cline.improveCode",
+					"group": "1_actions@3"
 				}
 			],
 			"terminal/context": [
diff --git a/package.nls.ca.json b/package.nls.ca.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Nova Tasca",
 	"command.explainCode.title": "Explicar Codi",
 	"command.fixCode.title": "Corregir Codi",
+	"command.improveCode.title": "Millorar Codi",
 	"command.addToContext.title": "Afegir al Context",
 	"command.openInNewTab.title": "Obrir en una Nova Pestanya",
 	"command.focusInput.title": "Enfocar Camp d'Entrada",
diff --git a/package.nls.de.json b/package.nls.de.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Neue Aufgabe",
 	"command.explainCode.title": "Code Erklären",
 	"command.fixCode.title": "Code Reparieren",
+	"command.improveCode.title": "Code Verbessern",
 	"command.addToContext.title": "Zum Kontext Hinzufügen",
 	"command.openInNewTab.title": "In Neuem Tab Öffnen",
 	"command.focusInput.title": "Eingabefeld Fokussieren",
diff --git a/package.nls.es.json b/package.nls.es.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Nueva Tarea",
 	"command.explainCode.title": "Explicar Código",
 	"command.fixCode.title": "Corregir Código",
+	"command.improveCode.title": "Mejorar Código",
 	"command.addToContext.title": "Añadir al Contexto",
 	"command.openInNewTab.title": "Abrir en Nueva Pestaña",
 	"command.focusInput.title": "Enfocar Campo de Entrada",
diff --git a/package.nls.fr.json b/package.nls.fr.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Nouvelle Tâche",
 	"command.explainCode.title": "Expliquer le Code",
 	"command.fixCode.title": "Corriger le Code",
+	"command.improveCode.title": "Améliorer le Code",
 	"command.addToContext.title": "Ajouter au Contexte",
 	"command.openInNewTab.title": "Ouvrir dans un Nouvel Onglet",
 	"command.focusInput.title": "Focus sur le Champ de Saisie",
diff --git a/package.nls.hi.json b/package.nls.hi.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "नया कार्य",
 	"command.explainCode.title": "कोड समझाएं",
 	"command.fixCode.title": "कोड ठीक करें",
+	"command.improveCode.title": "कोड सुधारें",
 	"command.addToContext.title": "संदर्भ में जोड़ें",
 	"command.openInNewTab.title": "नए टैब में खोलें",
 	"command.focusInput.title": "इनपुट फ़ील्ड पर फोकस करें",
diff --git a/package.nls.it.json b/package.nls.it.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Nuovo Task",
 	"command.explainCode.title": "Spiega Codice",
 	"command.fixCode.title": "Correggi Codice",
+	"command.improveCode.title": "Migliora Codice",
 	"command.addToContext.title": "Aggiungi al Contesto",
 	"command.openInNewTab.title": "Apri in Nuova Scheda",
 	"command.focusInput.title": "Focalizza Campo di Input",
diff --git a/package.nls.ja.json b/package.nls.ja.json
@@ -14,6 +14,7 @@
 	"command.openInNewTab.title": "新しいタブで開く",
 	"command.explainCode.title": "コードの説明",
 	"command.fixCode.title": "コードの修正",
+	"command.improveCode.title": "コードの改善",
 	"command.addToContext.title": "コンテキストに追加",
 	"command.focusInput.title": "入力フィールドにフォーカス",
 	"command.setCustomStoragePath.title": "カスタムストレージパスの設定",
diff --git a/package.nls.json b/package.nls.json
@@ -14,6 +14,7 @@
 	"command.openInNewTab.title": "Open In New Tab",
 	"command.explainCode.title": "Explain Code",
 	"command.fixCode.title": "Fix Code",
+	"command.improveCode.title": "Improve Code",
 	"command.addToContext.title": "Add To Context",
 	"command.focusInput.title": "Focus Input Field",
 	"command.setCustomStoragePath.title": "Set Custom Storage Path",
diff --git a/package.nls.ko.json b/package.nls.ko.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "새 작업",
 	"command.explainCode.title": "코드 설명",
 	"command.fixCode.title": "코드 수정",
+	"command.improveCode.title": "코드 개선",
 	"command.addToContext.title": "컨텍스트에 추가",
 	"command.openInNewTab.title": "새 탭에서 열기",
 	"command.focusInput.title": "입력 필드 포커스",
diff --git a/package.nls.pl.json b/package.nls.pl.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Nowe Zadanie",
 	"command.explainCode.title": "Wyjaśnij Kod",
 	"command.fixCode.title": "Napraw Kod",
+	"command.improveCode.title": "Ulepsz Kod",
 	"command.addToContext.title": "Dodaj do Kontekstu",
 	"command.openInNewTab.title": "Otwórz w Nowej Karcie",
 	"command.focusInput.title": "Fokus na Pole Wprowadzania",
diff --git a/package.nls.pt-BR.json b/package.nls.pt-BR.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Nova Tarefa",
 	"command.explainCode.title": "Explicar Código",
 	"command.fixCode.title": "Corrigir Código",
+	"command.improveCode.title": "Melhorar Código",
 	"command.addToContext.title": "Adicionar ao Contexto",
 	"command.openInNewTab.title": "Abrir em Nova Aba",
 	"command.focusInput.title": "Focar Campo de Entrada",
diff --git a/package.nls.ru.json b/package.nls.ru.json
@@ -14,6 +14,7 @@
 	"command.openInNewTab.title": "Открыть в новой вкладке",
 	"command.explainCode.title": "Объяснить код",
 	"command.fixCode.title": "Исправить код",
+	"command.improveCode.title": "Улучшить код",
 	"command.addToContext.title": "Добавить в контекст",
 	"command.focusInput.title": "Фокус на поле ввода",
 	"command.setCustomStoragePath.title": "Указать путь хранения",
diff --git a/package.nls.tr.json b/package.nls.tr.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Yeni Görev",
 	"command.explainCode.title": "Kodu Açıkla",
 	"command.fixCode.title": "Kodu Düzelt",
+	"command.improveCode.title": "Kodu İyileştir",
 	"command.addToContext.title": "Bağlama Ekle",
 	"command.openInNewTab.title": "Yeni Sekmede Aç",
 	"command.focusInput.title": "Giriş Alanına Odaklan",
diff --git a/package.nls.vi.json b/package.nls.vi.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "Tác Vụ Mới",
 	"command.explainCode.title": "Giải Thích Mã",
 	"command.fixCode.title": "Sửa Mã",
+	"command.improveCode.title": "Cải Thiện Mã",
 	"command.addToContext.title": "Thêm vào Ngữ Cảnh",
 	"command.openInNewTab.title": "Mở trong Tab Mới",
 	"command.focusInput.title": "Tập Trung vào Trường Nhập",
diff --git a/package.nls.zh-CN.json b/package.nls.zh-CN.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "新建任务",
 	"command.explainCode.title": "解释代码",
 	"command.fixCode.title": "修复代码",
+	"command.improveCode.title": "改进代码",
 	"command.addToContext.title": "添加到上下文",
 	"command.openInNewTab.title": "在新标签页中打开",
 	"command.focusInput.title": "聚焦输入框",
diff --git a/package.nls.zh-TW.json b/package.nls.zh-TW.json
@@ -4,6 +4,7 @@
 	"command.newTask.title": "新建任務",
 	"command.explainCode.title": "解釋程式碼",
 	"command.fixCode.title": "修復程式碼",
+	"command.improveCode.title": "改進程式碼",
 	"command.addToContext.title": "添加到上下文",
 	"command.openInNewTab.title": "在新分頁中開啟",
 	"command.focusInput.title": "聚焦輸入框",
diff --git a/src/activate/registerCodeActions.ts b/src/activate/registerCodeActions.ts
@@ -1,19 +1,16 @@
 import * as vscode from "vscode"
 
-import { ACTION_NAMES, COMMAND_IDS } from "../core/CodeActionProvider"
+import { type CodeActionName, type CodeActionId, COMMAND_IDS } from "../core/CodeActionProvider"
 import { EditorUtils } from "../core/EditorUtils"
 import { ClineProvider } from "../core/webview/ClineProvider"
 
 export const registerCodeActions = (context: vscode.ExtensionContext) => {
+	registerCodeAction(context, COMMAND_IDS.EXPLAIN, "EXPLAIN")
 	registerCodeAction(context, COMMAND_IDS.FIX, "FIX")
 	registerCodeAction(context, COMMAND_IDS.ADD_TO_CONTEXT, "ADD_TO_CONTEXT")
 }
 
-const registerCodeAction = (
-	context: vscode.ExtensionContext,
-	command: string,
-	promptType: keyof typeof ACTION_NAMES,
-) => {
+const registerCodeAction = (context: vscode.ExtensionContext, command: CodeActionId, promptType: CodeActionName) => {
 	let userInput: string | undefined
 
 	context.subscriptions.push(
@@ -31,7 +28,11 @@ const registerCodeAction = (
 			} else {
 				// Called directly from command palette.
 				const context = EditorUtils.getEditorContext()
-				if (!context) return
+
+				if (!context) {
+					return
+				}
+
 				;({ filePath, selectedText, startLine, endLine, diagnostics } = context)
 			}
 
diff --git a/src/core/CodeActionProvider.ts b/src/core/CodeActionProvider.ts
@@ -2,7 +2,13 @@ import * as vscode from "vscode"
 
 import { EditorUtils } from "./EditorUtils"
 
-type CodeActionName = "EXPLAIN" | "FIX" | "ADD_TO_CONTEXT" | "NEW_TASK"
+export type CodeActionName = "EXPLAIN" | "FIX" | "ADD_TO_CONTEXT" | "NEW_TASK"
+
+export type CodeActionId =
+	| "roo-cline.explainCode"
+	| "roo-cline.fixCode"
+	| "roo-cline.addToContext"
+	| "roo-cline.newTask"
 
 export const ACTION_NAMES: Record<CodeActionName, string> = {
 	EXPLAIN: "Explain with Roo Code",
@@ -11,7 +17,7 @@ export const ACTION_NAMES: Record<CodeActionName, string> = {
 	NEW_TASK: "New Roo Code Task",
 } as const
 
-export const COMMAND_IDS: Record<CodeActionName, string> = {
+export const COMMAND_IDS: Record<CodeActionName, CodeActionId> = {
 	EXPLAIN: "roo-cline.explainCode",
 	FIX: "roo-cline.fixCode",
 	ADD_TO_CONTEXT: "roo-cline.addToContext",

Original file line number	Diff line number	Diff line change
`@@ -12,6 +12,7 @@ suite("Roo Code Extension", () => {`
`12`	`12`	`"roo-cline.openInNewTab",`
`13`	`13`	`"roo-cline.explainCode",`
`14`	`14`	`"roo-cline.fixCode",`
	`15`	`+ "roo-cline.improveCode",`
`15`	`16`	`]`
`16`	`17`
`17`	`18`	`const commands = await vscode.commands.getCommands(true)`