introduce transient profiles

Author: sandy081

src/vs/code/electron-browser/sharedProcess/contrib/userDataProfilesCleaner.ts

@@ -0,0 +1,18 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { Disposable } from 'vs/base/common/lifecycle';
+import { IUserDataProfilesService } from 'vs/platform/userDataProfile/common/userDataProfile';
+
+export class UserDataProfilesCleaner extends Disposable {
+
+	constructor(
+		@IUserDataProfilesService userDataProfilesService: IUserDataProfilesService
+	) {
+		super();
+		userDataProfilesService.cleanUp();
+	}
+
+}

src/vs/code/electron-browser/sharedProcess/sharedProcessMain.ts

@@ -105,6 +105,7 @@ import { IPolicyService, NullPolicyService } from 'vs/platform/policy/common/pol
 import { UserDataProfilesNativeService } from 'vs/platform/userDataProfile/electron-sandbox/userDataProfile';
 import { SharedProcessRequestService } from 'vs/platform/request/electron-browser/sharedProcessRequestService';
 import { OneDataSystemAppender } from 'vs/platform/telemetry/node/1dsAppender';
+import { UserDataProfilesCleaner } from 'vs/code/electron-browser/sharedProcess/contrib/userDataProfilesCleaner';
 
 class SharedProcessMain extends Disposable {
 
@@ -169,7 +170,8 @@ class SharedProcessMain extends Disposable {
 			instantiationService.createInstance(StorageDataCleaner, this.configuration.backupWorkspacesPath),
 			instantiationService.createInstance(LogsDataCleaner),
 			instantiationService.createInstance(LocalizationsUpdater),
-			instantiationService.createInstance(ExtensionsCleaner)
+			instantiationService.createInstance(ExtensionsCleaner),
+			instantiationService.createInstance(UserDataProfilesCleaner)
 		));
 	}
 

src/vs/code/electron-main/app.ts

@@ -108,6 +108,7 @@ import { IRequestService } from 'vs/platform/request/common/request';
 import { ExtensionsProfileScannerService, IExtensionsProfileScannerService } from 'vs/platform/extensionManagement/common/extensionsProfileScannerService';
 import { IExtensionsScannerService } from 'vs/platform/extensionManagement/common/extensionsScannerService';
 import { ExtensionsScannerService } from 'vs/platform/extensionManagement/node/extensionsScannerService';
+import { UserDataTransientProfilesHandler } from 'vs/platform/userDataProfile/electron-main/userDataTransientProfilesHandler';
 
 /**
  * The main VS Code application. There will only ever be one instance,
@@ -562,6 +563,9 @@ export class CodeApplication extends Disposable {
 
 		// Default Extensions Profile Init Handler
 		this._register(instantiationService.createInstance(DefaultExtensionsProfileInitHandler));
+
+		// Transient profiles handler
+		this._register(instantiationService.createInstance(UserDataTransientProfilesHandler));
 	}
 
 	private async resolveMachineId(): Promise<string> {

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

@@ -92,6 +92,7 @@ export interface NativeParsedArgs {
 	editSessionId?: string;
 	'locate-shell-integration-path'?: string;
 	'profile'?: string;
+	'transient'?: boolean;
 
 	// chromium command line args: https://electronjs.org/docs/all#supported-chrome-command-line-switches
 	'no-proxy-server'?: boolean;

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

@@ -50,7 +50,7 @@ export const OPTIONS: OptionDescriptions<Required<NativeParsedArgs>> = {
 	'waitMarkerFilePath': { type: 'string' },
 	'locale': { type: 'string', cat: 'o', args: 'locale', description: localize('locale', "The locale to use (e.g. en-US or zh-TW).") },
 	'user-data-dir': { type: 'string', cat: 'o', args: 'dir', description: localize('userDataDir', "Specifies the directory that user data is kept in. Can be used to open multiple distinct instances of Code.") },
-	'profile': { type: 'string', 'cat': 'o', args: 'settingsProfileName', description: localize('settingsProfileName', "Opens the provided folder or workspace with the given profile. If the profile does not exist, a new empty one is created.") },
+	'profile': { type: 'string', 'cat': 'o', args: 'settingsProfileName', description: localize('settingsProfileName', "Opens the provided folder or workspace with the given profile. If the profile does not exist, a new empty one is created. Use '--transient' argument to not to persist the profile.") },
 	'help': { type: 'boolean', cat: 'o', alias: 'h', description: localize('help', "Print usage.") },
 
 	'extensions-dir': { type: 'string', deprecates: ['extensionHomePath'], cat: 'e', args: 'dir', description: localize('extensionHomePath', "Set the root path for extensions.") },
@@ -130,6 +130,7 @@ export const OPTIONS: OptionDescriptions<Required<NativeParsedArgs>> = {
 	'__enable-file-policy': { type: 'boolean' },
 	'editSessionId': { type: 'string' },
 	'locate-shell-integration-path': { type: 'string', args: ['bash', 'pwsh', 'zsh', 'fish'] },
+	'transient': { type: 'boolean' },
 
 	// chromium flags
 	'no-proxy-server': { type: 'boolean' },

src/vs/platform/userDataProfile/browser/userDataProfile.ts

@@ -31,16 +31,42 @@ export class BrowserUserDataProfilesService extends UserDataProfilesService impl
 		this._register(this.changesBroadcastChannel.onDidReceiveData(changes => {
 			try {
 				this._profilesObject = undefined;
+				const added = changes.added.map(p => reviveProfile(p, this.profilesHome.scheme));
+				const removed = changes.removed.map(p => reviveProfile(p, this.profilesHome.scheme));
+				const updated = changes.updated.map(p => reviveProfile(p, this.profilesHome.scheme));
+
+				this.updateTransientProfiles(
+					added.filter(a => a.isTransient),
+					removed.filter(a => a.isTransient),
+					updated.filter(a => a.isTransient)
+				);
+
 				this._onDidChangeProfiles.fire({
-					added: changes.added.map(p => reviveProfile(p, this.profilesHome.scheme)),
-					removed: changes.removed.map(p => reviveProfile(p, this.profilesHome.scheme)),
-					updated: changes.updated.map(p => reviveProfile(p, this.profilesHome.scheme)),
+					added,
+					removed,
+					updated,
 					all: this.profiles
 				});
 			} catch (error) {/* ignore */ }
 		}));
 	}
 
+	private updateTransientProfiles(added: IUserDataProfile[], removed: IUserDataProfile[], updated: IUserDataProfile[]): void {
+		if (added.length) {
+			this.transientProfilesObject.profiles.push(...added);
+		}
+		if (removed.length || updated.length) {
+			const allTransientProfiles = this.transientProfilesObject.profiles;
+			this.transientProfilesObject.profiles = [];
+			for (const profile of allTransientProfiles) {
+				if (removed.some(p => profile.id === p.id)) {
+					continue;
+				}
+				this.transientProfilesObject.profiles.push(updated.find(p => profile.id === p.id) ?? profile);
+			}
+		}
+	}
+
 	override setEnablement(enabled: boolean): void {
 		super.setEnablement(enabled);
 		window.localStorage.setItem(PROFILES_ENABLEMENT_CONFIG, enabled ? 'true' : 'false');