Implement CodeLens provider for TODO comments

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

Author: Copilot

src/issues/issueFeatureRegistrar.ts

@@ -575,8 +575,12 @@ export class IssueFeatureRegistrar extends Disposable {
 			this._register(
 				vscode.languages.registerHoverProvider('*', new UserHoverProvider(this.manager, this.telemetry)),
 			);
+			const todoProvider = new IssueTodoProvider(this.context, this.copilotRemoteAgentManager);
 			this._register(
-				vscode.languages.registerCodeActionsProvider('*', new IssueTodoProvider(this.context, this.copilotRemoteAgentManager), { providedCodeActionKinds: [vscode.CodeActionKind.QuickFix] }),
+				vscode.languages.registerCodeActionsProvider('*', todoProvider, { providedCodeActionKinds: [vscode.CodeActionKind.QuickFix] }),
+			);
+			this._register(
+				vscode.languages.registerCodeLensProvider('*', todoProvider),
 			);
 		});
 	}

src/issues/issueTodoProvider.ts

@@ -10,7 +10,7 @@ import { escapeRegExp } from '../common/utils';
 import { CopilotRemoteAgentManager } from '../github/copilotRemoteAgent';
 import { ISSUE_OR_URL_EXPRESSION } from '../github/utils';
 
-export class IssueTodoProvider implements vscode.CodeActionProvider {
+export class IssueTodoProvider implements vscode.CodeActionProvider, vscode.CodeLensProvider {
 	private expression: RegExp | undefined;
 
 	constructor(
@@ -87,4 +87,50 @@ export class IssueTodoProvider implements vscode.CodeActionProvider {
 		} while (range.end.line >= lineNumber);
 		return codeActions;
 	}
+
+	provideCodeLenses(
+		document: vscode.TextDocument,
+		_token: vscode.CancellationToken,
+	): vscode.CodeLens[] {
+		if (this.expression === undefined) {
+			return [];
+		}
+
+		const codeLenses: vscode.CodeLens[] = [];
+		for (let lineNumber = 0; lineNumber < document.lineCount; lineNumber++) {
+			const line = document.lineAt(lineNumber).text;
+			const truncatedLine = line.substring(0, MAX_LINE_LENGTH);
+			const matches = truncatedLine.match(ISSUE_OR_URL_EXPRESSION);
+			if (!matches) {
+				const match = truncatedLine.match(this.expression);
+				const search = match?.index ?? -1;
+				if (search >= 0 && match) {
+					const range = 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);
+
+					// Create GitHub Issue CodeLens
+					const createIssueCodeLens = new vscode.CodeLens(range, {
+						title: vscode.l10n.t('Create GitHub Issue'),
+						command: 'issue.createIssueFromSelection',
+						arguments: [{ document, lineNumber, line, insertIndex, range }],
+					});
+					codeLenses.push(createIssueCodeLens);
+
+					// Delegate to coding agent CodeLens (if copilot manager is available)
+					if (this.copilotRemoteAgentManager) {
+						const startAgentCodeLens = new vscode.CodeLens(range, {
+							title: vscode.l10n.t('Delegate to coding agent'),
+							command: 'issue.startCodingAgentFromTodo',
+							arguments: [{ document, lineNumber, line, insertIndex, range }],
+						});
+						codeLenses.push(startAgentCodeLens);
+					}
+				}
+			}
+		}
+		return codeLenses;
+	}
 }

src/test/issues/issueTodoProvider.test.ts

@@ -42,4 +42,40 @@ describe('IssueTodoProvider', function () {
 		assert.strictEqual(createIssueAction?.command?.command, 'issue.createIssueFromSelection');
 		assert.strictEqual(startAgentAction?.command?.command, 'issue.startCodingAgentFromTodo');
 	});
+
+	it('should provide code lenses for TODO comments', 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 codeLenses = provider.provideCodeLenses(document, new vscode.CancellationTokenSource().token);
+
+		assert.strictEqual(codeLenses.length, 2);
+
+		// Verify the code lenses
+		const createIssueLens = codeLenses.find(cl => cl.command?.title === 'Create GitHub Issue');
+		const startAgentLens = codeLenses.find(cl => cl.command?.title === 'Delegate to coding agent');
+
+		assert.ok(createIssueLens, 'Should have Create GitHub Issue CodeLens');
+		assert.ok(startAgentLens, 'Should have Delegate to coding agent CodeLens');
+
+		assert.strictEqual(createIssueLens?.command?.command, 'issue.createIssueFromSelection');
+		assert.strictEqual(startAgentLens?.command?.command, 'issue.startCodingAgentFromTodo');
+
+		// Verify the range points to the TODO text
+		assert.strictEqual(createIssueLens?.range.start.line, 1);
+		assert.strictEqual(startAgentLens?.range.start.line, 1);
+	});
 });