azuredatastudio/extensions/mssql/src/contextProvider.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 * as azdata from 'azdata';

import * as types from './types';
import * as Constants from './constants';

enum BuiltInCommands {
	SetContext = 'setContext',
}

enum ContextKeys {
	ISCLOUD = 'mssql:iscloud',
	EDITIONID = 'mssql:engineedition',
	ISCLUSTER = 'mssql:iscluster',
	SERVERMAJORVERSION = 'mssql:servermajorversion'
}

const isCloudEditions = [
	azdata.DatabaseEngineEdition.SqlDatabase,
	azdata.DatabaseEngineEdition.SqlDataWarehouse,
	azdata.DatabaseEngineEdition.SqlOnDemand
];

function setCommandContext(key: ContextKeys | string, value: any) {
	return vscode.commands.executeCommand(BuiltInCommands.SetContext, key, value);
}

export default class ContextProvider {
	private _disposables = new Array<vscode.Disposable>();

	constructor() {
		this._disposables.push(azdata.workspace.onDidOpenDashboard(this.onDashboardOpen, this));
		this._disposables.push(azdata.workspace.onDidChangeToDashboard(this.onDashboardOpen, this));
	}

	public onDashboardOpen(e: azdata.DashboardDocument): void {
		let iscloud: boolean;
		let edition: number;
		let isCluster: boolean = false;
		let serverMajorVersion: number;
		if (e.profile.providerName.toLowerCase() === 'mssql' && !types.isUndefinedOrNull(e.serverInfo) && !types.isUndefinedOrNull(e.serverInfo.engineEditionId)) {
			if (isCloudEditions.some(i => i === e.serverInfo.engineEditionId)) {
				iscloud = true;
			} else {
				iscloud = false;
			}

			edition = e.serverInfo.engineEditionId;

			if (!types.isUndefinedOrNull(e.serverInfo.options)) {
				let isBigDataCluster = e.serverInfo.options[Constants.isBigDataClusterProperty];
				if (isBigDataCluster) {
					isCluster = isBigDataCluster;
				}
			}
			serverMajorVersion = e.serverInfo.serverMajorVersion;
		}

		if (iscloud === true || iscloud === false) {
			void setCommandContext(ContextKeys.ISCLOUD, iscloud);
		}

		if (!types.isUndefinedOrNull(edition)) {
			void setCommandContext(ContextKeys.EDITIONID, edition);
		}

		if (!types.isUndefinedOrNull(isCluster)) {
			void setCommandContext(ContextKeys.ISCLUSTER, isCluster);
		}

		if (!types.isUndefinedOrNull(serverMajorVersion)) {
			void setCommandContext(ContextKeys.SERVERMAJORVERSION, serverMajorVersion);
		}
	}

	dispose(): void {
		this._disposables = this._disposables.map(i => i.dispose());
	}
}