From d0bd8225b3b4aa1b36f244fb58f1f86aa92caa94 Mon Sep 17 00:00:00 2001
From: Eric Amodio <eamodio@microsoft.com>
Date: Thu, 25 Jun 2020 13:22:06 -0400
Subject: [PATCH] Adds cs querystring to yarn web for editing

---
 resources/serverless/code-web.js | 28 ++++++++++++++++++++--------
 1 file changed, 20 insertions(+), 8 deletions(-)

diff --git a/resources/serverless/code-web.js b/resources/serverless/code-web.js
index b62185f107b..4d0ad61e7c1 100644
--- a/resources/serverless/code-web.js
+++ b/resources/serverless/code-web.js
@@ -100,7 +100,7 @@ async function initialize() {
 						mainFilePath += '.js';
 					}
 					if (!await exists(mainFilePath)) {
-						unbuiltExensions.push(path.relative(EXTENSIONS_ROOT, mainFilePath))
+						unbuiltExensions.push(path.relative(EXTENSIONS_ROOT, mainFilePath));
 					}
 				}
 				packageJSON.extensionKind = ['web']; // enable for Web
@@ -221,22 +221,34 @@ function handleStaticExtension(req, res, parsedUrl) {
  * @param {import('http').ServerResponse} res
  */
 async function handleRoot(req, res) {
+	let folderUri = { scheme: 'memfs', path: `/sample-folder` };
+
 	const match = req.url && req.url.match(/\?([^#]+)/);
-	let ghPath;
 	if (match) {
 		const qs = new URLSearchParams(match[1]);
-		ghPath = qs.get('gh');
-		if (ghPath && !ghPath.startsWith('/')) {
-			ghPath = '/' + ghPath;
+
+		let ghPath = qs.get('gh');
+		if (ghPath) {
+			if (!ghPath.startsWith('/')) {
+				ghPath = '/' + ghPath;
+			}
+			folderUri = { scheme: 'github', authority: 'HEAD', path: ghPath };
+		} else {
+
+			let csPath = qs.get('cs');
+			if (csPath) {
+				if (!csPath.startsWith('/')) {
+					csPath = '/' + csPath;
+				}
+				folderUri = { scheme: 'codespace', authority: 'HEAD', path: csPath };
+			}
 		}
 	}
 
 	const builtinExtensions = await builtinExtensionsPromise;
 
 	const webConfigJSON = escapeAttribute(JSON.stringify({
-		folderUri: ghPath
-			? { scheme: 'github', authority: 'HEAD', path: ghPath }
-			: { scheme: 'memfs', path: `/sample-folder` },
+		folderUri: folderUri,
 		builtinExtensionsServiceUrl: `${SCHEME}://${AUTHORITY}/static-extension`
 	}));