mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-01-15 09:35:37 -05:00
8570910a43
* close * connection is working * formatting * adds all * formatting * formatting and changing how features are initialized * formatting * changed named of typings file * update * updated to use dataprotocol npm * formatting * removed unneeded logging * readd npm shrinkwrap * still not working * removed unnecessary codfe * addressed comments * readded azure resource provider * fix capabilities cacheing * added backwards capat for older protocol * update shrinkwrap * update shrinkwrap * updated shrinkwrap * fixed tests * removed dead code * remove dead code * fix compile * remove backcompat stuff * change location of npm * vbump sqltools * merge master * fix imports * fix build breaks * update for sqlops * update yarn dependencies
65 lines
2.7 KiB
TypeScript
65 lines
2.7 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* 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 sqlops from 'sqlops';
|
|
import { TPromise } from 'vs/base/common/winjs.base';
|
|
import { IResourceProviderService } from 'sql/parts/accountManagement/common/interfaces';
|
|
import { dispose, IDisposable } from 'vs/base/common/lifecycle';
|
|
import {
|
|
ExtHostResourceProviderShape,
|
|
MainThreadResourceProviderShape,
|
|
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.MainThreadResourceProvider)
|
|
export class MainThreadResourceProvider implements MainThreadResourceProviderShape {
|
|
private _providerMetadata: { [handle: number]: sqlops.AccountProviderMetadata };
|
|
private _proxy: ExtHostResourceProviderShape;
|
|
private _toDispose: IDisposable[];
|
|
|
|
constructor(
|
|
extHostContext: IExtHostContext,
|
|
@IResourceProviderService private _resourceProviderService: IResourceProviderService
|
|
) {
|
|
this._providerMetadata = {};
|
|
if (extHostContext) {
|
|
this._proxy = extHostContext.get(SqlExtHostContext.ExtHostResourceProvider);
|
|
}
|
|
this._toDispose = [];
|
|
}
|
|
|
|
public $registerResourceProvider(providerMetadata: sqlops.ResourceProviderMetadata, handle: number): Thenable<any> {
|
|
let self = this;
|
|
|
|
// Create the account provider that interfaces with the extension via the proxy and register it
|
|
let resourceProvider: sqlops.ResourceProvider = {
|
|
createFirewallRule(account: sqlops.Account, firewallruleInfo: sqlops.FirewallRuleInfo): Thenable<sqlops.CreateFirewallRuleResponse> {
|
|
return self._proxy.$createFirewallRule(handle, account, firewallruleInfo);
|
|
},
|
|
handleFirewallRule(errorCode: number, errorMessage: string, connectionTypeId: string): Thenable<sqlops.HandleFirewallRuleResponse> {
|
|
return self._proxy.$handleFirewallRule(handle, errorCode, errorMessage, connectionTypeId);
|
|
}
|
|
};
|
|
this._resourceProviderService.registerProvider(providerMetadata.id, resourceProvider);
|
|
this._providerMetadata[handle] = providerMetadata;
|
|
|
|
return TPromise.as(null);
|
|
}
|
|
|
|
public $unregisterResourceProvider(handle: number): Thenable<any> {
|
|
this._resourceProviderService.unregisterProvider(this._providerMetadata[handle].id);
|
|
return TPromise.as(null);
|
|
}
|
|
|
|
public dispose(): void {
|
|
this._toDispose = dispose(this._toDispose);
|
|
}
|
|
}
|