Disables and hides code suggest on non-GitHub PRs

Author: axosoft-ramint

src/plus/drafts/draftsService.ts

@@ -25,7 +25,7 @@ import { getSettledValue } from '../../system/promise';
 import type { OrganizationMember } from '../gk/models/organization';
 import type { SubscriptionAccount } from '../gk/models/subscription';
 import type { ServerConnection } from '../gk/serverConnection';
-import { providersMetadata } from '../integrations/providers/models';
+import { providersMetadata, supportsCodeSuggest } from '../integrations/providers/models';
 import { getEntityIdentifierInput } from '../integrations/providers/utils';
 import type { LaunchpadItem } from '../launchpad/launchpadProvider';
 import type {
@@ -832,6 +832,8 @@ export class DraftService implements Disposable {
 		repositoryOrIntegrationId: Repository | IntegrationId,
 		options?: { includeArchived?: boolean },
 	): Promise<Draft[]> {
+		if (!supportsCodeSuggest(item.provider)) return [];
+
 		const entityIdentifier = getEntityIdentifierInput(item);
 		const prEntityId = EntityIdentifierUtils.encode(entityIdentifier);
 		const providerAuth = await this.getProviderAuthFromRepoOrIntegrationId(repositoryOrIntegrationId);
@@ -855,9 +857,11 @@ export class DraftService implements Disposable {
 
 		type Result = { data: CodeSuggestionCountsResponse };
 
-		const prEntityIds = pullRequests.map(pr => {
-			return EntityIdentifierUtils.encode(getEntityIdentifierInput(pr));
-		});
+		const prEntityIds = pullRequests
+			.filter(pr => supportsCodeSuggest(pr.provider))
+			.map(pr => {
+				return EntityIdentifierUtils.encode(getEntityIdentifierInput(pr));
+			});
 
 		const body = JSON.stringify({
 			prEntityIds: prEntityIds,

src/plus/integrations/integrationService.ts

@@ -52,9 +52,14 @@ import type {
 	SupportedSelfHostedIntegrationIds,
 } from './integration';
 import { isAzureCloudDomain } from './providers/azureDevOps';
-import { isCloudSelfHostedIntegrationId, isHostingIntegrationId, isSelfHostedIntegrationId } from './providers/models';
+import {
+	isCloudSelfHostedIntegrationId,
+	isGitHubDotCom,
+	isGitLabDotCom,
+	isHostingIntegrationId,
+	isSelfHostedIntegrationId,
+} from './providers/models';
 import type { ProvidersApi } from './providers/providersApi';
-import { isGitHubDotCom, isGitLabDotCom } from './providers/utils';
 
 export interface ConnectionStateChangeEvent {
 	key: string;

src/plus/integrations/providers/models.ts

@@ -53,6 +53,7 @@ import {
 	PullRequestStatusCheckRollupState,
 } from '../../../git/models/pullRequest';
 import type { ProviderReference } from '../../../git/models/remoteProvider';
+import { equalsIgnoreCase } from '../../../system/string';
 import type { EnrichableItem } from '../../launchpad/models/enrichedItem';
 import type { Integration, IntegrationType } from '../integration';
 import { getEntityIdentifierInput } from './utils';
@@ -1064,3 +1065,15 @@ export type GitConfigEntityIdentifier = AnyEntityIdentifierInput & {
 		isCloudEnterprise?: boolean;
 	};
 };
+
+export function isGitHubDotCom(domain: string): boolean {
+	return equalsIgnoreCase(domain, 'github.com');
+}
+
+export function isGitLabDotCom(domain: string): boolean {
+	return equalsIgnoreCase(domain, 'gitlab.com');
+}
+
+export function supportsCodeSuggest(provider: ProviderReference): boolean {
+	return isGitHubDotCom(provider.domain);
+}

src/plus/integrations/providers/utils.ts

@@ -7,21 +7,12 @@ import type { Issue, IssueShape } from '../../../git/models/issue';
 import type { IssueOrPullRequest } from '../../../git/models/issueOrPullRequest';
 import type { PullRequest } from '../../../git/models/pullRequest';
 import { Logger } from '../../../system/logger';
-import { equalsIgnoreCase } from '../../../system/string';
 import type { LaunchpadItem } from '../../launchpad/launchpadProvider';
 import type { IssueResourceDescriptor, RepositoryDescriptor } from '../integration';
 import { isIssueResourceDescriptor, isRepositoryDescriptor } from '../integration';
 import type { AzureProjectInputDescriptor } from './azure/models';
 import type { GitConfigEntityIdentifier } from './models';
-import { isCloudSelfHostedIntegrationId } from './models';
-
-export function isGitHubDotCom(domain: string): boolean {
-	return equalsIgnoreCase(domain, 'github.com');
-}
-
-export function isGitLabDotCom(domain: string): boolean {
-	return equalsIgnoreCase(domain, 'gitlab.com');
-}
+import { isCloudSelfHostedIntegrationId, isGitHubDotCom, isGitLabDotCom } from './models';
 
 function isLaunchpadItem(item: IssueOrPullRequest | LaunchpadItem): item is LaunchpadItem {
 	return (item as LaunchpadItem).uuid !== undefined;

src/plus/launchpad/launchpadProvider.ts

@@ -16,6 +16,7 @@ import type { Account } from '../../git/models/author';
 import type { GitBranch } from '../../git/models/branch';
 import type { PullRequest, SearchedPullRequest } from '../../git/models/pullRequest';
 import type { GitRemote } from '../../git/models/remote';
+import type { ProviderReference } from '../../git/models/remoteProvider';
 import type { Repository } from '../../git/models/repository';
 import { getOrOpenPullRequestRepository } from '../../git/utils/-webview/pullRequest.utils';
 import type { PullRequestUrlIdentity } from '../../git/utils/pullRequest.utils';
@@ -46,7 +47,11 @@ import type { ConnectionStateChangeEvent } from '../integrations/integrationServ
 import { isMaybeGitHubPullRequestUrl } from '../integrations/providers/github/github.utils';
 import { isMaybeGitLabPullRequestUrl } from '../integrations/providers/gitlab/gitlab.utils';
 import type { EnrichablePullRequest, ProviderActionablePullRequest } from '../integrations/providers/models';
-import { getActionablePullRequests, toProviderPullRequestWithUniqueId } from '../integrations/providers/models';
+import {
+	getActionablePullRequests,
+	supportsCodeSuggest,
+	toProviderPullRequestWithUniqueId,
+} from '../integrations/providers/models';
 import {
 	convertIntegrationIdToEnrichProvider,
 	convertRemoteProviderIdToEnrichProvider,
@@ -63,12 +68,26 @@ import {
 	sharedCategoryToLaunchpadActionCategoryMap,
 } from './models/launchpad';
 
-export function getSuggestedActions(category: LaunchpadActionCategory, isCurrentBranch: boolean): LaunchpadAction[] {
+export function getSuggestedActions(
+	category: LaunchpadActionCategory,
+	provider: ProviderReference,
+	isCurrentBranch: boolean,
+): LaunchpadAction[] {
 	const actions = [...prActionsMap.get(category)!];
 	if (isCurrentBranch) {
-		actions.push('show-overview', 'open-changes', 'code-suggest', 'open-in-graph');
+		actions.push('show-overview', 'open-changes');
+		if (supportsCodeSuggest(provider)) {
+			actions.push('code-suggest');
+		}
+
+		actions.push('open-in-graph');
 	} else {
-		actions.push('open-worktree', 'switch', 'switch-and-code-suggest', 'open-in-graph');
+		actions.push('open-worktree', 'switch');
+		if (supportsCodeSuggest(provider)) {
+			actions.push('switch-and-code-suggest');
+		}
+
+		actions.push('open-in-graph');
 	}
 	return actions;
 }
@@ -205,7 +224,9 @@ export class LaunchpadProvider implements Disposable {
 		if (prs?.value?.length && subscription?.account != null) {
 			try {
 				suggestionCounts = await withDurationAndSlowEventOnTimeout(
-					this.container.drafts.getCodeSuggestionCounts(prs.value.map(pr => pr.pullRequest)),
+					this.container.drafts.getCodeSuggestionCounts(
+						prs.value.map(pr => pr.pullRequest).filter(pr => supportsCodeSuggest(pr.provider)),
+					),
 					'getCodeSuggestionCounts',
 					this.container,
 				);
@@ -326,7 +347,7 @@ export class LaunchpadProvider implements Disposable {
 			this._codeSuggestions.get(item.uuid)!.expiresAt < Date.now()
 		) {
 			const providerId = item.provider.id;
-			if (!isSupportedLaunchpadIntegrationId(providerId)) {
+			if (!isSupportedLaunchpadIntegrationId(providerId) || !supportsCodeSuggest(item.provider)) {
 				return undefined;
 			}
 
@@ -790,6 +811,7 @@ export class LaunchpadProvider implements Disposable {
 
 					const suggestedActions = getSuggestedActions(
 						actionableCategory,
+						item.provider,
 						openRepository?.localBranch?.current ?? false,
 					);
 

src/webviews/apps/commitDetails/components/gl-wip-details.ts

@@ -4,6 +4,7 @@ import { css, html, nothing } from 'lit';
 import { customElement, property, state } from 'lit/decorators.js';
 import { repeat } from 'lit/directives/repeat.js';
 import { when } from 'lit/directives/when.js';
+import { equalsIgnoreCase } from '../../../../system/string';
 import type { DraftState, State, Wip } from '../../../commitDetails/protocol';
 import type { TreeItemAction, TreeItemBase } from '../../shared/components/tree/base';
 import type { File } from './gl-details-base';
@@ -128,7 +129,7 @@ export class GlWipDetails extends GlDetailsBase {
 		let label = 'Share as Cloud Patch';
 		let action = 'create-patch';
 		const pr = this.wip?.pullRequest;
-		if (pr != null && pr.state === 'opened') {
+		if (pr != null && pr.state === 'opened' && equalsIgnoreCase(pr.provider.domain, 'github.com')) {
 			// const isMe = pr.author.name.endsWith('(you)');
 			// if (isMe) {
 			// 	label = 'Share with PR Participants';

src/webviews/commitDetails/commitDetailsWebview.ts

@@ -47,6 +47,7 @@ import type { Subscription } from '../../plus/gk/models/subscription';
 import type { SubscriptionChangeEvent } from '../../plus/gk/subscriptionService';
 import { ensureAccount } from '../../plus/gk/utils/-webview/acount.utils';
 import type { ConnectionStateChangeEvent } from '../../plus/integrations/integrationService';
+import { supportsCodeSuggest } from '../../plus/integrations/providers/models';
 import { getEntityIdentifierInput } from '../../plus/integrations/providers/utils';
 import {
 	executeCommand,
@@ -1342,7 +1343,7 @@ export class CommitDetailsWebviewProvider
 		});
 
 		let codeSuggestions: Draft[] = [];
-		if (pullRequest != null) {
+		if (pullRequest != null && supportsCodeSuggest(pullRequest.provider)) {
 			const results = await this.getCodeSuggestions(pullRequest, repository);
 			if (results.length) {
 				codeSuggestions = results;
@@ -1363,7 +1364,7 @@ export class CommitDetailsWebviewProvider
 	}
 
 	private async getCodeSuggestions(pullRequest: PullRequest, repository: Repository): Promise<Draft[]> {
-		if (!(await this.canAccessDrafts())) return [];
+		if (!(await this.canAccessDrafts()) || !supportsCodeSuggest(pullRequest.provider)) return [];
 
 		const results = await this.container.drafts.getCodeSuggestions(pullRequest, repository);
 
@@ -1391,7 +1392,9 @@ export class CommitDetailsWebviewProvider
 		const wip = this._context.wip;
 		const { pullRequest, repo } = wip;
 
-		wip.codeSuggestions = await this.getCodeSuggestions(pullRequest!, repo);
+		wip.codeSuggestions = supportsCodeSuggest(pullRequest!.provider)
+			? await this.getCodeSuggestions(pullRequest!, repo)
+			: [];
 
 		if (this._pendingContext == null) {
 			const success = await this.host.notify(