azuredatastudio/extensions/arc/src/providers/arcControllersOptionsSourceProvider.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 arc from 'arc';
import * as azdata from 'azdata';
import * as rd from 'resource-deployment';
import { getControllerPassword, getRegisteredDataControllers, reacquireControllerPassword } from '../common/api';
import { CacheManager } from '../common/cacheManager';
import { throwUnless } from '../common/utils';
import * as loc from '../localizedConstants';
import { AzureArcTreeDataProvider } from '../ui/tree/azureArcTreeDataProvider';

/**
 * Class that provides options sources for an Arc Data Controller
 */
export class ArcControllersOptionsSourceProvider implements rd.IOptionsSourceProvider {
	private _cacheManager = new CacheManager<string, string>();
	readonly optionsSourceId = 'arc.controllers';
	constructor(private _treeProvider: AzureArcTreeDataProvider) { }

	async getOptions(): Promise<string[] | azdata.CategoryValue[]> {
		const controllers = await getRegisteredDataControllers(this._treeProvider);
		throwUnless(controllers !== undefined && controllers.length !== 0, loc.noControllersConnected);
		return controllers.map(ci => {
			return ci.label;
		});
	}

	private async retrieveVariable(key: string): Promise<string> {
		const [variableName, controllerLabel] = JSON.parse(key);
		const controller = (await getRegisteredDataControllers(this._treeProvider)).find(ci => ci.label === controllerLabel);
		throwUnless(controller !== undefined, loc.noControllerInfoFound(controllerLabel));
		switch (variableName) {
			case 'endpoint': return controller.info.url;
			case 'username': return controller.info.username;
			case 'password': return this.getPassword(controller);
			default: throw new Error(loc.variableValueFetchForUnsupportedVariable(variableName));
		}
	}

	getVariableValue(variableName: string, controllerLabel: string): Promise<string> {
		return this._cacheManager.getCacheEntry(JSON.stringify([variableName, controllerLabel]), this.retrieveVariable);
	}

	private async getPassword(controller: arc.DataController): Promise<string> {
		let password = await getControllerPassword(this._treeProvider, controller.info);
		if (!password) {
			password = await reacquireControllerPassword(this._treeProvider, controller.info);
		}
		throwUnless(password !== undefined, loc.noPasswordFound(controller.label));
		return password;
	}

	getIsPassword(variableName: string): boolean {
		switch (variableName) {
			case 'endpoint': return false;
			case 'username': return false;
			case 'password': return true;
			default: throw new Error(loc.isPasswordFetchForUnsupportedVariable(variableName));
		}
	}
}