Fixes integration connection sources not making it through

Author: axosoft-ramint

src/commands/patches.ts

@@ -344,7 +344,7 @@ export class OpenCloudPatchCommand extends Command {
 				return;
 			}
 
-			const session = await integration.getSession();
+			const session = await integration.getSession('cloud-patches');
 			if (session == null) {
 				void window.showErrorMessage(`Cannot open ${type}; provider not connected.`);
 				return;

src/plus/drafts/draftsService.ts

@@ -766,7 +766,7 @@ export class DraftService implements Disposable {
 		}
 		if (integration == null) return undefined;
 
-		const session = await integration.getSession();
+		const session = await integration.getSession('code-suggest');
 		if (session == null) return undefined;
 
 		return {

src/plus/integrations/integration.ts

@@ -129,15 +129,15 @@ export abstract class IntegrationBase<
 	}
 
 	protected _session: ProviderAuthenticationSession | null | undefined;
-	getSession() {
+	getSession(source: Sources) {
 		if (this._session === undefined) {
-			return this.ensureSession({ createIfNeeded: false });
+			return this.ensureSession({ createIfNeeded: false, source: source });
 		}
 		return this._session ?? undefined;
 	}
 
 	@log()
-	async connect(source?: Sources): Promise<boolean> {
+	async connect(source: Sources): Promise<boolean> {
 		try {
 			return Boolean(await this.ensureSession({ createIfNeeded: true, source: source }));
 		} catch (ex) {
@@ -290,7 +290,7 @@ export abstract class IntegrationBase<
 	@gate()
 	@debug({ exit: true })
 	async isConnected(): Promise<boolean> {
-		return (await this.getSession()) != null;
+		return (await this.getSession('integrations')) != null;
 	}
 
 	@gate()

src/plus/integrations/providers/github.ts

@@ -1,4 +1,5 @@
 import type { AuthenticationSession, CancellationToken } from 'vscode';
+import type { Sources } from '../../../constants';
 import type { Container } from '../../../container';
 import type { Account, UnidentifiedAuthor } from '../../../git/models/author';
 import type { DefaultBranch } from '../../../git/models/defaultBranch';
@@ -302,7 +303,7 @@ export class GitHubEnterpriseIntegration extends GitHubIntegrationBase<SelfHoste
 	}
 
 	@log()
-	override async connect(): Promise<boolean> {
+	override async connect(source: Sources): Promise<boolean> {
 		if (
 			!(await ensurePaidPlan(this.container, `Rich integration with ${this.name} is a Pro feature.`, {
 				source: 'integrations',
@@ -312,6 +313,6 @@ export class GitHubEnterpriseIntegration extends GitHubIntegrationBase<SelfHoste
 			return false;
 		}
 
-		return super.connect();
+		return super.connect(source);
 	}
 }

src/plus/integrations/providers/gitlab.ts

@@ -1,5 +1,6 @@
 import type { AuthenticationSession, CancellationToken } from 'vscode';
 import { window } from 'vscode';
+import type { Sources } from '../../../constants';
 import type { Container } from '../../../container';
 import type { Account } from '../../../git/models/author';
 import type { DefaultBranch } from '../../../git/models/defaultBranch';
@@ -349,7 +350,7 @@ export class GitLabSelfHostedIntegration extends GitLabIntegrationBase<SelfHoste
 	}
 
 	@log()
-	override async connect(): Promise<boolean> {
+	override async connect(source: Sources): Promise<boolean> {
 		if (
 			!(await ensurePaidPlan(this.container, `Rich integration with ${this.name} is a Pro feature.`, {
 				source: 'integrations',
@@ -359,6 +360,6 @@ export class GitLabSelfHostedIntegration extends GitLabIntegrationBase<SelfHoste
 			return false;
 		}
 
-		return super.connect();
+		return super.connect(source);
 	}
 }