azuredatastudio/src/vs/workbench/contrib/debug/browser/debugProgress.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import { Event } from 'vs/base/common/event';
import { IWorkbenchContribution } from 'vs/workbench/common/contributions';
import { IDebugService, VIEWLET_ID, IDebugSession } from 'vs/workbench/contrib/debug/common/debug';
import { IProgressService, ProgressLocation } from 'vs/platform/progress/common/progress';
import { dispose, IDisposable } from 'vs/base/common/lifecycle';

export class DebugProgressContribution implements IWorkbenchContribution {

	private toDispose: IDisposable[] = [];

	constructor(
		@IDebugService private readonly debugService: IDebugService,
		@IProgressService private readonly progressService: IProgressService
	) {
		let progressListener: IDisposable;
		const onFocusSession = (session: IDebugSession | undefined) => {
			if (progressListener) {
				progressListener.dispose();
			}
			if (session) {
				progressListener = session.onDidProgressStart(async progressStartEvent => {
					const promise = new Promise<void>(r => {
						// Show progress until a progress end event comes or the session ends
						const listener = Event.any(Event.filter(session.onDidProgressEnd, e => e.body.progressId === progressStartEvent.body.progressId),
							session.onDidEndAdapter)(() => {
								listener.dispose();
								r();
							});
					});

					this.progressService.withProgress({ location: VIEWLET_ID }, () => promise);
					const source = this.debugService.getConfigurationManager().getDebuggerLabel(session);
					this.progressService.withProgress({
						location: ProgressLocation.Notification,
						title: progressStartEvent.body.title,
						cancellable: progressStartEvent.body.cancellable,
						silent: true,
						source,
						delay: 500
					}, progressStep => {
						let increment = 0;
						const progressUpdateListener = session.onDidProgressUpdate(e => {
							if (e.body.progressId === progressStartEvent.body.progressId) {
								if (typeof e.body.percentage === 'number') {
									increment = e.body.percentage - increment;
								}
								progressStep.report({
									message: e.body.message,
									increment: typeof e.body.percentage === 'number' ? increment : undefined,
									total: typeof e.body.percentage === 'number' ? 100 : undefined,
								});
							}
						});

						return promise.then(() => progressUpdateListener.dispose());
					}, () => session.cancel(progressStartEvent.body.progressId));
				});
			}
		};
		this.toDispose.push(this.debugService.getViewModel().onDidFocusSession(onFocusSession));
		onFocusSession(this.debugService.getViewModel().focusedSession);
	}

	dispose(): void {
		dispose(this.toDispose);
	}
}