Add "Start Coding Agent Session" code action for TODO comments (#7678)

* Initial plan

* Add Start Coding Agent Session code action for TODO comments

* Fix formatting issues and finalize TODO to coding agent feature

* 💄

* Complete TODO coding agent feature by adding command registration to package.json

Co-authored-by: joshspicer <23246594+joshspicer@users.noreply.github.com>

* Revert "Complete TODO coding agent feature by adding command registration to package.json"

This reverts commit 8b31e74.

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: Alex Ross <38270282+alexr00@users.noreply.github.com>
Co-authored-by: joshspicer <23246594+joshspicer@users.noreply.github.com>

Author: 3 people

src/extension.ts

@@ -235,7 +235,7 @@ async function init(
 	const layout = vscode.workspace.getConfiguration(PR_SETTINGS_NAMESPACE).get<string>(FILE_LIST_LAYOUT);
 	await vscode.commands.executeCommand('setContext', 'fileListLayout:flat', layout === 'flat');
 
-	const issuesFeatures = new IssueFeatureRegistrar(git, reposManager, reviewsManager, context, telemetry);
+	const issuesFeatures = new IssueFeatureRegistrar(git, reposManager, reviewsManager, context, telemetry, copilotRemoteAgentManager);
 	context.subscriptions.push(issuesFeatures);
 	await issuesFeatures.initialize();
 

src/issues/issueFeatureRegistrar.ts

@@ -24,6 +24,7 @@ import { ITelemetry } from '../common/telemetry';
 import { fromRepoUri, RepoUriParams, Schemes, toNewIssueUri } from '../common/uri';
 import { EXTENSION_ID } from '../constants';
 import { OctokitCommon } from '../github/common';
+import { CopilotRemoteAgentManager } from '../github/copilotRemoteAgent';
 import { FolderRepositoryManager, PullRequestDefaults } from '../github/folderRepositoryManager';
 import { IProject } from '../github/interface';
 import { IssueModel } from '../github/issueModel';
@@ -91,6 +92,7 @@ export class IssueFeatureRegistrar extends Disposable {
 		private reviewsManager: ReviewsManager,
 		private context: vscode.ExtensionContext,
 		private telemetry: ITelemetry,
+		private copilotRemoteAgentManager: CopilotRemoteAgentManager,
 	) {
 		super();
 		this._stateManager = new StateManager(gitAPI, this.manager, this.context);
@@ -140,6 +142,19 @@ export class IssueFeatureRegistrar extends Disposable {
 				this,
 			),
 		);
+		this._register(
+			vscode.commands.registerCommand(
+				'issue.startCodingAgentFromTodo',
+				(todoInfo?: { document: vscode.TextDocument; lineNumber: number; line: string; insertIndex: number; range: vscode.Range }) => {
+					/* __GDPR__
+				"issue.startCodingAgentFromTodo" : {}
+			*/
+					this.telemetry.sendTelemetryEvent('issue.startCodingAgentFromTodo');
+					return this.startCodingAgentFromTodo(todoInfo);
+				},
+				this,
+			),
+		);
 		this._register(
 			vscode.commands.registerCommand(
 				'issue.copyGithubPermalink',
@@ -548,7 +563,7 @@ export class IssueFeatureRegistrar extends Disposable {
 				vscode.languages.registerHoverProvider('*', new UserHoverProvider(this.manager, this.telemetry)),
 			);
 			this._register(
-				vscode.languages.registerCodeActionsProvider('*', new IssueTodoProvider(this.context), { providedCodeActionKinds: [vscode.CodeActionKind.QuickFix] }),
+				vscode.languages.registerCodeActionsProvider('*', new IssueTodoProvider(this.context, this.copilotRemoteAgentManager), { providedCodeActionKinds: [vscode.CodeActionKind.QuickFix] }),
 			);
 		});
 	}
@@ -1453,4 +1468,34 @@ ${options?.body ?? ''}\n
 		}
 		return undefined;
 	}
+
+	async startCodingAgentFromTodo(todoInfo?: { document: vscode.TextDocument; lineNumber: number; line: string; insertIndex: number; range: vscode.Range }) {
+		if (!todoInfo) {
+			return;
+		}
+		
+		const { document, line, insertIndex } = todoInfo;
+		
+		// Extract the TODO text after the trigger word
+		const todoText = line.substring(insertIndex).trim();
+		
+		if (!todoText) {
+			vscode.window.showWarningMessage(vscode.l10n.t('No task description found in TODO comment'));
+			return;
+		}
+
+		// Create a prompt for the coding agent
+		const relativePath = vscode.workspace.asRelativePath(document.uri);
+		const prompt = vscode.l10n.t('Work on TODO: {0} (from {1})', todoText, relativePath);
+
+		// Start the coding agent session
+		try {
+			await this.copilotRemoteAgentManager.commandImpl({
+				userPrompt: prompt,
+				source: 'todo'
+			});
+		} catch (error) {
+			vscode.window.showErrorMessage(vscode.l10n.t('Failed to start coding agent session: {0}', error.message));
+		}
+	}
 }

src/issues/issueTodoProvider.ts

@@ -6,13 +6,17 @@
 import * as vscode from 'vscode';
 import { CREATE_ISSUE_TRIGGERS, ISSUES_SETTINGS_NAMESPACE } from '../common/settingKeys';
 import { escapeRegExp } from '../common/utils';
+import { CopilotRemoteAgentManager } from '../github/copilotRemoteAgent';
 import { ISSUE_OR_URL_EXPRESSION } from '../github/utils';
 import { MAX_LINE_LENGTH } from './util';
 
 export class IssueTodoProvider implements vscode.CodeActionProvider {
 	private expression: RegExp | undefined;
 
-	constructor(context: vscode.ExtensionContext) {
+	constructor(
+		context: vscode.ExtensionContext,
+		private copilotRemoteAgentManager: CopilotRemoteAgentManager
+	) {
 		context.subscriptions.push(
 			vscode.workspace.onDidChangeConfiguration(() => {
 				this.updateTriggers();
@@ -45,21 +49,37 @@ export class IssueTodoProvider implements vscode.CodeActionProvider {
 				const match = truncatedLine.match(this.expression);
 				const search = match?.index ?? -1;
 				if (search >= 0 && match) {
-					const codeAction: vscode.CodeAction = new vscode.CodeAction(
+					// Create GitHub Issue action
+					const createIssueAction: vscode.CodeAction = new vscode.CodeAction(
 						vscode.l10n.t('Create GitHub Issue'),
 						vscode.CodeActionKind.QuickFix,
 					);
-					codeAction.ranges = [new vscode.Range(lineNumber, search, lineNumber, search + match[0].length)];
+					createIssueAction.ranges = [new vscode.Range(lineNumber, search, lineNumber, search + match[0].length)];
 					const indexOfWhiteSpace = truncatedLine.substring(search).search(/\s/);
 					const insertIndex =
 						search +
 						(indexOfWhiteSpace > 0 ? indexOfWhiteSpace : truncatedLine.match(this.expression)![0].length);
-					codeAction.command = {
+					createIssueAction.command = {
 						title: vscode.l10n.t('Create GitHub Issue'),
 						command: 'issue.createIssueFromSelection',
 						arguments: [{ document, lineNumber, line, insertIndex, range }],
 					};
-					codeActions.push(codeAction);
+					codeActions.push(createIssueAction);
+
+					// Start Coding Agent Session action (if copilot manager is available)
+					if (this.copilotRemoteAgentManager) {
+						const startAgentAction: vscode.CodeAction = new vscode.CodeAction(
+							vscode.l10n.t('Start Coding Agent Session'),
+							vscode.CodeActionKind.QuickFix,
+						);
+						startAgentAction.ranges = [new vscode.Range(lineNumber, search, lineNumber, search + match[0].length)];
+						startAgentAction.command = {
+							title: vscode.l10n.t('Start Coding Agent Session'),
+							command: 'issue.startCodingAgentFromTodo',
+							arguments: [{ document, lineNumber, line, insertIndex, range }],
+						};
+						codeActions.push(startAgentAction);
+					}
 					break;
 				}
 			}

src/test/issues/issueTodoProvider.test.ts

@@ -0,0 +1,45 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { default as assert } from 'assert';
+import * as vscode from 'vscode';
+import { IssueTodoProvider } from '../../issues/issueTodoProvider';
+
+describe('IssueTodoProvider', function () {
+	it('should provide both actions when CopilotRemoteAgentManager is available', async function () {
+		const mockContext = {
+			subscriptions: []
+		} as any as vscode.ExtensionContext;
+
+		const mockCopilotManager = {} as any; // Mock CopilotRemoteAgentManager
+
+		const provider = new IssueTodoProvider(mockContext, mockCopilotManager);
+
+		// Create a mock document with TODO comment
+		const document = {
+			lineAt: (line: number) => ({ text: line === 1 ? '  // TODO: Fix this' : 'function test() {' }),
+			lineCount: 4
+		} as vscode.TextDocument;
+
+		const range = new vscode.Range(1, 0, 1, 20);
+		const context = {
+			only: vscode.CodeActionKind.QuickFix
+		} as vscode.CodeActionContext;
+
+		const actions = await provider.provideCodeActions(document, range, context, new vscode.CancellationTokenSource().token);
+
+		assert.strictEqual(actions.length, 2);
+
+		// Find the actions
+		const createIssueAction = actions.find(a => a.title === 'Create GitHub Issue');
+		const startAgentAction = actions.find(a => a.title === 'Start Coding Agent Session');
+
+		assert.ok(createIssueAction, 'Should have Create GitHub Issue action');
+		assert.ok(startAgentAction, 'Should have Start Coding Agent Session action');
+
+		assert.strictEqual(createIssueAction?.command?.command, 'issue.createIssueFromSelection');
+		assert.strictEqual(startAgentAction?.command?.command, 'issue.startCodingAgentFromTodo');
+	});
+});