Makes AI usage in create PR optional, adds button that creates PR using AI

Author: sergeibbb

src/api/gitlens.d.ts

@@ -1,4 +1,5 @@
 import type { Disposable } from 'vscode';
+import type { Sources } from '../constants.telemetry';
 
 export type { Disposable } from 'vscode';
 
@@ -24,6 +25,8 @@ export interface CreatePullRequestActionContext {
 				readonly url?: string;
 		  }
 		| undefined;
+	readonly source?: Sources;
+	readonly useAI?: boolean;
 }
 
 export interface OpenPullRequestActionContext {

src/commands/createPullRequestOnRemote.ts

@@ -1,4 +1,5 @@
 import { window } from 'vscode';
+import type { Sources } from '../constants.telemetry';
 import type { Container } from '../container';
 import type { GitRemote } from '../git/models/remote';
 import type { RemoteResource } from '../git/models/remoteResource';
@@ -17,6 +18,8 @@ export interface CreatePullRequestOnRemoteCommandArgs {
 	repoPath: string;
 
 	clipboard?: boolean;
+	source?: Sources;
+	useAI?: boolean;
 }
 
 @command()
@@ -74,26 +77,31 @@ export class CreatePullRequestOnRemoteCommand extends GlCommandBase {
 				branch: args.compare,
 				remote: { path: compareRemote.path, url: compareRemote.url, name: compareRemote.name },
 			},
-			describePullRequest: async (
-				completedResource: RemoteResource & { type: RemoteResourceType.CreatePullRequest },
-			) => {
-				const base = completedResource.base;
-				const compare = completedResource.compare;
-				if (!base?.remote || !compare?.remote || !base?.branch || !compare?.branch) {
-					return undefined;
-				}
-				const baseRef = `${base.remote.name}/${base.branch}`;
-				const compareRef = `${compare.remote.name}/${compare.branch}`;
-				try {
-					const result = await this.container.ai.generatePullRequestMessage(repo, baseRef, compareRef, {
-						source: 'scm-input',
-					});
-					return result?.parsed;
-				} catch (e) {
-					void window.showErrorMessage(`Unable to generate pull request message: ${e}`);
-					return undefined;
-				}
-			},
+			describePullRequest: !args.useAI
+				? undefined
+				: async (completedResource: RemoteResource & { type: RemoteResourceType.CreatePullRequest }) => {
+						const base = completedResource.base;
+						const compare = completedResource.compare;
+						if (!base?.remote || !compare?.remote || !base?.branch || !compare?.branch) {
+							return undefined;
+						}
+						const baseRef = `${base.remote.name}/${base.branch}`;
+						const compareRef = `${compare.remote.name}/${compare.branch}`;
+						try {
+							const result = await this.container.ai.generatePullRequestMessage(
+								repo,
+								baseRef,
+								compareRef,
+								{
+									source: args.source ?? 'scm-input',
+								},
+							);
+							return result?.parsed;
+						} catch (e) {
+							void window.showErrorMessage(`Unable to generate pull request message: ${e}`);
+							return undefined;
+						}
+				  },
 		};
 
 		void (await executeCommand<OpenOnRemoteCommandArgs>('gitlens.openOnRemote', {

src/extension.ts

@@ -309,6 +309,8 @@ function registerBuiltInActionRunners(container: Container): void {
 						  : ctx.branch.name,
 					remote: ctx.remote?.name ?? '',
 					repoPath: ctx.repoPath,
+					source: ctx.source,
+					useAI: ctx.useAI,
 				}));
 			},
 		}),

src/webviews/apps/plus/home/components/branch-card.ts

@@ -640,7 +640,7 @@ export abstract class GlBranchCardBase extends GlElement {
 	}
 
 	protected createCommandLink<T>(command: GlCommands, args?: T | any): string {
-		return createCommandLink<T>(command, args ?? this.branchRef);
+		return createCommandLink<T>(command, args ? { ...args, ...this.branchRef } : this.branchRef);
 	}
 
 	protected renderTimestamp(): TemplateResult | NothingType {
@@ -722,6 +722,15 @@ export abstract class GlBranchCardBase extends GlElement {
 						href="${this.createCommandLink('gitlens.home.createPullRequest')}"
 						>Create a Pull Request</gl-button
 					>
+						<gl-button
+							class="branch-item__missing"
+							appearance="secondary"
+							href="${this.createCommandLink('gitlens.home.createPullRequest', {
+								source: 'home',
+								useAI: true,
+							})}"
+							>AI</gl-button
+						>
 				`;
 			}
 			return nothing;

src/webviews/home/homeWebview.ts

@@ -12,7 +12,7 @@ import {
 	supportedCloudIntegrationDescriptors,
 	supportedOrderedCloudIntegrationIds,
 } from '../../constants.integrations';
-import type { HomeTelemetryContext, Source } from '../../constants.telemetry';
+import type { HomeTelemetryContext, Source, Sources } from '../../constants.telemetry';
 import type { Container } from '../../container';
 import { executeGitCommand } from '../../git/actions';
 import { openComparisonChanges } from '../../git/actions/commit';
@@ -1346,7 +1346,7 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 	@log<HomeWebviewProvider['pullRequestCreate']>({
 		args: { 0: r => `${r.branchId}, upstream: ${r.branchUpstreamName}` },
 	})
-	private async pullRequestCreate(ref: BranchRef) {
+	private async pullRequestCreate(ref: BranchRef & { source?: Sources; useAI?: boolean }) {
 		const { branch } = await this.getRepoInfoFromRef(ref);
 		if (branch == null) return;
 
@@ -1374,6 +1374,8 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 				upstream: branch.upstream?.name,
 				isRemote: branch.remote,
 			},
+			source: ref.source,
+			useAI: ref.useAI,
 		});
 	}