azuredatastudio/extensions/vscode-account/src/extension.ts

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

import * as vscode from 'vscode';
import { AzureActiveDirectoryService, onDidChangeSessions } from './AADHelper';
import * as nls from 'vscode-nls';

const localize = nls.loadMessageBundle();

export const DEFAULT_SCOPES = 'https://management.core.windows.net/.default offline_access';

export async function activate(context: vscode.ExtensionContext) {

	const loginService = new AzureActiveDirectoryService();

	await loginService.initialize();

	context.subscriptions.push(vscode.authentication.registerAuthenticationProvider({
		id: 'microsoft',
		displayName: 'Microsoft',
		onDidChangeSessions: onDidChangeSessions.event,
		getSessions: () => Promise.resolve(loginService.sessions),
		login: async (scopes: string[]) => {
			try {
				await loginService.login(scopes.sort().join(' '));
				return loginService.sessions[0]!;
			} catch (e) {
				throw e;
			}
		},
		logout: async (id: string) => {
			return loginService.logout(id);
		}
	}));

	context.subscriptions.push(vscode.commands.registerCommand('microsoft.signin', () => {
		return loginService.login(DEFAULT_SCOPES);
	}));

	context.subscriptions.push(vscode.commands.registerCommand('microsoft.signout', async () => {
		const sessions = loginService.sessions;
		if (sessions.length === 0) {
			return;
		}

		if (sessions.length === 1) {
			await loginService.logout(loginService.sessions[0].id);
			onDidChangeSessions.fire();
			vscode.window.showInformationMessage(localize('signedOut', "Successfully signed out."));
			return;
		}

		const selectedSession = await vscode.window.showQuickPick(sessions.map(session => {
			return {
				id: session.id,
				label: session.accountName
			};
		}));

		if (selectedSession) {
			await loginService.logout(selectedSession.id);
			onDidChangeSessions.fire();
			vscode.window.showInformationMessage(localize('signedOut', "Successfully signed out."));
			return;
		}
	}));

	return;
}

// this method is called when your extension is deactivated
export function deactivate() { }