Add azcli extension (#16029)

extensions/azcli/src/extension.ts

@@ -0,0 +1,73 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as azdataExt from 'azdata-ext';
+import * as rd from 'resource-deployment';
+import * as vscode from 'vscode';
+import { getExtensionApi } from './api';
+import { checkAndInstallAzdata, checkAndUpdateAzdata, findAzdata, isEulaAccepted, promptForEula } from './azdata';
+import Logger from './common/logger';
+import * as constants from './constants';
+import * as loc from './localizedConstants';
+import { ArcControllerConfigProfilesOptionsSource } from './providers/arcControllerConfigProfilesOptionsSource';
+import { AzdataToolService } from './services/azdataToolService';
+
+export async function activate(context: vscode.ExtensionContext): Promise<azdataExt.IExtension> {
+	const azdataToolService = new AzdataToolService();
+	let eulaAccepted: boolean = false;
+	vscode.commands.registerCommand('azdata.acceptEula', async () => {
+		await promptForEula(context.globalState, true /* userRequested */);
+	});
+
+	vscode.commands.registerCommand('azdata.install', async () => {
+		azdataToolService.localAzdata = await checkAndInstallAzdata(true /* userRequested */);
+	});
+
+	vscode.commands.registerCommand('azdata.update', async () => {
+		if (await checkAndUpdateAzdata(azdataToolService.localAzdata, true /* userRequested */)) { // if an update was performed
+			azdataToolService.localAzdata = await findAzdata(); // find and save the currently installed azdata
+		}
+	});
+
+	eulaAccepted = isEulaAccepted(context.globalState); // fetch eula acceptance state from memento
+	await vscode.commands.executeCommand('setContext', constants.eulaAccepted, eulaAccepted); // set a context key for current value of eulaAccepted state retrieved from memento so that command for accepting eula is available/unavailable in commandPalette appropriately.
+	Logger.log(loc.eulaAcceptedStateOnStartup(eulaAccepted));
+
+	// Don't block on this since we want the extension to finish activating without needing user input
+	const localAzdataDiscovered = checkAndInstallAzdata() // install if not installed and user wants it.
+		.then(async azdataTool => {
+			if (azdataTool !== undefined) {
+				azdataToolService.localAzdata = azdataTool;
+				if (!eulaAccepted) {
+					// Don't block on this since we want extension to finish activating without requiring user actions.
+					// If EULA has not been accepted then we will check again while executing azdata commands.
+					promptForEula(context.globalState)
+						.then(async (userResponse: boolean) => {
+							eulaAccepted = userResponse;
+						})
+						.catch((err) => console.log(err));
+				}
+				try {
+					//update if available and user wants it.
+					if (await checkAndUpdateAzdata(azdataToolService.localAzdata)) { // if an update was performed
+						azdataToolService.localAzdata = await findAzdata(); // find and save the currently installed azdata
+					}
+				} catch (err) {
+					vscode.window.showWarningMessage(loc.updateError(err));
+				}
+			}
+			return azdataTool;
+		});
+
+	const azdataApi = getExtensionApi(context.globalState, azdataToolService, localAzdataDiscovered);
+
+	// register option source(s)
+	const rdApi = <rd.IExtension>vscode.extensions.getExtension(rd.extension.name)?.exports;
+	context.subscriptions.push(rdApi.registerOptionsSourceProvider(new ArcControllerConfigProfilesOptionsSource(azdataApi)));
+
+	return azdataApi;
+}
+
+export function deactivate(): void { }