app.ts 32.6 KB
Newer Older
1 2 3 4 5
/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

6
import { app, ipcMain as ipc, systemPreferences, shell, Event, contentTracing, protocol, powerMonitor } from 'electron';
7
import { IProcessEnvironment, isWindows, isMacintosh } from 'vs/base/common/platform';
8
import { WindowsManager } from 'vs/code/electron-main/windows';
9
import { IWindowsService, OpenContext, ActiveWindowManager, IURIToOpen } from 'vs/platform/windows/common/windows';
B
Benjamin Pasero 已提交
10
import { WindowsChannel } from 'vs/platform/windows/common/windowsIpc';
11
import { WindowsService } from 'vs/platform/windows/electron-main/windowsService';
12
import { ILifecycleService, LifecycleMainPhase } from 'vs/platform/lifecycle/electron-main/lifecycleMain';
B
Benjamin Pasero 已提交
13
import { getShellEnvironment } from 'vs/code/node/shellEnv';
14
import { IUpdateService } from 'vs/platform/update/common/update';
J
Joao Moreno 已提交
15
import { UpdateChannel } from 'vs/platform/update/node/updateIpc';
16
import { Server as ElectronIPCServer } from 'vs/base/parts/ipc/electron-main/ipc.electron-main';
A
Alex Dima 已提交
17 18
import { Client } from 'vs/base/parts/ipc/common/ipc.net';
import { Server, connect } from 'vs/base/parts/ipc/node/ipc.net';
19
import { SharedProcess } from 'vs/code/electron-main/sharedProcess';
B
Benjamin Pasero 已提交
20
import { LaunchService, LaunchChannel, ILaunchService } from 'vs/platform/launch/electron-main/launchService';
21 22 23
import { IInstantiationService, ServicesAccessor } from 'vs/platform/instantiation/common/instantiation';
import { ServiceCollection } from 'vs/platform/instantiation/common/serviceCollection';
import { SyncDescriptor } from 'vs/platform/instantiation/common/descriptors';
24
import { ILogService } from 'vs/platform/log/common/log';
B
Benjamin Pasero 已提交
25
import { IStateService } from 'vs/platform/state/common/state';
26 27 28
import { IEnvironmentService } from 'vs/platform/environment/common/environment';
import { IConfigurationService } from 'vs/platform/configuration/common/configuration';
import { IURLService } from 'vs/platform/url/common/url';
J
Joao Moreno 已提交
29
import { URLHandlerChannelClient, URLServiceChannel } from 'vs/platform/url/node/urlIpc';
30
import { ITelemetryService } from 'vs/platform/telemetry/common/telemetry';
31
import { NullTelemetryService, combinedAppender, LogAppender } from 'vs/platform/telemetry/common/telemetryUtils';
J
Joao Moreno 已提交
32
import { TelemetryAppenderClient } from 'vs/platform/telemetry/node/telemetryIpc';
33
import { TelemetryService, ITelemetryServiceConfig } from 'vs/platform/telemetry/common/telemetryService';
34
import { resolveCommonProperties } from 'vs/platform/telemetry/node/commonProperties';
A
Alex Dima 已提交
35
import { getDelayedChannel, StaticRouter } from 'vs/base/parts/ipc/common/ipc';
36 37
import product from 'vs/platform/product/node/product';
import pkg from 'vs/platform/product/node/package';
38
import { ProxyAuthHandler } from 'vs/code/electron-main/auth';
39
import { Disposable } from 'vs/base/common/lifecycle';
B
Benjamin Pasero 已提交
40
import { IWindowsMainService, ICodeWindow } from 'vs/platform/windows/electron-main/windows';
B
Benjamin Pasero 已提交
41
import { IHistoryMainService } from 'vs/platform/history/common/history';
42
import { withUndefinedAsNull } from 'vs/base/common/types';
43
import { URI } from 'vs/base/common/uri';
J
Joao Moreno 已提交
44
import { WorkspacesChannel } from 'vs/platform/workspaces/node/workspacesIpc';
M
Martin Aeschlimann 已提交
45
import { IWorkspacesMainService, hasWorkspaceFileExtension } from 'vs/platform/workspaces/common/workspaces';
46
import { getMachineId } from 'vs/base/node/id';
47 48 49
import { Win32UpdateService } from 'vs/platform/update/electron-main/updateService.win32';
import { LinuxUpdateService } from 'vs/platform/update/electron-main/updateService.linux';
import { DarwinUpdateService } from 'vs/platform/update/electron-main/updateService.darwin';
50
import { IIssueService } from 'vs/platform/issue/common/issue';
J
Joao Moreno 已提交
51
import { IssueChannel } from 'vs/platform/issue/node/issueIpc';
P
Pine Wu 已提交
52
import { IssueService } from 'vs/platform/issue/electron-main/issueService';
53
import { LogLevelSetterChannel } from 'vs/platform/log/common/logIpc';
54
import { setUnexpectedErrorHandler, onUnexpectedError } from 'vs/base/common/errors';
J
Joao Moreno 已提交
55
import { ElectronURLListener } from 'vs/platform/url/electron-main/electronUrlListener';
J
Joao Moreno 已提交
56
import { serve as serveDriver } from 'vs/platform/driver/electron-main/driver';
A
Alex Dima 已提交
57
import { connectRemoteAgentManagement, ManagementPersistentConnection, IConnectionOptions } from 'vs/platform/remote/common/remoteAgentConnection';
58 59
import { IMenubarService } from 'vs/platform/menubar/common/menubar';
import { MenubarService } from 'vs/platform/menubar/electron-main/menubarService';
J
Joao Moreno 已提交
60
import { MenubarChannel } from 'vs/platform/menubar/node/menubarIpc';
M
Martin Aeschlimann 已提交
61
import { hasArgs } from 'vs/platform/environment/node/argv';
62
import { RunOnceScheduler } from 'vs/base/common/async';
63
import { registerContextMenuListener } from 'vs/base/parts/contextmenu/electron-main/contextmenu';
B
Benjamin Pasero 已提交
64
import { homedir } from 'os';
B
Benjamin Pasero 已提交
65
import { join, sep } from 'vs/base/common/path';
B
Benjamin Pasero 已提交
66
import { localize } from 'vs/nls';
B
Benjamin Pasero 已提交
67
import { Schemas } from 'vs/base/common/network';
A
Alex Dima 已提交
68
import { REMOTE_FILE_SYSTEM_CHANNEL_NAME } from 'vs/platform/remote/common/remoteAgentFileSystemChannel';
A
Tweaks  
Alex Dima 已提交
69
import { ResolvedAuthority } from 'vs/platform/remote/common/remoteAuthorityResolver';
J
Joao Moreno 已提交
70
import { SnapUpdateService } from 'vs/platform/update/electron-main/updateService.snap';
B
wip  
Benjamin Pasero 已提交
71 72
import { IStorageMainService, StorageMainService } from 'vs/platform/storage/node/storageMainService';
import { GlobalStorageDatabaseChannel } from 'vs/platform/storage/node/storageIpc';
73
import { startsWith } from 'vs/base/common/strings';
74 75 76 77 78
import { BackupMainService } from 'vs/platform/backup/electron-main/backupMainService';
import { IBackupMainService } from 'vs/platform/backup/common/backup';
import { HistoryMainService } from 'vs/platform/history/electron-main/historyMainService';
import { URLService } from 'vs/platform/url/common/urlService';
import { WorkspacesMainService } from 'vs/platform/workspaces/electron-main/workspacesMainService';
79
import { RemoteAgentConnectionContext } from 'vs/platform/remote/common/remoteAgentEnvironment';
A
Alex Dima 已提交
80
import { nodeSocketFactory } from 'vs/platform/remote/node/nodeSocketFactory';
A
Alex Dima 已提交
81
import { VSBuffer } from 'vs/base/common/buffer';
B
Benjamin Pasero 已提交
82
import { statSync } from 'fs';
I
isidor 已提交
83
import { ISignService } from 'vs/platform/sign/common/sign';
84 85
import { IDiagnosticsService } from 'vs/platform/diagnostics/common/diagnosticsService';
import { DiagnosticsService } from 'vs/platform/diagnostics/node/diagnosticsIpc';
86 87 88
import { FileService } from 'vs/platform/files/common/fileService';
import { IFileService } from 'vs/platform/files/common/files';
import { DiskFileSystemProvider } from 'vs/platform/files/node/diskFileSystemProvider';
89
import { ExtensionHostDebugBroadcastChannel } from 'vs/platform/debug/common/extensionHostDebugIpc';
90

91
export class CodeApplication extends Disposable {
B
Benjamin Pasero 已提交
92

93
	private static readonly MACHINE_ID_KEY = 'telemetry.machineId';
94
	private static readonly TRUE_MACHINE_ID_KEY = 'telemetry.trueMachineId';
B
Benjamin Pasero 已提交
95

96
	private windowsMainService: IWindowsMainService | undefined;
97 98

	constructor(
99 100
		private readonly mainIpcServer: Server,
		private readonly userEnv: IProcessEnvironment,
101 102 103 104
		@IInstantiationService private readonly instantiationService: IInstantiationService,
		@ILogService private readonly logService: ILogService,
		@IEnvironmentService private readonly environmentService: IEnvironmentService,
		@ILifecycleService private readonly lifecycleService: ILifecycleService,
S
Sandeep Somavarapu 已提交
105
		@IConfigurationService private readonly configurationService: IConfigurationService,
I
isidor 已提交
106 107
		@IStateService private readonly stateService: IStateService,
		@ISignService private readonly signService: ISignService
108
	) {
109 110
		super();

111 112 113 114 115 116
		this.registerListeners();
	}

	private registerListeners(): void {

		// We handle uncaught exceptions here to prevent electron from opening a dialog to the user
117
		setUnexpectedErrorHandler(err => this.onUnexpectedError(err));
B
Benjamin Pasero 已提交
118
		process.on('uncaughtException', err => this.onUnexpectedError(err));
119
		process.on('unhandledRejection', (reason: unknown) => onUnexpectedError(reason));
120

B
Benjamin Pasero 已提交
121 122
		// Dispose on shutdown
		this.lifecycleService.onWillShutdown(() => this.dispose());
123

124 125 126
		// Contextmenu via IPC support
		registerContextMenuListener();

127 128 129 130 131 132 133
		app.on('accessibility-support-changed', (event: Event, accessibilitySupportEnabled: boolean) => {
			if (this.windowsMainService) {
				this.windowsMainService.sendToAll('vscode:accessibilitySupportChanged', accessibilitySupportEnabled);
			}
		});

		app.on('activate', (event: Event, hasVisibleWindows: boolean) => {
J
Joao Moreno 已提交
134
			this.logService.trace('App#activate');
135 136 137 138 139 140 141

			// Mac only event: open new window when we get activated
			if (!hasVisibleWindows && this.windowsMainService) {
				this.windowsMainService.openNewWindow(OpenContext.DOCK);
			}
		});

142
		// Security related measures (https://electronjs.org/docs/tutorial/security)
143 144 145 146
		//
		// !!! DO NOT CHANGE without consulting the documentation !!!
		//
		// app.on('remote-get-guest-web-contents', event => event.preventDefault()); // TODO@Ben TODO@Matt revisit this need for <webview>
147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169
		app.on('remote-require', (event, sender, module) => {
			this.logService.trace('App#on(remote-require): prevented');

			event.preventDefault();
		});
		app.on('remote-get-global', (event, sender, module) => {
			this.logService.trace(`App#on(remote-get-global): prevented on ${module}`);

			event.preventDefault();
		});
		app.on('remote-get-builtin', (event, sender, module) => {
			this.logService.trace(`App#on(remote-get-builtin): prevented on ${module}`);

			if (module !== 'clipboard') {
				event.preventDefault();
			}
		});
		app.on('remote-get-current-window', event => {
			this.logService.trace(`App#on(remote-get-current-window): prevented`);

			event.preventDefault();
		});
		app.on('remote-get-current-web-contents', event => {
D
Daniel Imms 已提交
170 171 172 173 174
			// The driver needs access to web contents
			if (!this.environmentService.args.driver) {
				this.logService.trace(`App#on(remote-get-current-web-contents): prevented`);
				event.preventDefault();
			}
175
		});
176
		app.on('web-contents-created', (_event: Electron.Event, contents) => {
177
			contents.on('will-attach-webview', (event: Electron.Event, webPreferences, params) => {
178

179 180 181 182 183 184 185 186 187
				const isValidWebviewSource = (source: string): boolean => {
					if (!source) {
						return false;
					}

					if (source === 'data:text/html;charset=utf-8,%3C%21DOCTYPE%20html%3E%0D%0A%3Chtml%20lang%3D%22en%22%20style%3D%22width%3A%20100%25%3B%20height%3A%20100%25%22%3E%0D%0A%3Chead%3E%0D%0A%09%3Ctitle%3EVirtual%20Document%3C%2Ftitle%3E%0D%0A%3C%2Fhead%3E%0D%0A%3Cbody%20style%3D%22margin%3A%200%3B%20overflow%3A%20hidden%3B%20width%3A%20100%25%3B%20height%3A%20100%25%22%3E%0D%0A%3C%2Fbody%3E%0D%0A%3C%2Fhtml%3E') {
						return true;
					}

188
					const srcUri = URI.parse(source).fsPath.toLowerCase();
189 190
					const rootUri = URI.file(this.environmentService.appRoot).fsPath.toLowerCase();

191
					return startsWith(srcUri, rootUri + sep);
192 193
				};

194
				// Ensure defaults
M
Matt Bierner 已提交
195 196 197 198
				delete webPreferences.preload;
				webPreferences.nodeIntegration = false;

				// Verify URLs being loaded
199
				if (isValidWebviewSource(params.src) && isValidWebviewSource(webPreferences.preloadURL)) {
M
Matt Bierner 已提交
200 201
					return;
				}
M
Matt Bierner 已提交
202

203 204
				delete webPreferences.preloadUrl;

M
Matt Bierner 已提交
205
				// Otherwise prevent loading
B
Benjamin Pasero 已提交
206
				this.logService.error('webContents#web-contents-created: Prevented webview attach');
207

M
Matt Bierner 已提交
208 209 210 211
				event.preventDefault();
			});

			contents.on('will-navigate', event => {
B
Benjamin Pasero 已提交
212
				this.logService.error('webContents#will-navigate: Prevented webcontent navigation');
213

M
Matt Bierner 已提交
214 215
				event.preventDefault();
			});
216 217 218 219 220 221

			contents.on('new-window', (event: Event, url: string) => {
				event.preventDefault(); // prevent code that wants to open links

				shell.openExternal(url);
			});
M
Matt Bierner 已提交
222 223
		});

224
		let macOpenFileURIs: IURIToOpen[] = [];
225
		let runningTimeout: NodeJS.Timeout | null = null;
226
		app.on('open-file', (event: Event, path: string) => {
J
Joao Moreno 已提交
227
			this.logService.trace('App#open-file: ', path);
228 229 230
			event.preventDefault();

			// Keep in array because more might come!
231
			macOpenFileURIs.push(this.getURIToOpenFromPathSync(path));
232 233 234 235 236 237 238 239 240 241 242 243 244

			// Clear previous handler if any
			if (runningTimeout !== null) {
				clearTimeout(runningTimeout);
				runningTimeout = null;
			}

			// Handle paths delayed in case more are coming!
			runningTimeout = setTimeout(() => {
				if (this.windowsMainService) {
					this.windowsMainService.open({
						context: OpenContext.DOCK /* can also be opening from finder while app is running */,
						cli: this.environmentService.args,
S
Sandeep Somavarapu 已提交
245
						urisToOpen: macOpenFileURIs,
246
						gotoLineMode: false,
247 248
						preferNewWindow: true /* dropping on the dock or opening from finder prefers to open in a new window */
					});
249

250
					macOpenFileURIs = [];
251 252 253 254 255
					runningTimeout = null;
				}
			}, 100);
		});

256
		app.on('new-window-for-tab', () => {
257 258 259
			if (this.windowsMainService) {
				this.windowsMainService.openNewWindow(OpenContext.DESKTOP); //macOS native tab "+" button
			}
260 261
		});

262
		ipc.on('vscode:exit', (event: Event, code: number) => {
J
Joao Moreno 已提交
263
			this.logService.trace('IPC#vscode:exit', code);
264 265 266 267 268

			this.dispose();
			this.lifecycleService.kill(code);
		});

269
		ipc.on('vscode:fetchShellEnv', async (event: Event) => {
270
			const webContents = event.sender;
271 272

			try {
273
				const shellEnv = await getShellEnvironment(this.logService, this.environmentService);
J
Johannes Rieken 已提交
274 275 276
				if (!webContents.isDestroyed()) {
					webContents.send('vscode:acceptShellEnv', shellEnv);
				}
277
			} catch (error) {
J
Johannes Rieken 已提交
278 279 280
				if (!webContents.isDestroyed()) {
					webContents.send('vscode:acceptShellEnv', {});
				}
B
Benjamin Pasero 已提交
281

282 283
				this.logService.error('Error fetching shell env', error);
			}
284
		});
285

286 287
		ipc.on('vscode:toggleDevTools', (event: Event) => event.sender.toggleDevTools());
		ipc.on('vscode:openDevTools', (event: Event) => event.sender.openDevTools());
B
Benjamin Pasero 已提交
288

289
		ipc.on('vscode:reloadWindow', (event: Event) => event.sender.reload());
290

B
Benjamin Pasero 已提交
291
		// Some listeners after window opened
292 293 294
		(async () => {
			await this.lifecycleService.when(LifecycleMainPhase.AfterWindowOpen);

B
Benjamin Pasero 已提交
295
			// After waking up from sleep  (after window opened)
296 297 298 299 300 301
			powerMonitor.on('resume', () => {
				if (this.windowsMainService) {
					this.windowsMainService.sendToAll('vscode:osResume', undefined);
				}
			});

B
Benjamin Pasero 已提交
302
			// Keyboard layout changes (after window opened)
303 304 305 306 307 308 309
			const nativeKeymap = await import('native-keymap');
			nativeKeymap.onDidChangeKeyboardLayout(() => {
				if (this.windowsMainService) {
					this.windowsMainService.sendToAll('vscode:keyboardLayoutChanged', false);
				}
			});
		})();
310 311
	}

B
Benjamin Pasero 已提交
312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332
	private onUnexpectedError(err: Error): void {
		if (err) {

			// take only the message and stack property
			const friendlyError = {
				message: err.message,
				stack: err.stack
			};

			// handle on client side
			if (this.windowsMainService) {
				this.windowsMainService.sendToFocused('vscode:reportError', JSON.stringify(friendlyError));
			}
		}

		this.logService.error(`[uncaught exception in main]: ${err}`);
		if (err.stack) {
			this.logService.error(err.stack);
		}
	}

333
	async startup(): Promise<void> {
J
Joao Moreno 已提交
334 335 336
		this.logService.debug('Starting VS Code');
		this.logService.debug(`from: ${this.environmentService.appRoot}`);
		this.logService.debug('args:', this.environmentService.args);
337

338 339 340 341
		// Make sure we associate the program with the app user model id
		// This will help Windows to associate the running program with
		// any shortcut that is pinned to the taskbar and prevent showing
		// two icons in the taskbar for the same app.
342
		if (isWindows && product.win32AppUserModelId) {
343 344
			app.setAppUserModelId(product.win32AppUserModelId);
		}
345

346 347 348 349 350 351 352
		// Fix native tabs on macOS 10.13
		// macOS enables a compatibility patch for any bundle ID beginning with
		// "com.microsoft.", which breaks native tabs for VS Code when using this
		// identifier (from the official build).
		// Explicitly opt out of the patch here before creating any windows.
		// See: https://github.com/Microsoft/vscode/issues/35361#issuecomment-399794085
		try {
353
			if (isMacintosh && this.configurationService.getValue<boolean>('window.nativeTabs') === true && !systemPreferences.getUserDefault('NSUseImprovedLayoutPass', 'boolean')) {
354
				systemPreferences.setUserDefault('NSUseImprovedLayoutPass', 'boolean', true as any);
355 356 357 358 359
			}
		} catch (error) {
			this.logService.error(error);
		}

360
		// Create Electron IPC Server
361
		const electronIpcServer = new ElectronIPCServer();
362

363 364
		// Resolve unique machine ID
		this.logService.trace('Resolving machine identifier...');
365 366
		const { machineId, trueMachineId } = await this.resolveMachineId();
		this.logService.trace(`Resolved machine identifier: ${machineId} (trueMachineId: ${trueMachineId})`);
367

368 369 370 371 372
		// Spawn shared process after the first window has opened and 3s have passed
		const sharedProcess = this.instantiationService.createInstance(SharedProcess, machineId, this.userEnv);
		const sharedProcessClient = sharedProcess.whenReady().then(() => connect(this.environmentService.sharedIPCHandle, 'main'));
		this.lifecycleService.when(LifecycleMainPhase.AfterWindowOpen).then(() => {
			this._register(new RunOnceScheduler(async () => {
373
				const userEnv = await getShellEnvironment(this.logService, this.environmentService);
374 375 376 377

				sharedProcess.spawn(userEnv);
			}, 3000)).schedule();
		});
378

379
		// Services
380
		const appInstantiationService = await this.createServices(machineId, trueMachineId, sharedProcess, sharedProcessClient);
381

382 383
		// Create driver
		if (this.environmentService.driverHandle) {
B
Benjamin Pasero 已提交
384
			const server = await serveDriver(electronIpcServer, this.environmentService.driverHandle!, this.environmentService, appInstantiationService);
385

B
Benjamin Pasero 已提交
386 387
			this.logService.info('Driver started at:', this.environmentService.driverHandle);
			this._register(server);
388
		}
389

390 391 392
		// Setup Auth Handler
		const authHandler = appInstantiationService.createInstance(ProxyAuthHandler);
		this._register(authHandler);
B
Benjamin Pasero 已提交
393

394
		// Open Windows
395
		const windows = appInstantiationService.invokeFunction(accessor => this.openFirstWindow(accessor, electronIpcServer, sharedProcessClient));
B
Benjamin Pasero 已提交
396

397
		// Post Open Windows Tasks
398
		this.afterWindowOpen();
B
Benjamin Pasero 已提交
399

400 401 402
		// Tracing: Stop tracing after windows are ready if enabled
		if (this.environmentService.args.trace) {
			this.stopTracingEventually(windows);
B
Benjamin Pasero 已提交
403 404 405
		}
	}

406
	private async resolveMachineId(): Promise<{ machineId: string, trueMachineId?: string }> {
B
Benjamin Pasero 已提交
407

408 409 410 411 412
		// We cache the machineId for faster lookups on startup
		// and resolve it only once initially if not cached
		let machineId = this.stateService.getItem<string>(CodeApplication.MACHINE_ID_KEY);
		if (!machineId) {
			machineId = await getMachineId();
B
Benjamin Pasero 已提交
413

414 415
			this.stateService.setItem(CodeApplication.MACHINE_ID_KEY, machineId);
		}
416

417 418 419 420 421 422 423 424 425 426 427 428
		// Check if machineId is hashed iBridge Device
		let trueMachineId: string | undefined;
		if (isMacintosh && machineId === '6c9d2bc8f91b89624add29c0abeae7fb42bf539fa1cdb2e3e57cd668fa9bcead') {
			trueMachineId = this.stateService.getItem<string>(CodeApplication.TRUE_MACHINE_ID_KEY);
			if (!trueMachineId) {
				trueMachineId = await getMachineId();

				this.stateService.setItem(CodeApplication.TRUE_MACHINE_ID_KEY, trueMachineId);
			}
		}

		return { machineId, trueMachineId };
429 430
	}

431
	private async createServices(machineId: string, trueMachineId: string | undefined, sharedProcess: SharedProcess, sharedProcessClient: Promise<Client<string>>): Promise<IInstantiationService> {
432
		const services = new ServiceCollection();
433

434 435 436 437 438 439 440
		// Files
		const fileService = this._register(new FileService(this.logService));
		services.set(IFileService, fileService);

		const diskFileSystemProvider = this._register(new DiskFileSystemProvider(this.logService));
		fileService.registerProvider(Schemas.file, diskFileSystemProvider);

441 442 443 444
		switch (process.platform) {
			case 'win32':
				services.set(IUpdateService, new SyncDescriptor(Win32UpdateService));
				break;
445

446 447 448
			case 'linux':
				if (process.env.SNAP && process.env.SNAP_REVISION) {
					services.set(IUpdateService, new SyncDescriptor(SnapUpdateService, [process.env.SNAP, process.env.SNAP_REVISION]));
449
				} else {
450
					services.set(IUpdateService, new SyncDescriptor(LinuxUpdateService));
B
Benjamin Pasero 已提交
451
				}
452
				break;
453

454 455 456
			case 'darwin':
				services.set(IUpdateService, new SyncDescriptor(DarwinUpdateService));
				break;
457 458
		}

459 460
		services.set(IWindowsMainService, new SyncDescriptor(WindowsManager, [machineId, this.userEnv]));
		services.set(IWindowsService, new SyncDescriptor(WindowsService, [sharedProcess]));
461
		services.set(ILaunchService, new SyncDescriptor(LaunchService));
462 463 464 465

		const diagnosticsChannel = getDelayedChannel(sharedProcessClient.then(client => client.getChannel('diagnostics')));
		services.set(IDiagnosticsService, new SyncDescriptor(DiagnosticsService, [diagnosticsChannel]));

466
		services.set(IIssueService, new SyncDescriptor(IssueService, [machineId, this.userEnv]));
467
		services.set(IMenubarService, new SyncDescriptor(MenubarService));
468 469 470 471 472 473 474 475

		const storageMainService = new StorageMainService(this.logService, this.environmentService);
		services.set(IStorageMainService, storageMainService);
		this.lifecycleService.onWillShutdown(e => e.join(storageMainService.close()));

		const backupMainService = new BackupMainService(this.environmentService, this.configurationService, this.logService);
		services.set(IBackupMainService, backupMainService);

476 477 478
		services.set(IHistoryMainService, new SyncDescriptor(HistoryMainService));
		services.set(IURLService, new SyncDescriptor(URLService));
		services.set(IWorkspacesMainService, new SyncDescriptor(WorkspacesMainService));
479

480
		// Telemetry
481
		if (!this.environmentService.isExtensionDevelopment && !this.environmentService.args['disable-telemetry'] && !!product.enableTelemetry) {
482
			const channel = getDelayedChannel(sharedProcessClient.then(client => client.getChannel('telemetryAppender')));
483
			const appender = combinedAppender(new TelemetryAppenderClient(channel), new LogAppender(this.logService));
484
			const commonProperties = resolveCommonProperties(product.commit, pkg.version, machineId, this.environmentService.installSourcePath);
485
			const piiPaths = this.environmentService.extensionsPath ? [this.environmentService.appRoot, this.environmentService.extensionsPath] : [this.environmentService.appRoot];
486
			const config: ITelemetryServiceConfig = { appender, commonProperties, piiPaths, trueMachineId };
487

488
			services.set(ITelemetryService, new SyncDescriptor(TelemetryService, [config]));
489 490 491 492
		} else {
			services.set(ITelemetryService, NullTelemetryService);
		}

493
		// Init services that require it
494
		await backupMainService.initialize();
495

496
		return this.instantiationService.createChild(services);
497 498
	}

499 500
	private stopTracingEventually(windows: ICodeWindow[]): void {
		this.logService.info(`Tracing: waiting for windows to get ready...`);
501

502 503 504 505 506
		let recordingStopped = false;
		const stopRecording = (timeout: boolean) => {
			if (recordingStopped) {
				return;
			}
B
wip  
Benjamin Pasero 已提交
507

508
			recordingStopped = true; // only once
509

510 511 512 513 514 515 516 517 518 519 520 521 522 523 524
			contentTracing.stopRecording(join(homedir(), `${product.applicationName}-${Math.random().toString(16).slice(-4)}.trace.txt`), path => {
				if (!timeout) {
					if (this.windowsMainService) {
						this.windowsMainService.showMessageBox({
							type: 'info',
							message: localize('trace.message', "Successfully created trace."),
							detail: localize('trace.detail', "Please create an issue and manually attach the following file:\n{0}", path),
							buttons: [localize('trace.ok', "Ok")]
						}, this.windowsMainService.getLastActiveWindow());
					}
				} else {
					this.logService.info(`Tracing: data recorded (after 30s timeout) to ${path}`);
				}
			});
		};
525

526 527 528 529 530 531 532 533
		// Wait up to 30s before creating the trace anyways
		const timeoutHandle = setTimeout(() => stopRecording(true), 30000);

		// Wait for all windows to get ready and stop tracing then
		Promise.all(windows.map(window => window.ready())).then(() => {
			clearTimeout(timeoutHandle);
			stopRecording(false);
		});
534 535
	}

536
	private openFirstWindow(accessor: ServicesAccessor, electronIpcServer: ElectronIPCServer, sharedProcessClient: Promise<Client<string>>): ICodeWindow[] {
537 538 539 540 541 542 543 544 545

		// Register more Main IPC services
		const launchService = accessor.get(ILaunchService);
		const launchChannel = new LaunchChannel(launchService);
		this.mainIpcServer.registerChannel('launch', launchChannel);

		// Register more Electron IPC services
		const updateService = accessor.get(IUpdateService);
		const updateChannel = new UpdateChannel(updateService);
546
		electronIpcServer.registerChannel('update', updateChannel);
547

548 549
		const issueService = accessor.get(IIssueService);
		const issueChannel = new IssueChannel(issueService);
550
		electronIpcServer.registerChannel('issue', issueChannel);
551

B
Benjamin Pasero 已提交
552
		const workspacesService = accessor.get(IWorkspacesMainService);
553 554
		const workspacesChannel = new WorkspacesChannel(workspacesService);
		electronIpcServer.registerChannel('workspaces', workspacesChannel);
B
Benjamin Pasero 已提交
555

556 557
		const windowsService = accessor.get(IWindowsService);
		const windowsChannel = new WindowsChannel(windowsService);
558 559
		electronIpcServer.registerChannel('windows', windowsChannel);
		sharedProcessClient.then(client => client.registerChannel('windows', windowsChannel));
560

561 562
		const menubarService = accessor.get(IMenubarService);
		const menubarChannel = new MenubarChannel(menubarService);
563
		electronIpcServer.registerChannel('menubar', menubarChannel);
564

J
Joao Moreno 已提交
565 566
		const urlService = accessor.get(IURLService);
		const urlChannel = new URLServiceChannel(urlService);
567
		electronIpcServer.registerChannel('url', urlChannel);
J
Joao Moreno 已提交
568

569
		const storageMainService = accessor.get(IStorageMainService);
570
		const storageChannel = this._register(new GlobalStorageDatabaseChannel(this.logService, storageMainService as StorageMainService));
571
		electronIpcServer.registerChannel('storage', storageChannel);
B
wip  
Benjamin Pasero 已提交
572

S
Sandeep Somavarapu 已提交
573
		// Log level management
574
		const logLevelChannel = new LogLevelSetterChannel(accessor.get(ILogService));
575 576
		electronIpcServer.registerChannel('loglevel', logLevelChannel);
		sharedProcessClient.then(client => client.registerChannel('loglevel', logLevelChannel));
S
Sandeep Somavarapu 已提交
577

578 579 580
		// ExtensionHost Debug broadcast service
		electronIpcServer.registerChannel(ExtensionHostDebugBroadcastChannel.ChannelName, new ExtensionHostDebugBroadcastChannel());

581 582
		// Signal phase: ready (services set)
		this.lifecycleService.phase = LifecycleMainPhase.Ready;
583 584

		// Propagate to clients
J
Joao Moreno 已提交
585
		const windowsMainService = this.windowsMainService = accessor.get(IWindowsMainService);
J
Joao Moreno 已提交
586

J
Joao Moreno 已提交
587
		// Create a URL handler which forwards to the last active window
J
Joao Moreno 已提交
588
		const activeWindowManager = new ActiveWindowManager(windowsService);
J
Joao Moreno 已提交
589
		const activeWindowRouter = new StaticRouter(ctx => activeWindowManager.getActiveClientId().then(id => ctx === id));
590
		const urlHandlerChannel = electronIpcServer.getChannel('urlHandler', activeWindowRouter);
J
Joao Moreno 已提交
591
		const multiplexURLHandler = new URLHandlerChannelClient(urlHandlerChannel);
J
Joao Moreno 已提交
592

593 594
		// On Mac, Code can be running without any open windows, so we must create a window to handle urls,
		// if there is none
595
		if (isMacintosh) {
596 597 598
			const environmentService = accessor.get(IEnvironmentService);

			urlService.registerHandler({
599
				async handleURL(uri: URI): Promise<boolean> {
600
					if (windowsMainService.getWindowCount() === 0) {
601 602
						const cli = { ...environmentService.args };
						const [window] = windowsMainService.open({ context: OpenContext.API, cli, forceEmpty: true, gotoLineMode: true });
603

604 605 606
						await window.ready();

						return urlService.open(uri);
607 608
					}

609
					return false;
610 611 612 613
				}
			});
		}

614
		// Register the multiple URL handler
J
Joao Moreno 已提交
615 616
		urlService.registerHandler(multiplexURLHandler);

J
Joao Moreno 已提交
617
		// Watch Electron URLs and forward them to the UrlService
618
		const args = this.environmentService.args;
J
Joao Moreno 已提交
619
		const urls = args['open-url'] ? args._urls : [];
620
		const urlListener = new ElectronURLListener(urls || [], urlService, windowsMainService);
621
		this._register(urlListener);
J
Joao Moreno 已提交
622

623
		// Open our first window
624
		const macOpenFiles: string[] = (<any>global).macOpenFiles;
625
		const context = !!process.env['VSCODE_CLI'] ? OpenContext.CLI : OpenContext.DESKTOP;
M
Martin Aeschlimann 已提交
626 627 628
		const hasCliArgs = hasArgs(args._);
		const hasFolderURIs = hasArgs(args['folder-uri']);
		const hasFileURIs = hasArgs(args['file-uri']);
629
		const noRecentEntry = args['skip-add-to-recently-opened'] === true;
M
Martin Aeschlimann 已提交
630
		const waitMarkerFileURI = args.wait && args.waitMarkerFilePath ? URI.file(args.waitMarkerFilePath) : undefined;
631

632
		// new window if "-n" was used without paths
M
Martin Aeschlimann 已提交
633
		if (args['new-window'] && !hasCliArgs && !hasFolderURIs && !hasFileURIs) {
634
			return windowsMainService.open({
M
Martin Aeschlimann 已提交
635 636 637 638 639 640 641 642
				context,
				cli: args,
				forceNewWindow: true,
				forceEmpty: true,
				noRecentEntry,
				waitMarkerFileURI,
				initialStartup: true
			});
643
		}
B
Benjamin Pasero 已提交
644

645
		// mac: open-file event received on startup
B
Benjamin Pasero 已提交
646
		if (macOpenFiles && macOpenFiles.length && !hasCliArgs && !hasFolderURIs && !hasFileURIs) {
647
			return windowsMainService.open({
M
Martin Aeschlimann 已提交
648 649
				context: OpenContext.DOCK,
				cli: args,
650
				urisToOpen: macOpenFiles.map(file => this.getURIToOpenFromPathSync(file)),
M
Martin Aeschlimann 已提交
651 652
				noRecentEntry,
				waitMarkerFileURI,
653
				gotoLineMode: false,
M
Martin Aeschlimann 已提交
654 655
				initialStartup: true
			});
B
Benjamin Pasero 已提交
656 657
		}

M
Martin Aeschlimann 已提交
658
		// default: read paths from cli
659
		return windowsMainService.open({
M
Martin Aeschlimann 已提交
660 661 662 663 664 665
			context,
			cli: args,
			forceNewWindow: args['new-window'] || (!hasCliArgs && args['unity-launch']),
			diffMode: args.diff,
			noRecentEntry,
			waitMarkerFileURI,
666
			gotoLineMode: args.goto,
M
Martin Aeschlimann 已提交
667 668
			initialStartup: true
		});
669 670
	}

671 672 673 674 675
	private getURIToOpenFromPathSync(path: string): IURIToOpen {
		try {
			const fileStat = statSync(path);
			if (fileStat.isDirectory()) {
				return { folderUri: URI.file(path) };
676
			}
677

678 679
			if (hasWorkspaceFileExtension(path)) {
				return { workspaceUri: URI.file(path) };
680
			}
681 682
		} catch (error) {
			// ignore errors
683
		}
684

685 686
		return { fileUri: URI.file(path) };
	}
687

688
	private afterWindowOpen(): void {
689

690 691
		// Signal phase: after window open
		this.lifecycleService.phase = LifecycleMainPhase.AfterWindowOpen;
692

693 694
		// Remote Authorities
		this.handleRemoteAuthorities();
695
	}
696 697 698 699 700

	private handleRemoteAuthorities(): void {
		const connectionPool: Map<string, ActiveConnection> = new Map<string, ActiveConnection>();

		class ActiveConnection {
701
			private readonly _authority: string;
A
Alex Dima 已提交
702
			private readonly _connection: Promise<ManagementPersistentConnection>;
703
			private readonly _disposeRunner: RunOnceScheduler;
704

I
isidor 已提交
705
			constructor(authority: string, host: string, port: number, signService: ISignService) {
706
				this._authority = authority;
707

A
Alex Dima 已提交
708 709
				const options: IConnectionOptions = {
					commit: product.commit,
A
Alex Dima 已提交
710
					socketFactory: nodeSocketFactory,
A
Alex Dima 已提交
711 712 713 714
					addressProvider: {
						getAddress: () => {
							return Promise.resolve({ host, port });
						}
I
isidor 已提交
715
					},
I
Isidor Nikolic 已提交
716
					signService
A
Alex Dima 已提交
717
				};
718

A
Alex Dima 已提交
719
				this._connection = connectRemoteAgentManagement(options, authority, `main`);
720
				this._disposeRunner = new RunOnceScheduler(() => this.dispose(), 5000);
721 722
			}

723
			dispose(): void {
724 725
				this._disposeRunner.dispose();
				connectionPool.delete(this._authority);
726
				this._connection.then(connection => connection.dispose());
727 728
			}

A
Alex Dima 已提交
729
			async getClient(): Promise<Client<RemoteAgentConnectionContext>> {
730
				this._disposeRunner.schedule();
A
Alex Dima 已提交
731
				const connection = await this._connection;
732

A
Alex Dima 已提交
733
				return connection.client;
734 735 736 737
			}
		}

		const resolvedAuthorities = new Map<string, ResolvedAuthority>();
738
		ipc.on('vscode:remoteAuthorityResolved', (event: Electron.Event, data: ResolvedAuthority) => {
739
			this.logService.info('Received resolved authority', data.authority);
740

741
			resolvedAuthorities.set(data.authority, data);
742

743 744
			// Make sure to close and remove any existing connections
			if (connectionPool.has(data.authority)) {
A
Alex Dima 已提交
745
				connectionPool.get(data.authority)!.dispose();
746
			}
747 748 749
		});

		const resolveAuthority = (authority: string): ResolvedAuthority | null => {
750
			this.logService.info('Resolving authority', authority);
751

752 753
			if (authority.indexOf('+') >= 0) {
				if (resolvedAuthorities.has(authority)) {
M
Matt Bierner 已提交
754
					return withUndefinedAsNull(resolvedAuthorities.get(authority));
755
				}
756

757
				this.logService.info('Didnot find resolved authority for', authority);
758

759 760 761 762
				return null;
			} else {
				const [host, strPort] = authority.split(':');
				const port = parseInt(strPort, 10);
763

764
				return { authority, host, port };
765 766 767
			}
		};

B
Benjamin Pasero 已提交
768
		protocol.registerBufferProtocol(Schemas.vscodeRemote, async (request, callback) => {
769
			if (request.method !== 'GET') {
770
				return callback(undefined);
771
			}
772

773 774
			const uri = URI.parse(request.url);

775
			let activeConnection: ActiveConnection | undefined;
776 777 778
			if (connectionPool.has(uri.authority)) {
				activeConnection = connectionPool.get(uri.authority);
			} else {
779
				const resolvedAuthority = resolveAuthority(uri.authority);
780
				if (!resolvedAuthority) {
781
					callback(undefined);
782 783
					return;
				}
784

I
isidor 已提交
785
				activeConnection = new ActiveConnection(uri.authority, resolvedAuthority.host, resolvedAuthority.port, this.signService);
786 787
				connectionPool.set(uri.authority, activeConnection);
			}
788

789
			try {
790
				const rawClient = await activeConnection!.getClient();
791 792 793 794
				if (connectionPool.has(uri.authority)) { // not disposed in the meantime
					const channel = rawClient.getChannel(REMOTE_FILE_SYSTEM_CHANNEL_NAME);

					// TODO@alex don't use call directly, wrap it around a `RemoteExtensionsFileSystemProvider`
A
Alex Dima 已提交
795 796
					const fileContents = await channel.call<VSBuffer>('readFile', [uri]);
					callback(<Buffer>fileContents.buffer);
797
				} else {
798
					callback(undefined);
799 800
				}
			} catch (err) {
801
				onUnexpectedError(err);
802
				callback(undefined);
803 804 805
			}
		});
	}
806
}