azuredatastudio/extensions/resource-deployment/src/services/optionSourcesService.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 rd from 'resource-deployment';
import * as loc from '../localizedConstants';

class OptionsSourcesService {
	private _optionsSourceStore = new Map<string, rd.IOptionsSourceProvider>();
	registerOptionsSourceProvider(provider: rd.IOptionsSourceProvider): vscode.Disposable {
		if (this._optionsSourceStore.has(provider.id)) {
			throw new Error(loc.optionsSourceAlreadyDefined(provider.id));
		}
		this._optionsSourceStore.set(provider.id, provider);
		return {
			dispose: () => this.unregisterOptionsSourceProvider(provider.id)
		};
	}

	private unregisterOptionsSourceProvider(providerId: string): void {
		this._optionsSourceStore.delete(providerId);
	}

	getOptionsSource(optionsSourceProviderId: string): rd.IOptionsSourceProvider {
		const optionsSource = this._optionsSourceStore.get(optionsSourceProviderId);
		if (optionsSource === undefined) {
			throw new Error(loc.noOptionsSourceDefined(optionsSourceProviderId));
		}
		return optionsSource;
	}
}

export const optionsSourcesService = new OptionsSourcesService();