mirror of
https://github.com/ckaczor/azuredatastudio.git
synced 2026-01-14 01:25:37 -05:00
492 lines
20 KiB
TypeScript
492 lines
20 KiB
TypeScript
/*---------------------------------------------------------------------------------------------
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
|
* Licensed under the Source EULA. See License.txt in the project root for license information.
|
|
*--------------------------------------------------------------------------------------------*/
|
|
|
|
import {
|
|
IConnectionManagementService,
|
|
ConnectionType, INewConnectionParams, IConnectionCompletionOptions, IConnectionResult
|
|
} from 'sql/platform/connection/common/connectionManagement';
|
|
import { ConnectionDialogWidget, OnShowUIResponse } from 'sql/workbench/services/connection/browser/connectionDialogWidget';
|
|
import { ConnectionController } from 'sql/workbench/services/connection/browser/connectionController';
|
|
import * as WorkbenchUtils from 'sql/workbench/common/sqlWorkbenchUtils';
|
|
import * as Constants from 'sql/platform/connection/common/constants';
|
|
import { IConnectionProfile } from 'sql/platform/connection/common/interfaces';
|
|
import { ICapabilitiesService } from 'sql/platform/capabilities/common/capabilitiesService';
|
|
import { ConnectionProfile } from 'sql/platform/connection/common/connectionProfile';
|
|
import { entries } from 'sql/base/common/objects';
|
|
import { Deferred } from 'sql/base/common/promise';
|
|
import { IErrorMessageService } from 'sql/platform/errorMessage/common/errorMessageService';
|
|
import { IConnectionDialogService } from 'sql/workbench/services/connection/common/connectionDialogService';
|
|
import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
|
|
import * as platform from 'vs/base/common/platform';
|
|
import Severity from 'vs/base/common/severity';
|
|
import { Action, IAction } from 'vs/base/common/actions';
|
|
import { IClipboardService } from 'vs/platform/clipboard/common/clipboardService';
|
|
import { ICommandService } from 'vs/platform/commands/common/commands';
|
|
import * as types from 'vs/base/common/types';
|
|
import { trim } from 'vs/base/common/strings';
|
|
import { localize } from 'vs/nls';
|
|
import { IConfigurationService } from 'vs/platform/configuration/common/configuration';
|
|
import { IWorkbenchLayoutService } from 'vs/workbench/services/layout/browser/layoutService';
|
|
import { CmsConnectionController } from 'sql/workbench/services/connection/browser/cmsConnectionController';
|
|
|
|
export interface IConnectionValidateResult {
|
|
isValid: boolean;
|
|
connection: IConnectionProfile;
|
|
}
|
|
|
|
export interface IConnectionComponentCallbacks {
|
|
onSetConnectButton: (enable: boolean) => void;
|
|
onCreateNewServerGroup?: () => void;
|
|
onAdvancedProperties?: () => void;
|
|
onSetAzureTimeOut?: () => void;
|
|
onFetchDatabases?: (serverName: string, authenticationType: string, userName?: string, password?: string) => Promise<string[]>;
|
|
}
|
|
|
|
export interface IConnectionComponentController {
|
|
showUiComponent(container: HTMLElement, didChange?: boolean): void;
|
|
initDialog(providers: string[], model: IConnectionProfile): void;
|
|
validateConnection(): IConnectionValidateResult;
|
|
fillInConnectionInputs(connectionInfo: IConnectionProfile): void;
|
|
handleOnConnecting(): void;
|
|
handleResetConnection(): void;
|
|
focusOnOpen(): void;
|
|
closeDatabaseDropdown(): void;
|
|
databaseDropdownExpanded: boolean;
|
|
}
|
|
|
|
export class ConnectionDialogService implements IConnectionDialogService {
|
|
|
|
_serviceBrand: any;
|
|
|
|
private _connectionDialog: ConnectionDialogWidget;
|
|
private _connectionControllerMap: { [providerName: string]: IConnectionComponentController } = {};
|
|
private _model: ConnectionProfile;
|
|
private _params: INewConnectionParams;
|
|
private _options: IConnectionCompletionOptions;
|
|
private _inputModel: IConnectionProfile;
|
|
private _providerNameToDisplayNameMap: { [providerDisplayName: string]: string } = {};
|
|
private _providerTypes: string[] = [];
|
|
private _currentProviderType: string = Constants.mssqlProviderName;
|
|
private _previousProviderType: string = undefined;
|
|
private _connecting: boolean = false;
|
|
private _connectionErrorTitle = localize('connectionError', 'Connection error');
|
|
private _dialogDeferredPromise: Deferred<IConnectionProfile>;
|
|
private _toDispose = [];
|
|
|
|
/**
|
|
* This is used to work around the interconnectedness of this code
|
|
*/
|
|
private ignoreNextConnect = false;
|
|
private _connectionManagementService: IConnectionManagementService;
|
|
|
|
constructor(
|
|
@IWorkbenchLayoutService private layoutService: IWorkbenchLayoutService,
|
|
@IInstantiationService private _instantiationService: IInstantiationService,
|
|
@ICapabilitiesService private _capabilitiesService: ICapabilitiesService,
|
|
@IErrorMessageService private _errorMessageService: IErrorMessageService,
|
|
@IConfigurationService private _configurationService: IConfigurationService,
|
|
@IClipboardService private _clipboardService: IClipboardService,
|
|
@ICommandService private _commandService: ICommandService
|
|
) {
|
|
this.initializeConnectionProviders();
|
|
}
|
|
|
|
/**
|
|
* Set the initial value for the connection provider and listen to the provider change event
|
|
*/
|
|
private initializeConnectionProviders() {
|
|
this.setConnectionProviders();
|
|
if (this._capabilitiesService) {
|
|
this._capabilitiesService.onCapabilitiesRegistered(() => {
|
|
this.setConnectionProviders();
|
|
if (this._connectionDialog) {
|
|
this._connectionDialog.updateConnectionProviders(this._providerTypes, this._providerNameToDisplayNameMap);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Update the available provider types using the values from capabilities service
|
|
*/
|
|
private setConnectionProviders() {
|
|
if (this._capabilitiesService) {
|
|
this._providerTypes = [];
|
|
this._providerNameToDisplayNameMap = {};
|
|
entries(this._capabilitiesService.providers).forEach(p => {
|
|
this._providerTypes.push(p[1].connection.displayName);
|
|
this._providerNameToDisplayNameMap[p[0]] = p[1].connection.displayName;
|
|
});
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Gets the default provider with the following actions
|
|
* 1. Checks if master provider(map) has data
|
|
* 2. If so, filters provider parameter against master map
|
|
* 3. Fetches the result array and extracts the first element
|
|
* 4. If none of the above data exists, returns 'MSSQL'
|
|
* @returns: Default provider as string
|
|
*/
|
|
private getDefaultProviderName(): string {
|
|
let defaultProvider: string;
|
|
if (this._providerNameToDisplayNameMap) {
|
|
let keys = Object.keys(this._providerNameToDisplayNameMap);
|
|
let filteredKeys: string[];
|
|
if (keys && keys.length > 0) {
|
|
if (this._params && this._params.providers && this._params.providers.length > 0) {
|
|
//Filter providers from master keys.
|
|
filteredKeys = keys.filter(key => this._params.providers.includes(key));
|
|
}
|
|
if (filteredKeys && filteredKeys.length > 0) {
|
|
defaultProvider = filteredKeys[0];
|
|
}
|
|
else {
|
|
defaultProvider = keys[0];
|
|
}
|
|
}
|
|
}
|
|
if (!defaultProvider && this._configurationService) {
|
|
defaultProvider = WorkbenchUtils.getSqlConfigValue<string>(this._configurationService, Constants.defaultEngine);
|
|
}
|
|
// as a fallback, default to MSSQL if the value from settings is not available
|
|
return defaultProvider || Constants.mssqlProviderName;
|
|
}
|
|
|
|
private handleOnConnect(params: INewConnectionParams, profile?: IConnectionProfile): void {
|
|
if (!this._connecting) {
|
|
this._connecting = true;
|
|
this.handleProviderOnConnecting();
|
|
if (!profile) {
|
|
let result = this.uiController.validateConnection();
|
|
if (!result.isValid) {
|
|
this._connecting = false;
|
|
this._connectionDialog.resetConnection();
|
|
return;
|
|
}
|
|
profile = result.connection;
|
|
|
|
profile.serverName = trim(profile.serverName);
|
|
|
|
// append the port to the server name for SQL Server connections
|
|
if (this._currentProviderType === Constants.mssqlProviderName ||
|
|
this._currentProviderType === Constants.cmsProviderName) {
|
|
let portPropertyName: string = 'port';
|
|
let portOption: string = profile.options[portPropertyName];
|
|
if (portOption && portOption.indexOf(',') === -1) {
|
|
profile.serverName = profile.serverName + ',' + portOption;
|
|
}
|
|
profile.options[portPropertyName] = undefined;
|
|
profile.providerName = Constants.mssqlProviderName;
|
|
}
|
|
|
|
// Disable password prompt during reconnect if connected with an empty password
|
|
if (profile.password === '' && profile.savePassword === false) {
|
|
profile.savePassword = true;
|
|
}
|
|
|
|
this.handleDefaultOnConnect(params, profile);
|
|
} else {
|
|
profile.serverName = trim(profile.serverName);
|
|
this._connectionManagementService.addSavedPassword(profile).then(connectionWithPassword => {
|
|
this.handleDefaultOnConnect(params, connectionWithPassword);
|
|
});
|
|
}
|
|
}
|
|
}
|
|
|
|
private handleOnCancel(params: INewConnectionParams): void {
|
|
if (this.ignoreNextConnect) {
|
|
this._connectionDialog.resetConnection();
|
|
this._connectionDialog.close();
|
|
this.ignoreNextConnect = false;
|
|
this._dialogDeferredPromise.resolve(undefined);
|
|
return;
|
|
}
|
|
if (this.uiController.databaseDropdownExpanded) {
|
|
this.uiController.closeDatabaseDropdown();
|
|
} else {
|
|
if (params && params.input && params.connectionType === ConnectionType.editor) {
|
|
this._connectionManagementService.cancelEditorConnection(params.input);
|
|
} else {
|
|
this._connectionManagementService.cancelConnection(this._model);
|
|
}
|
|
if (params && params.input && params.input.onConnectCanceled) {
|
|
params.input.onConnectCanceled();
|
|
}
|
|
this._connectionDialog.resetConnection();
|
|
this._connecting = false;
|
|
}
|
|
this.uiController.databaseDropdownExpanded = false;
|
|
if (this._dialogDeferredPromise) {
|
|
this._dialogDeferredPromise.resolve(undefined);
|
|
}
|
|
}
|
|
|
|
private handleDefaultOnConnect(params: INewConnectionParams, connection: IConnectionProfile): Thenable<void> {
|
|
if (this.ignoreNextConnect) {
|
|
this._connectionDialog.resetConnection();
|
|
this._connectionDialog.close();
|
|
this.ignoreNextConnect = false;
|
|
this._connecting = false;
|
|
this._dialogDeferredPromise.resolve(connection);
|
|
return Promise.resolve();
|
|
}
|
|
let fromEditor = params && params.connectionType === ConnectionType.editor;
|
|
let isTemporaryConnection = params && params.connectionType === ConnectionType.temporary;
|
|
let uri: string = undefined;
|
|
if (fromEditor && params && params.input) {
|
|
uri = params.input.uri;
|
|
}
|
|
let options: IConnectionCompletionOptions = this._options || {
|
|
params: params,
|
|
saveTheConnection: !isTemporaryConnection,
|
|
showDashboard: params && params.showDashboard !== undefined ? params.showDashboard : !fromEditor && !isTemporaryConnection,
|
|
showConnectionDialogOnError: false,
|
|
showFirewallRuleOnError: true
|
|
};
|
|
|
|
return this._connectionManagementService.connectAndSaveProfile(connection, uri, options, params && params.input).then(connectionResult => {
|
|
this._connecting = false;
|
|
if (connectionResult && connectionResult.connected) {
|
|
this._connectionDialog.close();
|
|
if (this._dialogDeferredPromise) {
|
|
this._dialogDeferredPromise.resolve(connectionResult.connectionProfile);
|
|
}
|
|
} else if (connectionResult && connectionResult.errorHandled) {
|
|
this._connectionDialog.resetConnection();
|
|
} else {
|
|
this._connectionDialog.resetConnection();
|
|
this.showErrorDialog(Severity.Error, this._connectionErrorTitle, connectionResult.errorMessage, connectionResult.callStack);
|
|
}
|
|
}).catch(err => {
|
|
this._connecting = false;
|
|
this._connectionDialog.resetConnection();
|
|
this.showErrorDialog(Severity.Error, this._connectionErrorTitle, err);
|
|
});
|
|
}
|
|
|
|
private get uiController(): IConnectionComponentController {
|
|
// Find the provider name from the selected provider type, or throw an error if it does not correspond to a known provider
|
|
let providerName = this._currentProviderType;
|
|
if (!providerName) {
|
|
throw Error('Invalid provider type');
|
|
}
|
|
|
|
// Set the model name, initialize the controller if needed, and return the controller
|
|
this._model.providerName = providerName;
|
|
if (!this._connectionControllerMap[providerName]) {
|
|
if (providerName === Constants.cmsProviderName) {
|
|
this._connectionControllerMap[providerName] =
|
|
this._instantiationService.createInstance(CmsConnectionController,
|
|
this._connectionManagementService,
|
|
this._capabilitiesService.getCapabilities(providerName).connection, {
|
|
onSetConnectButton: (enable: boolean) => this.handleSetConnectButtonEnable(enable)
|
|
}, providerName);
|
|
} else {
|
|
this._connectionControllerMap[providerName] =
|
|
this._instantiationService.createInstance(ConnectionController,
|
|
this._connectionManagementService,
|
|
this._capabilitiesService.getCapabilities(providerName).connection, {
|
|
onSetConnectButton: (enable: boolean) => this.handleSetConnectButtonEnable(enable)
|
|
}, providerName);
|
|
}
|
|
}
|
|
return this._connectionControllerMap[providerName];
|
|
}
|
|
|
|
private handleSetConnectButtonEnable(enable: boolean): void {
|
|
this._connectionDialog.connectButtonState = enable;
|
|
}
|
|
|
|
private handleShowUiComponent(input: OnShowUIResponse) {
|
|
if (input.selectedProviderType) {
|
|
// If the call is for specific providers
|
|
let isParamProvider: boolean = false;
|
|
if (this._params && this._params.providers) {
|
|
this._params.providers.forEach((provider) => {
|
|
if (input.selectedProviderType === this._providerNameToDisplayNameMap[provider]) {
|
|
isParamProvider = true;
|
|
this._currentProviderType = provider;
|
|
}
|
|
});
|
|
}
|
|
if (!isParamProvider) {
|
|
this._currentProviderType = Object.keys(this._providerNameToDisplayNameMap).find((key) =>
|
|
this._providerNameToDisplayNameMap[key] === input.selectedProviderType
|
|
);
|
|
}
|
|
}
|
|
this._model.providerName = this._currentProviderType;
|
|
|
|
this._model = new ConnectionProfile(this._capabilitiesService, this._model);
|
|
this.uiController.showUiComponent(input.container);
|
|
|
|
}
|
|
|
|
private handleInitDialog() {
|
|
this.uiController.initDialog(this._params && this._params.providers, this._model);
|
|
}
|
|
|
|
private handleFillInConnectionInputs(connectionInfo: IConnectionProfile): void {
|
|
this._connectionManagementService.addSavedPassword(connectionInfo).then(connectionWithPassword => {
|
|
let model = this.createModel(connectionWithPassword);
|
|
this._model = model;
|
|
this.uiController.fillInConnectionInputs(model);
|
|
});
|
|
this._connectionDialog.updateProvider(this._providerNameToDisplayNameMap[connectionInfo.providerName]);
|
|
}
|
|
|
|
private handleProviderOnResetConnection(): void {
|
|
this.uiController.handleResetConnection();
|
|
}
|
|
|
|
private handleProviderOnConnecting(): void {
|
|
this.uiController.handleOnConnecting();
|
|
}
|
|
|
|
private updateModelServerCapabilities(model: IConnectionProfile) {
|
|
this._model = this.createModel(model);
|
|
if (this._model.providerName) {
|
|
this._currentProviderType = this._model.providerName;
|
|
if (this._connectionDialog) {
|
|
this._connectionDialog.updateProvider(this._providerNameToDisplayNameMap[this._currentProviderType]);
|
|
}
|
|
}
|
|
}
|
|
|
|
private createModel(model: IConnectionProfile): ConnectionProfile {
|
|
let defaultProvider = this.getDefaultProviderName();
|
|
let providerName = model ? model.providerName : defaultProvider;
|
|
providerName = providerName ? providerName : defaultProvider;
|
|
let newProfile = new ConnectionProfile(this._capabilitiesService, model || providerName);
|
|
newProfile.saveProfile = true;
|
|
newProfile.generateNewId();
|
|
// If connecting from a query editor set "save connection" to false
|
|
if (this._params && this._params.input && this._params.connectionType === ConnectionType.editor) {
|
|
newProfile.saveProfile = false;
|
|
}
|
|
return newProfile;
|
|
}
|
|
|
|
private showDialogWithModel(): Promise<void> {
|
|
return new Promise<void>((resolve, reject) => {
|
|
this.updateModelServerCapabilities(this._inputModel);
|
|
this.doShowDialog(this._params);
|
|
resolve(null);
|
|
});
|
|
}
|
|
|
|
public openDialogAndWait(
|
|
connectionManagementService: IConnectionManagementService,
|
|
params?: INewConnectionParams,
|
|
model?: IConnectionProfile,
|
|
connectionResult?: IConnectionResult,
|
|
doConnect: boolean = true): Thenable<IConnectionProfile> {
|
|
|
|
if (!doConnect) {
|
|
this.ignoreNextConnect = true;
|
|
}
|
|
this._dialogDeferredPromise = new Deferred<IConnectionProfile>();
|
|
|
|
this.showDialog(connectionManagementService, params,
|
|
model,
|
|
connectionResult).then(() => {
|
|
}, error => {
|
|
this._dialogDeferredPromise.reject(error);
|
|
});
|
|
return this._dialogDeferredPromise;
|
|
}
|
|
|
|
public showDialog(
|
|
connectionManagementService: IConnectionManagementService,
|
|
params?: INewConnectionParams,
|
|
model?: IConnectionProfile,
|
|
connectionResult?: IConnectionResult,
|
|
connectionOptions?: IConnectionCompletionOptions): Thenable<void> {
|
|
|
|
this._connectionManagementService = connectionManagementService;
|
|
|
|
this._options = connectionOptions;
|
|
this._params = params;
|
|
this._inputModel = model;
|
|
return new Promise<void>((resolve, reject) => {
|
|
this.updateModelServerCapabilities(model);
|
|
// If connecting from a query editor set "save connection" to false
|
|
if (params && (params.input && params.connectionType === ConnectionType.editor ||
|
|
params.connectionType === ConnectionType.temporary)) {
|
|
this._model.saveProfile = false;
|
|
}
|
|
|
|
resolve(this.showDialogWithModel().then(() => {
|
|
if (connectionResult && connectionResult.errorMessage) {
|
|
this.showErrorDialog(Severity.Error, this._connectionErrorTitle, connectionResult.errorMessage, connectionResult.callStack);
|
|
}
|
|
}));
|
|
});
|
|
}
|
|
|
|
private doShowDialog(params: INewConnectionParams): Promise<void> {
|
|
if (!this._connectionDialog) {
|
|
this._connectionDialog = this._instantiationService.createInstance(ConnectionDialogWidget, this._providerTypes, this._providerNameToDisplayNameMap[this._model.providerName], this._providerNameToDisplayNameMap);
|
|
this._connectionDialog.onCancel(() => {
|
|
this._connectionDialog.databaseDropdownExpanded = this.uiController.databaseDropdownExpanded;
|
|
this.handleOnCancel(this._connectionDialog.newConnectionParams);
|
|
});
|
|
this._connectionDialog.onConnect((profile) => {
|
|
this.handleOnConnect(this._connectionDialog.newConnectionParams, profile);
|
|
});
|
|
this._connectionDialog.onShowUiComponent((input) => this.handleShowUiComponent(input));
|
|
this._connectionDialog.onInitDialog(() => this.handleInitDialog());
|
|
this._connectionDialog.onFillinConnectionInputs((input) => this.handleFillInConnectionInputs(input));
|
|
this._connectionDialog.onResetConnection(() => this.handleProviderOnResetConnection());
|
|
this._connectionDialog.render();
|
|
this._previousProviderType = this._currentProviderType;
|
|
}
|
|
this._connectionDialog.newConnectionParams = params;
|
|
// if provider changed
|
|
if ((this._previousProviderType !== this._currentProviderType) ||
|
|
// or if currentProvider not set correctly yet
|
|
!(this._currentProviderType === Constants.cmsProviderName && params.providers && params.providers.length > 1)) {
|
|
this._previousProviderType = undefined;
|
|
this._connectionDialog.updateProvider(this._providerNameToDisplayNameMap[this.getDefaultProviderName()]);
|
|
} else {
|
|
this._connectionDialog.newConnectionParams = params;
|
|
}
|
|
return new Promise<void>(() => {
|
|
this._connectionDialog.open(this._connectionManagementService.getRecentConnections(params.providers).length > 0);
|
|
this.uiController.focusOnOpen();
|
|
});
|
|
}
|
|
|
|
private showErrorDialog(severity: Severity, headerTitle: string, message: string, messageDetails?: string): void {
|
|
// Kerberos errors are currently very hard to understand, so adding handling of these to solve the common scenario
|
|
// note that ideally we would have an extensible service to handle errors by error code and provider, but for now
|
|
// this solves the most common "hard error" that we've noticed
|
|
const helpLink = 'https://aka.ms/sqlopskerberos';
|
|
let actions: IAction[] = [];
|
|
if (!platform.isWindows && types.isString(message) && message.toLowerCase().includes('kerberos') && message.toLowerCase().includes('kinit')) {
|
|
message = [
|
|
localize('kerberosErrorStart', "Connection failed due to Kerberos error."),
|
|
localize('kerberosHelpLink', "Help configuring Kerberos is available at {0}", helpLink),
|
|
localize('kerberosKinit', "If you have previously connected you may need to re-run kinit.")
|
|
].join('\r\n');
|
|
actions.push(new Action('Kinit', 'Run kinit', null, true, () => {
|
|
this._connectionDialog.close();
|
|
this._clipboardService.writeText('kinit\r');
|
|
this._commandService.executeCommand('workbench.action.terminal.focus').then(resolve => {
|
|
// setTimeout to allow for terminal Instance to load.
|
|
setTimeout(() => {
|
|
return this._commandService.executeCommand('workbench.action.terminal.paste');
|
|
}, 10);
|
|
}).then(resolve => null, reject => null);
|
|
return null;
|
|
}));
|
|
|
|
}
|
|
this._errorMessageService.showDialog(severity, headerTitle, message, messageDetails, actions);
|
|
}
|
|
}
|