azuredatastudio/src/sql/workbench/browser/scriptingActions.ts

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

import { Action } from 'vs/base/common/actions';
import * as nls from 'vs/nls';
import { IQueryEditorService } from 'sql/workbench/services/queryEditor/common/queryEditorService';
import { IConnectionManagementService } from 'sql/platform/connection/common/connectionManagement';
import { IScriptingService, ScriptOperation } from 'sql/platform/scripting/common/scriptingService';
import { BaseActionContext } from 'sql/workbench/browser/actions';
import { scriptSelect, script, scriptEditSelect } from 'sql/workbench/browser/scriptingUtils';
import { IErrorMessageService } from 'sql/platform/errorMessage/common/errorMessageService';

export class ScriptSelectAction extends Action {
	public static ID = 'selectTop';
	public static LABEL = nls.localize('scriptSelect', "Select Top 1000");
	public static KUSTOLABEL = nls.localize('scriptKustoSelect', "Take 10");

	constructor(
		id: string, label: string,
		@IQueryEditorService protected _queryEditorService: IQueryEditorService,
		@IConnectionManagementService protected _connectionManagementService: IConnectionManagementService,
		@IScriptingService protected _scriptingService: IScriptingService
	) {
		super(id, label);
	}

	public override async run(actionContext: BaseActionContext): Promise<void> {
		await scriptSelect(
			actionContext.profile!,
			actionContext.object!,
			this._connectionManagementService,
			this._queryEditorService,
			this._scriptingService
		);
	}
}

export class ScriptExecuteAction extends Action {
	public static ID = 'scriptExecute';
	public static LABEL = nls.localize('scriptExecute', "Script as Execute");

	constructor(
		id: string, label: string,
		@IQueryEditorService protected _queryEditorService: IQueryEditorService,
		@IConnectionManagementService protected _connectionManagementService: IConnectionManagementService,
		@IScriptingService protected _scriptingService: IScriptingService,
		@IErrorMessageService protected _errorMessageService: IErrorMessageService
	) {
		super(id, label);
	}

	public override async run(actionContext: BaseActionContext): Promise<void> {
		await script(
			actionContext.profile!,
			actionContext.object!,
			this._connectionManagementService,
			this._queryEditorService,
			this._scriptingService,
			ScriptOperation.Execute,
			this._errorMessageService
		);
	}
}

export class ScriptAlterAction extends Action {
	public static ID = 'scriptAlter';
	public static LABEL = nls.localize('scriptAlter', "Script as Alter");

	constructor(
		id: string, label: string,
		@IQueryEditorService protected _queryEditorService: IQueryEditorService,
		@IConnectionManagementService protected _connectionManagementService: IConnectionManagementService,
		@IScriptingService protected _scriptingService: IScriptingService,
		@IErrorMessageService protected _errorMessageService: IErrorMessageService
	) {
		super(id, label);
	}

	public override async run(actionContext: BaseActionContext): Promise<void> {
		await script(
			actionContext.profile!,
			actionContext.object!,
			this._connectionManagementService,
			this._queryEditorService,
			this._scriptingService,
			ScriptOperation.Alter,
			this._errorMessageService
		);
	}
}

export class EditDataAction extends Action {
	public static ID = 'editData';
	public static LABEL = nls.localize('editData', "Edit Data");

	constructor(
		id: string, label: string,
		@IQueryEditorService protected _queryEditorService: IQueryEditorService,
		@IConnectionManagementService protected _connectionManagementService: IConnectionManagementService,
		@IScriptingService protected _scriptingService: IScriptingService
	) {
		super(id, label);
	}

	public override async run(actionContext: BaseActionContext): Promise<void> {
		await scriptEditSelect(
			actionContext.profile!,
			actionContext.object!,
			this._connectionManagementService,
			this._queryEditorService,
			this._scriptingService
		);
	}
}

export class ScriptCreateAction extends Action {
	public static ID = 'scriptCreate';
	public static LABEL = nls.localize('scriptCreate', "Script as Create");

	constructor(
		id: string, label: string,
		@IQueryEditorService protected _queryEditorService: IQueryEditorService,
		@IConnectionManagementService protected _connectionManagementService: IConnectionManagementService,
		@IScriptingService protected _scriptingService: IScriptingService,
		@IErrorMessageService protected _errorMessageService: IErrorMessageService
	) {
		super(id, label);
	}

	public override async run(actionContext: BaseActionContext): Promise<void> {
		await script(
			actionContext.profile!,
			actionContext.object!,
			this._connectionManagementService,
			this._queryEditorService,
			this._scriptingService,
			ScriptOperation.Create,
			this._errorMessageService
		);
	}
}

export class ScriptDeleteAction extends Action {
	public static ID = 'scriptDelete';
	public static LABEL = nls.localize('scriptDelete', "Script as Drop");

	constructor(
		id: string, label: string,
		@IQueryEditorService protected _queryEditorService: IQueryEditorService,
		@IConnectionManagementService protected _connectionManagementService: IConnectionManagementService,
		@IScriptingService protected _scriptingService: IScriptingService,
		@IErrorMessageService protected _errorMessageService: IErrorMessageService
	) {
		super(id, label);
	}

	public override async run(actionContext: BaseActionContext): Promise<void> {
		await script(
			actionContext.profile!,
			actionContext.object!,
			this._connectionManagementService,
			this._queryEditorService,
			this._scriptingService,
			ScriptOperation.Delete,
			this._errorMessageService
		);
	}
}