add --download-extensions-locally

Author: joshspicer

cli/src/commands/args.rs

@@ -266,6 +266,7 @@ pub enum ExtensionSubcommand {
 	Uninstall(UninstallExtensionArgs),
 	/// Update the installed extensions.
 	Update,
+	DownloadExtensionsLocally,
 }
 
 impl ExtensionSubcommand {
@@ -302,6 +303,9 @@ impl ExtensionSubcommand {
 			ExtensionSubcommand::Update => {
 				target.push("--update-extensions".to_string());
 			}
+			ExtensionSubcommand::DownloadExtensionsLocally => {
+				target.push("--download-extensions-locally".to_string());
+			}
 		}
 	}
 }

cli/src/tunnels/code_server.rs

@@ -63,6 +63,7 @@ pub struct CodeServerArgs {
 	pub install_extensions: Vec<String>,
 	pub uninstall_extensions: Vec<String>,
 	pub update_extensions: bool,
+	pub download_extensions_locally: bool,
 	pub list_extensions: bool,
 	pub show_versions: bool,
 	pub category: Option<String>,
@@ -138,6 +139,9 @@ impl CodeServerArgs {
 		if self.update_extensions {
 			args.push(String::from("--update-extensions"));
 		}
+		if self.download_extensions_locally {
+			args.push(String::from("--download-extensions-locally"));
+		}
 		if self.list_extensions {
 			args.push(String::from("--list-extensions"));
 			if self.show_versions {

src/server-main.ts

@@ -28,7 +28,7 @@ perf.mark('code/server/start');
 // Do a quick parse to determine if a server or the cli needs to be started
 const parsedArgs = minimist(process.argv.slice(2), {
 	boolean: ['start-server', 'list-extensions', 'print-ip-address', 'help', 'version', 'accept-server-license-terms', 'update-extensions'],
-	string: ['install-extension', 'install-builtin-extension', 'uninstall-extension', 'locate-extension', 'socket-path', 'host', 'port', 'compatibility'],
+	string: ['install-extension', 'install-builtin-extension', 'uninstall-extension', 'locate-extension', 'socket-path', 'host', 'port', 'compatibility', 'download-extensions-locally'],
 	alias: { help: 'h', version: 'v' }
 });
 ['host', 'port', 'accept-server-license-terms'].forEach(e => {
@@ -41,7 +41,7 @@ const parsedArgs = minimist(process.argv.slice(2), {
 });
 
 const extensionLookupArgs = ['list-extensions', 'locate-extension'];
-const extensionInstallArgs = ['install-extension', 'install-builtin-extension', 'uninstall-extension', 'update-extensions'];
+const extensionInstallArgs = ['install-extension', 'install-builtin-extension', 'uninstall-extension', 'update-extensions', 'download-extensions-locally'];
 
 const shouldSpawnCli = parsedArgs.help || parsedArgs.version || extensionLookupArgs.some(a => !!parsedArgs[a]) || (extensionInstallArgs.some(a => !!parsedArgs[a]) && !parsedArgs['start-server']);
 

src/vs/platform/environment/common/argv.ts

@@ -83,6 +83,7 @@ export interface NativeParsedArgs {
 	'pre-release'?: boolean;
 	'install-builtin-extension'?: string[]; // undefined or array of 1 or more
 	'uninstall-extension'?: string[]; // undefined or array of 1 or more
+	'download-extensions-locally'?: boolean;
 	'update-extensions'?: boolean;
 	'do-not-include-pack-dependencies'?: boolean;
 	'locate-extension'?: string[]; // undefined or array of 1 or more

src/vs/platform/environment/node/argv.ts

@@ -101,6 +101,7 @@ export const OPTIONS: OptionDescriptions<Required<NativeParsedArgs>> = {
 	'install-extension': { type: 'string[]', cat: 'e', args: 'ext-id | path', description: localize('installExtension', "Installs or updates an extension. The argument is either an extension id or a path to a VSIX. The identifier of an extension is '${publisher}.${name}'. Use '--force' argument to update to latest version. To install a specific version provide '@${version}'. For example: 'vscode.csharp@1.2.3'.") },
 	'pre-release': { type: 'boolean', cat: 'e', description: localize('install prerelease', "Installs the pre-release version of the extension, when using --install-extension") },
 	'uninstall-extension': { type: 'string[]', cat: 'e', args: 'ext-id', description: localize('uninstallExtension', "Uninstalls an extension.") },
+	'download-extensions-locally': { type: 'boolean', cat: 'e', description: localize('downloadExtensionsLocally', "When enabled extensions are downloaded locally and installed on remote") },
 	'update-extensions': { type: 'boolean', cat: 'e', description: localize('updateExtensions', "Update the installed extensions.") },
 	'enable-proposed-api': { type: 'string[]', allowEmptyValue: true, cat: 'e', args: 'ext-id', description: localize('experimentalApis', "Enables proposed API features for extensions. Can receive one or more extension IDs to enable individually.") },
 

src/vs/server/node/remoteExtensionsScanner.ts

@@ -60,12 +60,22 @@ export class RemoteExtensionsScannerService implements IRemoteExtensionsScannerS
 
 		const extensionsToInstall = environmentService.args['install-extension'];
 		if (extensionsToInstall) {
-			_logService.trace('Installing extensions passed via args...');
 			const installOptions: InstallOptions = {
 				isMachineScoped: !!environmentService.args['do-not-sync'],
 				installPreReleaseVersion: !!environmentService.args['pre-release'],
 				isApplicationScoped: true // extensions installed during server startup are available to all profiles
 			};
+
+			if (!!environmentService.args['download-extensions-locally']) {
+				_logService.trace('Installing extensions later...');
+				this._whenExtensionsReady = this._whenBuiltinExtensionsReady
+					.then(async () => {
+						return { failed: extensionsToInstall.map(id => ({ id, installOptions })) };
+					});
+				return;
+			}
+
+			_logService.trace('Installing extensions passed via args...');
 			this._whenExtensionsReady = this._whenBuiltinExtensionsReady
 				.then(() => _extensionManagementCLI.installExtensions(this._asExtensionIdOrVSIX(extensionsToInstall), [], installOptions, !!environmentService.args['force']))
 				.then(async () => {

src/vs/server/node/server.cli.ts

@@ -70,6 +70,7 @@ const isSupportedForPipe = (optionId: keyof RemoteParsedArgs) => {
 		case 'uninstall-extension':
 		case 'update-extensions':
 		case 'list-extensions':
+		case 'download-extensions-locally':
 		case 'force':
 		case 'do-not-include-pack-dependencies':
 		case 'show-versions':
@@ -228,7 +229,7 @@ export async function main(desc: ProductDescription, args: string[]): Promise<vo
 	}
 
 	if (cliCommand) {
-		if (parsedArgs['install-extension'] !== undefined || parsedArgs['uninstall-extension'] !== undefined || parsedArgs['list-extensions'] || parsedArgs['update-extensions']) {
+		if (parsedArgs['install-extension'] !== undefined || parsedArgs['uninstall-extension'] !== undefined || parsedArgs['list-extensions'] || parsedArgs['update-extensions'] || parsedArgs['download-extensions-locally']) {
 			const cmdLine: string[] = [];
 			parsedArgs['install-extension']?.forEach(id => cmdLine.push('--install-extension', id));
 			parsedArgs['uninstall-extension']?.forEach(id => cmdLine.push('--uninstall-extension', id));
@@ -242,6 +243,10 @@ export async function main(desc: ProductDescription, args: string[]): Promise<vo
 				cmdLine.push('--update-extensions');
 			}
 
+			if (parsedArgs['download-extensions-locally']) {
+				cmdLine.push('--download-extensions-locally');
+			}
+
 			const childProcess = cp.fork(FileAccess.asFileUri('server-main').fsPath, cmdLine, { stdio: 'inherit' });
 			childProcess.on('error', err => console.log(err));
 			return;

src/vs/server/node/serverEnvironmentService.ts

@@ -64,7 +64,7 @@ export const serverOptions: OptionDescriptions<Required<ServerParsedArgs>> = {
 	'uninstall-extension': OPTIONS['uninstall-extension'],
 	'list-extensions': OPTIONS['list-extensions'],
 	'locate-extension': OPTIONS['locate-extension'],
-
+	'download-extensions-locally': OPTIONS['download-extensions-locally'],
 	'show-versions': OPTIONS['show-versions'],
 	'category': OPTIONS['category'],
 	'force': OPTIONS['force'],
@@ -189,6 +189,7 @@ export interface ServerParsedArgs {
 	'update-extensions'?: boolean;
 	'uninstall-extension'?: string[];
 	'list-extensions'?: boolean;
+	'download-extensions-locally'?: boolean;
 	'locate-extension'?: string[];
 	'show-versions'?: boolean;
 	'category'?: string;