Associate a Cloud GitHub Enterprise provider with remote by its domain

(#3901)

Author: sergeibbb

src/env/node/git/localGitProvider.ts

@@ -5039,8 +5039,12 @@ export class LocalGitProvider implements GitProvider, Disposable {
 		let remotesPromise = this.useCaching ? this._remotesCache.get(repoPath) : undefined;
 		if (remotesPromise == null) {
 			async function load(this: LocalGitProvider): Promise<GitRemote[]> {
+				const ci = await this.container.cloudIntegrations;
+				const connections = await ci?.getConnections();
+
 				const providers = loadRemoteProviders(
 					configuration.get('remotes', this.container.git.getRepository(repoPath!)?.folder?.uri ?? null),
+					connections,
 				);
 
 				try {
@@ -5049,7 +5053,7 @@ export class LocalGitProvider implements GitProvider, Disposable {
 						this.container,
 						data,
 						repoPath!,
-						getRemoteProviderMatcher(this.container, providers),
+						await getRemoteProviderMatcher(this.container, providers),
 					);
 					return remotes;
 				} catch (ex) {

src/git/parsers/remoteParser.ts

@@ -6,11 +6,13 @@ import type { getRemoteProviderMatcher } from '../remotes/remoteProviders';
 
 const remoteRegex = /^(.*)\t(.*)\s\((.*)\)$/gm;
 
+type PromiseType<T> = T extends Promise<infer U> ? U : T;
+
 export function parseGitRemotes(
 	container: Container,
 	data: string,
 	repoPath: string,
-	remoteProviderMatcher: ReturnType<typeof getRemoteProviderMatcher>,
+	remoteProviderMatcher: PromiseType<ReturnType<typeof getRemoteProviderMatcher>>,
 ): GitRemote[] {
 	using sw = maybeStopWatch(`Git.parseRemotes(${repoPath})`, { log: false, logLevel: 'debug' });
 	if (!data) return [];

src/git/remotes/remoteProviders.ts

@@ -1,7 +1,11 @@
 import type { RemotesConfig } from '../../config';
+import { SelfHostedIntegrationId } from '../../constants.integrations';
 import type { Container } from '../../container';
+import type { CloudIntegrationConnection } from '../../plus/integrations/authentication/models';
+import { toIntegrationId } from '../../plus/integrations/authentication/models';
 import { Logger } from '../../system/logger';
 import { configuration } from '../../system/vscode/configuration';
+import { parseGitRemoteUrl } from '../parsers/remoteParser';
 import { AzureDevOpsRemote } from './azure-devops';
 import { BitbucketRemote } from './bitbucket';
 import { BitbucketServerRemote } from './bitbucket-server';
@@ -73,7 +77,10 @@ const builtInProviders: RemoteProviders = [
 	},
 ];
 
-export function loadRemoteProviders(cfg: RemotesConfig[] | null | undefined): RemoteProviders {
+export function loadRemoteProviders(
+	cfg: RemotesConfig[] | null | undefined,
+	connectedIntegrations: CloudIntegrationConnection[] | undefined,
+): RemoteProviders {
 	const providers: RemoteProviders = [];
 
 	if (cfg?.length) {
@@ -97,6 +104,37 @@ export function loadRemoteProviders(cfg: RemotesConfig[] | null | undefined): Re
 		}
 	}
 
+	if (connectedIntegrations?.length) {
+		for (const ci of connectedIntegrations) {
+			if (toIntegrationId[ci.provider] === SelfHostedIntegrationId.CloudGitHubEnterprise) {
+				const host = new URL(ci.domain).host;
+				const rc: RemotesConfig = {
+					domain: host,
+					regex: null,
+					name: ci.domain,
+					protocol: 'https',
+					type: 'GitHub',
+				};
+				const providerCreator = getCustomProviderCreator(rc);
+				if (providerCreator == null) continue;
+
+				let matcher: string | RegExp | undefined;
+				try {
+					matcher = rc.domain?.toLowerCase();
+					if (matcher == null) throw new Error('No matcher found');
+				} catch (ex) {
+					Logger.error(ex, `Loading remote provider '${rc.name ?? ''}' failed`);
+				}
+
+				providers.push({
+					custom: true,
+					matcher: matcher!,
+					creator: providerCreator,
+				});
+			}
+		}
+	}
+
 	providers.push(...builtInProviders);
 
 	return providers;
@@ -136,12 +174,14 @@ function getCustomProviderCreator(cfg: RemotesConfig) {
 	}
 }
 
-export function getRemoteProviderMatcher(
+export async function getRemoteProviderMatcher(
 	container: Container,
 	providers?: RemoteProviders,
-): (url: string, domain: string, path: string) => RemoteProvider | undefined {
+): Promise<(url: string, domain: string, path: string) => RemoteProvider | undefined> {
 	if (providers == null) {
-		providers = loadRemoteProviders(configuration.get('remotes', null));
+		const ci = await container.cloudIntegrations;
+		const c = await ci?.getConnections();
+		providers = loadRemoteProviders(configuration.get('remotes', null), c);
 	}
 
 	return (url: string, domain: string, path: string) =>

src/plus/drafts/draftsService.ts

@@ -729,7 +729,7 @@ export class DraftService implements Disposable {
 		} else if (data.provider?.repoName != null) {
 			name = data.provider.repoName;
 		} else if (data.remote?.url != null && data.remote?.domain != null && data.remote?.path != null) {
-			const matcher = getRemoteProviderMatcher(this.container);
+			const matcher = await getRemoteProviderMatcher(this.container);
 			const provider = matcher(data.remote.url, data.remote.domain, data.remote.path);
 			name = provider?.repoName ?? data.remote.path;
 		} else {

src/plus/integrations/providers/github/githubGitProvider.ts

@@ -2781,7 +2781,7 @@ export class GitHubGitProvider implements GitProvider, Disposable {
 	): Promise<GitRemote[]> {
 		if (repoPath == null) return [];
 
-		const providers = loadRemoteProviders(configuration.get('remotes', null));
+		const providers = loadRemoteProviders(configuration.get('remotes', null), undefined);
 
 		const uri = Uri.parse(repoPath, true);
 		const [, owner, repo] = uri.path.split('/', 3);
@@ -2798,7 +2798,7 @@ export class GitHubGitProvider implements GitProvider, Disposable {
 				'https',
 				domain,
 				path,
-				getRemoteProviderMatcher(this.container, providers)(url, domain, path),
+				(await getRemoteProviderMatcher(this.container, providers))(url, domain, path),
 				[
 					{ type: 'fetch', url: url },
 					{ type: 'push', url: url },