ckaczor/azuredatastudio
1
0
Fork 0
mirror of https://github.com/ckaczor/azuredatastudio.git synced 2026-02-05 01:25:38 -05:00
Code Issues Packages Projects Releases Wiki Activity

Feature/sqlcmd - ADS changes to enable SQLCMD (#6555)

Browse Source 
* Basic SQLCMD functionality to run query and toggle mode

* icons and  some cleanup

* Adding PR comments
This commit is contained in:
Udeesha Gautam
2019-08-20 11:40:16 -07:00
committed by GitHub
parent c513204501
commit 9117b02c72
8 changed files with 147 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
src/sql/workbench/parts/query/browser/queryActions.ts
View File

@@ -15,7 +15,7 @@ import { IExtensionTipsService } from 'vs/workbench/services/extensionManagement
import Severity from 'vs/base/common/severity';
import { append, $ } from 'vs/base/browser/dom';
import { ISelectionData } from 'azdata';
import { ISelectionData, QueryExecutionOptions } from 'azdata';
import {
IConnectionManagementService,
IConnectionParams,
@@ -28,6 +28,7 @@ import { IQueryModelService } from 'sql/platform/query/common/queryModel';
import { SelectBox } from 'sql/base/browser/ui/selectBox/selectBox';
import { attachEditableDropdownStyler, attachSelectBoxStyler } from 'sql/platform/theme/common/styler';
import { Dropdown } from 'sql/base/parts/editableDropdown/browser/dropdown';
import { IQueryManagementService } from 'sql/platform/query/common/queryManagement';
/**
* Action class that query-based Actions will extend. This base class automatically handles activating and
@@ -415,6 +416,61 @@ export class ListDatabasesAction extends QueryTaskbarAction {
}
}
/**
* Action class that toggles SQLCMD mode for the editor
*/
export class ToggleSqlCmdModeAction extends QueryTaskbarAction {
public static EnableSqlcmdClass = 'enablesqlcmd';
public static DisableSqlcmdClass = 'disablesqlcmd';
public static ID = 'ToggleSqlCmdModeAction';
private _enablesqlcmdLabel = nls.localize('enablesqlcmdLabel', "Enable SQLCMD Mode");
private _disablesqlcmdLabel = nls.localize('disablesqlcmdLabel', "Disable SQLCMD Mode");
constructor(
editor: QueryEditor,
private _isSqlCmdMode: boolean,
@IQueryManagementService protected readonly queryManagementService: IQueryManagementService,
@IConfigurationService protected readonly configurationService: IConfigurationService,
@IConnectionManagementService connectionManagementService: IConnectionManagementService
) {
super(connectionManagementService, editor, ToggleSqlCmdModeAction.ID, undefined);
}
public get isSqlCmdMode(): boolean {
return this._isSqlCmdMode;
}
public set isSqlCmdMode(value: boolean) {
this._isSqlCmdMode = value;
this.updateLabelAndIcon();
}
private updateLabelAndIcon(): void {
// show option to disable sql cmd mode if already enabled
this.label = this.isSqlCmdMode ? this._disablesqlcmdLabel : this._enablesqlcmdLabel;
this.isSqlCmdMode ? this.updateCssClass(ToggleSqlCmdModeAction.DisableSqlcmdClass) : this.updateCssClass(ToggleSqlCmdModeAction.EnableSqlcmdClass);
}
private setSqlCmdModeFalse() {
}
public run(): Promise<void> {
const toSqlCmdState = !this.isSqlCmdMode; // input.state change triggers event that changes this.isSqlCmdMode, so store it before using
this.editor.input.state.isSqlCmdMode = toSqlCmdState;
// set query options
let queryoptions: QueryExecutionOptions = { options: new Map<string, any>() };
queryoptions.options['isSqlCmdMode'] = toSqlCmdState;
this.queryManagementService.setQueryExecutionOptions(this.editor.input.uri, queryoptions);
// set intellisense options
toSqlCmdState ? this.connectionManagementService.doChangeLanguageFlavor(this.editor.input.uri, 'sqlcmd', 'MSSQL') : this.connectionManagementService.doChangeLanguageFlavor(this.editor.input.uri, 'sql', 'MSSQL');
return Promise.resolve(null);
}
}
/*
* Action item that handles the dropdown (combobox) that lists the available databases.
* Based off StartDebugActionItem.