azuredatastudio/extensions/notebook/src/common/appContext.ts

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

'use strict';

import * as vscode from 'vscode';
import { ApiWrapper } from './apiWrapper';

/**
 * Global context for the application
 */
export class AppContext {

	private serviceMap: Map<string, any> = new Map();
	constructor(public readonly extensionContext: vscode.ExtensionContext, public readonly apiWrapper: ApiWrapper) {
		this.apiWrapper = apiWrapper || new ApiWrapper();
	}

	public getService<T>(serviceName: string): T {
		return this.serviceMap.get(serviceName) as T;
	}

	public registerService<T>(serviceName: string, service: T): void {
		this.serviceMap.set(serviceName, service);
	}
}