SQL Operations Studio Public Preview 1 (0.23) release source code

src/sql/workbench/api/node/mainThreadAccountManagement.ts

@@ -0,0 +1,78 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 data from 'data';
+import {TPromise} from 'vs/base/common/winjs.base';
+import {IAccountManagementService} from 'sql/services/accountManagement/interfaces';
+import {dispose, IDisposable} from 'vs/base/common/lifecycle';
+import {
+	ExtHostAccountManagementShape,
+	MainThreadAccountManagementShape,
+	SqlExtHostContext,
+	SqlMainContext
+} from 'sql/workbench/api/node/sqlExtHost.protocol';
+import { IExtHostContext } from 'vs/workbench/api/node/extHost.protocol';
+import { extHostNamedCustomer } from 'vs/workbench/api/electron-browser/extHostCustomers';
+
+
+@extHostNamedCustomer(SqlMainContext.MainThreadAccountManagement)
+export class MainThreadAccountManagement extends MainThreadAccountManagementShape {
+	private _providerMetadata: {[handle: number]: data.AccountProviderMetadata};
+	private _proxy: ExtHostAccountManagementShape;
+	private _toDispose: IDisposable[];
+
+	constructor(
+		extHostContext: IExtHostContext,
+		@IAccountManagementService private _accountManagementService: IAccountManagementService
+	) {
+		super();
+		this._providerMetadata = {};
+		if (extHostContext) {
+			this._proxy = extHostContext.get(SqlExtHostContext.ExtHostAccountManagement);
+		}
+		this._toDispose = [];
+	}
+
+	public $performOAuthAuthorization(url, silent: boolean): Thenable<string> {
+		return this._accountManagementService.performOAuthAuthorization(url, silent);
+	}
+
+	public $registerAccountProvider(providerMetadata: data.AccountProviderMetadata, handle: number): Thenable<any> {
+		let self = this;
+
+		// Create the account provider that interfaces with the extension via the proxy and register it
+		let accountProvider: data.AccountProvider = {
+			clear(accountKey: data.AccountKey): Thenable<void> {
+				return self._proxy.$clear(handle, accountKey);
+			},
+			getSecurityToken(account: data.Account): Thenable<{}> {
+				return self._proxy.$getSecurityToken(handle, account);
+			},
+			initialize(restoredAccounts: data.Account[]): Thenable<data.Account[]> {
+				return self._proxy.$initialize(handle, restoredAccounts);
+			},
+			prompt(): Thenable<data.Account> {
+				return self._proxy.$prompt(handle);
+			},
+			refresh(account: data.Account): Thenable<data.Account> {
+				return self._proxy.$refresh(handle, account);
+			}
+		};
+		this._accountManagementService.registerProvider(providerMetadata, accountProvider);
+		this._providerMetadata[handle] = providerMetadata;
+
+		return TPromise.as(null);
+	}
+
+	public $unregisterAccountProvider(handle: number): Thenable<any> {
+		this._accountManagementService.unregisterProvider(this._providerMetadata[handle]);
+		return TPromise.as(null);
+	}
+
+	public dispose(): void {
+		this._toDispose = dispose(this._toDispose);
+	}
+}