Implement GitHub Coding Agent availability and assignability context setting (#7105)

* Initial plan

* Changes before error encountered

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

* Remove excessive tests to simplify implementation

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

* remove gate for pr tool

* Initial plan

* Implement GitHub Coding Agent availability and assignability context setting

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

* Update remoteCodingAgents contribution with id and context condition

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

* tidy up

---------

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

Author: Copilot

package.json

@@ -67,10 +67,11 @@
 	"contributes": {
 		"remoteCodingAgents": [
 			{
+				"id": "githubCodingAgent",
 				"command": "githubpr.remoteAgent",
 				"displayName": "GitHub Coding Agent",
 				"description": "The GitHub Coding Agent.",
-				"when": "config.githubPullRequests.codingAgent.enabled"
+				"when": "config.githubPullRequests.codingAgent.enabled && copilotCodingAgentAssignable"
 			}
 		],
 		"chatParticipants": [

src/github/copilotRemoteAgent.ts

@@ -6,6 +6,8 @@
 import vscode from 'vscode';
 import { Repository } from '../api/api';
 import { AuthProvider } from '../common/authentication';
+import { COPILOT_LOGINS } from '../common/copilot';
+import { commands } from '../common/executeCommands';
 import { Disposable } from '../common/lifecycle';
 import { Remote } from '../common/remote';
 import { CODING_AGENT, CODING_AGENT_AUTO_COMMIT_AND_PUSH, CODING_AGENT_ENABLED } from '../common/settingKeys';
@@ -55,6 +57,28 @@ export class CopilotRemoteAgentManager extends Disposable {
 		this._register(new CopilotPRWatcher(this.repositoriesManager, this._stateModel));
 		this._register(this._stateModel.onDidChangeStates(() => this._onDidChangeStates.fire()));
 		this._register(this._stateModel.onDidChangeNotifications(() => this._onDidChangeNotifications.fire()));
+
+		this._register(this.repositoriesManager.onDidChangeFolderRepositories((event) => {
+			if (event.added) {
+				this._register(event.added.onDidChangeAssignableUsers(() => {
+					this.updateAssignabilityContext();
+				}));
+			}
+			this.updateAssignabilityContext();
+		}));
+		this.repositoriesManager.folderManagers.forEach(manager => {
+			this._register(manager.onDidChangeAssignableUsers(() => {
+				this.updateAssignabilityContext();
+			}));
+		});
+		this._register(vscode.workspace.onDidChangeConfiguration((e) => {
+			if (e.affectsConfiguration(CODING_AGENT)) {
+				this.updateAssignabilityContext();
+			}
+		}));
+
+		// Set initial context
+		this.updateAssignabilityContext();
 	}
 
 	private _copilotApiPromise: Promise<CopilotApi | undefined> | undefined;
@@ -79,6 +103,61 @@ export class CopilotRemoteAgentManager extends Disposable {
 			.getConfiguration(CODING_AGENT).get(CODING_AGENT_ENABLED, false);
 	}
 
+	async isAssignable(): Promise<boolean> {
+		const repoInfo = await this.repoInfo();
+		if (!repoInfo) {
+			return false;
+		}
+
+		const { owner, repo } = repoInfo;
+		const folderManager = this.getFolderManagerForRepo(owner, repo);
+
+		try {
+			// Ensure assignable users are loaded
+			await folderManager.getAssignableUsers();
+			const allAssignableUsers = folderManager.getAllAssignableUsers();
+
+			if (!allAssignableUsers) {
+				return false;
+			}
+
+			// Check if any of the copilot logins are in the assignable users
+			return allAssignableUsers.some(user => COPILOT_LOGINS.includes(user.login));
+		} catch (error) {
+			// If there's an error fetching assignable users, assume not assignable
+			return false;
+		}
+	}
+
+	async isAvailable(): Promise<boolean> {
+		// Check if the manager is enabled, copilot API is available, and it's assignable
+		if (!this.enabled()) {
+			return false;
+		}
+
+		const repoInfo = await this.repoInfo();
+		if (!repoInfo) {
+			return false;
+		}
+
+		const copilotApi = await this.copilotApi;
+		if (!copilotApi) {
+			return false;
+		}
+
+		return await this.isAssignable();
+	}
+
+	private async updateAssignabilityContext(): Promise<void> {
+		try {
+			const available = await this.isAvailable();
+			commands.setContext('copilotCodingAgentAssignable', available);
+		} catch (error) {
+			// Presume false
+			commands.setContext('copilotCodingAgentAssignable', false);
+		}
+	}
+
 	autoCommitAndPushEnabled(): boolean {
 		return vscode.workspace
 			.getConfiguration(CODING_AGENT).get(CODING_AGENT_AUTO_COMMIT_AND_PUSH, false);
@@ -114,6 +193,13 @@ export class CopilotRemoteAgentManager extends Disposable {
 	}
 
 	async commandImpl(args?: any) {
+		// Check if the coding agent is available (enabled and assignable)
+		const isAvailable = await this.isAvailable();
+		if (!isAvailable) {
+			vscode.window.showWarningMessage(vscode.l10n.t('GitHub Coding Agent is not available for this repository. Make sure the agent is enabled and assignable to this repository.'));
+			return;
+		}
+
 		// https://github.com/microsoft/vscode-copilot/issues/18918
 		const userPrompt: string | undefined = args.userPrompt;
 		const summary: string | undefined = args.summary;

src/lm/tools/copilotRemoteAgentTool.ts

@@ -25,6 +25,13 @@ export class CopilotRemoteAgentTool implements vscode.LanguageModelTool<CopilotR
 
 	async prepareInvocation(options: vscode.LanguageModelToolInvocationPrepareOptions<CopilotRemoteAgentToolParameters>): Promise<vscode.PreparedToolInvocation> {
 		const { title } = options.input;
+		
+		// Check if the coding agent is available (enabled and assignable)
+		const isAvailable = await this.manager.isAvailable();
+		if (!isAvailable) {
+			throw new Error(vscode.l10n.t('GitHub Coding Agent is not available for this repository. Make sure the agent is enabled and assignable to this repository.'));
+		}
+
 		const targetRepo = await this.manager.repoInfo();
 		const autoPushEnabled = this.manager.autoCommitAndPushEnabled();
 		return {