1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
/*---------------------------------------------------------------------------------------------
* 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 { localize } from 'vs/nls';
import { IDialogService } from 'vs/platform/dialogs/common/dialogs';
import { InstantiationType, registerSingleton } from 'vs/platform/instantiation/common/extensions';
import { DidChangeProfilesEvent, IUserDataProfile, IUserDataProfileOptions, IUserDataProfilesService, IUserDataProfileUpdateOptions, WorkspaceIdentifier } from 'vs/platform/userDataProfile/common/userDataProfile';
import { IWorkspaceContextService, WorkbenchState } from 'vs/platform/workspace/common/workspace';
import { IWorkbenchEnvironmentService } from 'vs/workbench/services/environment/common/environmentService';
import { IExtensionService } from 'vs/workbench/services/extensions/common/extensions';
import { IHostService } from 'vs/workbench/services/host/browser/host';
import { DidChangeUserDataProfileEvent, IUserDataProfileManagementService, IUserDataProfileService } from 'vs/workbench/services/userDataProfile/common/userDataProfile';
export class UserDataProfileManagementService extends Disposable implements IUserDataProfileManagementService {
readonly _serviceBrand: undefined;
constructor(
@IUserDataProfilesService private readonly userDataProfilesService: IUserDataProfilesService,
@IUserDataProfileService private readonly userDataProfileService: IUserDataProfileService,
@IHostService private readonly hostService: IHostService,
@IDialogService private readonly dialogService: IDialogService,
@IWorkspaceContextService private readonly workspaceContextService: IWorkspaceContextService,
@IExtensionService private readonly extensionService: IExtensionService,
@IWorkbenchEnvironmentService private readonly environmentService: IWorkbenchEnvironmentService,
) {
super();
this._register(userDataProfilesService.onDidChangeProfiles(e => this.onDidChangeProfiles(e)));
this._register(userDataProfilesService.onDidResetWorkspaces(() => this.onDidResetWorkspaces()));
this._register(userDataProfileService.onDidChangeCurrentProfile(e => this.onDidChangeCurrentProfile(e)));
}
private onDidChangeProfiles(e: DidChangeProfilesEvent): void {
if (e.removed.some(profile => profile.id === this.userDataProfileService.currentProfile.id)) {
this.enterProfile(this.userDataProfilesService.defaultProfile, false, localize('reload message when removed', "The current profile has been removed. Please reload to switch back to default profile"));
return;
}
}
private onDidResetWorkspaces(): void {
if (!this.userDataProfileService.currentProfile.isDefault) {
this.enterProfile(this.userDataProfilesService.defaultProfile, false, localize('reload message when removed', "The current profile has been removed. Please reload to switch back to default profile"));
return;
}
}
private async onDidChangeCurrentProfile(e: DidChangeUserDataProfileEvent): Promise<void> {
if (e.previous.isTransient) {
await this.userDataProfilesService.cleanUpTransientProfiles();
}
}
async createAndEnterProfile(name: string, options?: IUserDataProfileOptions, fromExisting?: boolean): Promise<IUserDataProfile> {
const profile = await this.userDataProfilesService.createNamedProfile(name, options, this.getWorkspaceIdentifier());
await this.enterProfile(profile, !!fromExisting);
return profile;
}
async createAndEnterTransientProfile(): Promise<IUserDataProfile> {
const profile = await this.userDataProfilesService.createTransientProfile(this.getWorkspaceIdentifier());
await this.enterProfile(profile, false);
return profile;
}
async updateProfile(profile: IUserDataProfile, updateOptions: IUserDataProfileUpdateOptions): Promise<void> {
if (!this.userDataProfilesService.profiles.some(p => p.id === profile.id)) {
throw new Error(`Profile ${profile.name} does not exist`);
}
if (profile.isDefault) {
throw new Error(localize('cannotRenameDefaultProfile', "Cannot rename the default profile"));
}
await this.userDataProfilesService.updateProfile(profile, updateOptions);
}
async removeProfile(profile: IUserDataProfile): Promise<void> {
if (!this.userDataProfilesService.profiles.some(p => p.id === profile.id)) {
throw new Error(`Profile ${profile.name} does not exist`);
}
if (profile.isDefault) {
throw new Error(localize('cannotDeleteDefaultProfile', "Cannot delete the default profile"));
}
await this.userDataProfilesService.removeProfile(profile);
}
async switchProfile(profile: IUserDataProfile): Promise<void> {
const workspaceIdentifier = this.getWorkspaceIdentifier();
if (!this.userDataProfilesService.profiles.some(p => p.id === profile.id)) {
throw new Error(`Profile ${profile.name} does not exist`);
}
if (this.userDataProfileService.currentProfile.id === profile.id) {
return;
}
await this.userDataProfilesService.setProfileForWorkspace(workspaceIdentifier, profile);
await this.enterProfile(profile, false);
}
private getWorkspaceIdentifier(): WorkspaceIdentifier {
const workspace = this.workspaceContextService.getWorkspace();
switch (this.workspaceContextService.getWorkbenchState()) {
case WorkbenchState.FOLDER:
return { uri: workspace.folders[0].uri, id: workspace.id };
case WorkbenchState.WORKSPACE:
return { configPath: workspace.configuration!, id: workspace.id };
}
return 'empty-window';
}
private async enterProfile(profile: IUserDataProfile, preserveData: boolean, reloadMessage?: string): Promise<void> {
const isRemoteWindow = !!this.environmentService.remoteAuthority;
if (!isRemoteWindow) {
this.extensionService.stopExtensionHosts();
}
// In a remote window update current profile before reloading so that data is preserved from current profile if asked to preserve
await this.userDataProfileService.updateCurrentProfile(profile, preserveData);
if (isRemoteWindow) {
const result = await this.dialogService.confirm({
type: 'info',
message: reloadMessage ?? localize('reload message', "Switching a profile requires reloading VS Code."),
primaryButton: localize('reload button', "&&Reload"),
});
if (result.confirmed) {
await this.hostService.reload();
}
} else {
await this.extensionService.startExtensionHosts();
}
}
}
registerSingleton(IUserDataProfileManagementService, UserDataProfileManagementService, InstantiationType.Eager /* Eager because it updates the current window profile by listening to profiles changes */);
|
