Adds support for fork on push for Codespaces

extensions/github/src/pushErrorHandler.ts

@@ -3,35 +3,91 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
-import { PushErrorHandler, GitErrorCodes, Repository, Remote } from './typings/git';
-import { window, ProgressLocation, commands, Uri } from 'vscode';
+import { commands, env, ProgressLocation, UIKind, Uri, window } from 'vscode';
 import * as nls from 'vscode-nls';
 import { getOctokit } from './auth';
+import { GitErrorCodes, PushErrorHandler, Remote, Repository } from './typings/git';
 
 const localize = nls.loadMessageBundle();
 
+type Awaited<T> = T extends PromiseLike<infer U> ? Awaited<U> : T;
+
+export function isInCodespaces(): boolean {
+	return env.remoteName === 'codespaces';
+}
+
 async function handlePushError(repository: Repository, remote: Remote, refspec: string, owner: string, repo: string): Promise<void> {
+	const inCodespaces = isInCodespaces();
+	let codespace: string | undefined;
+	if (inCodespaces) {
+		if (env.uiKind === UIKind.Web) {
+			// TODO@eamodio Find a better way to get the codespace id
+			// HACK to get the codespace id
+			try {
+				const codespaceUrl = (await env.asExternalUri(Uri.parse(`${env.uriScheme}://codespace/`))).authority;
+				if (codespaceUrl.endsWith('.github.dev')) {
+					codespace = codespaceUrl.slice(0, -11);
+				} else {
+					[codespace] = codespaceUrl.split('.');
+				}
+			} catch { }
+		} else {
+			// TODO@eamodio Figure out how to get the codespace id when on the desktop
+		}
+
+		if (!codespace) {
+			const ok = localize('ok', "OK");
+			await window.showErrorMessage(localize('fork unable', "You don't have permissions to push to '{0}/{1}' on GitHub.", owner, repo), ok);
+			return;
+		}
+	}
+
 	const yes = localize('create a fork', "Create Fork");
 	const no = localize('no', "No");
 
 	const answer = await window.showInformationMessage(localize('fork', "You don't have permissions to push to '{0}/{1}' on GitHub. Would you like to create a fork and push to it instead?", owner, repo), yes, no);
-
 	if (answer === no) {
 		return;
 	}
 
 	const match = /^([^:]*):([^:]*)$/.exec(refspec);
 	const localName = match ? match[1] : refspec;
-	const remoteName = match ? match[2] : refspec;
+	let remoteName = match ? match[2] : refspec;
 
 	const [octokit, ghRepository] = await window.withProgress({ location: ProgressLocation.Notification, cancellable: false, title: localize('create fork', 'Create GitHub fork') }, async progress => {
 		progress.report({ message: localize('forking', "Forking '{0}/{1}'...", owner, repo), increment: 33 });
 
 		const octokit = await getOctokit();
 
+		type CreateForkResponseData = Awaited<ReturnType<typeof octokit.repos.createFork>>['data'];
+
 		// Issue: what if the repo already exists?
-		const res = await octokit.repos.createFork({ owner, repo });
-		const ghRepository = res.data;
+		let ghRepository: CreateForkResponseData;
+		try {
+			if (inCodespaces) {
+				const userResp = await octokit.users.getAuthenticated();
+				const user = userResp.data.login;
+
+				const resp = await octokit.request<{ repository: CreateForkResponseData, ref: string }>({ method: 'POST', url: `/vscs_internal/user/${user}/codespaces/${codespace}/fork_repo` });
+				ghRepository = resp.data.repository;
+
+				if (resp.data.ref) {
+					let ref = resp.data.ref;
+					if (ref.startsWith('refs/heads/')) {
+						ref = ref.substr(11);
+					}
+
+					remoteName = ref;
+				}
+			} else {
+				const resp = await octokit.repos.createFork({ owner, repo });
+				ghRepository = resp.data;
+			}
+		} catch (ex) {
+			console.error(ex);
+			throw ex;
+		}
+
 
 		progress.report({ message: localize('forking_pushing', "Pushing changes..."), increment: 33 });