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

src/sql/parts/query/execution/queryStatus.ts

@@ -0,0 +1,118 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { $, append, show, hide } from 'vs/base/browser/dom';
+import { IDisposable, combinedDisposable } from 'vs/base/common/lifecycle';
+import { IStatusbarItem } from 'vs/workbench/browser/parts/statusbar/statusbar';
+import { IEditorCloseEvent } from 'vs/workbench/common/editor';
+import { IWorkbenchEditorService } from 'vs/workbench/services/editor/common/editorService';
+import { IEditorGroupService } from 'vs/workbench/services/group/common/groupService';
+import { IQueryModelService } from 'sql/parts/query/execution/queryModel';
+import LocalizedConstants = require('sql/parts/query/common/localizedConstants');
+import * as WorkbenchUtils from 'sql/workbench/common/sqlWorkbenchUtils';
+
+// Query execution status
+enum QueryExecutionStatus{
+	Executing,
+	Completed
+}
+
+// Shows query status in the editor
+export class QueryStatusbarItem implements IStatusbarItem {
+
+	private _element: HTMLElement;
+	private _queryElement: HTMLElement;
+	private _queryStatusEditors: { [editorUri: string]: QueryExecutionStatus };
+	private _toDispose: IDisposable[];
+
+	constructor(
+		@IQueryModelService private _queryModelService: IQueryModelService,
+		@IWorkbenchEditorService private _editorService: IWorkbenchEditorService,
+		@IEditorGroupService private _editorGroupService: IEditorGroupService,
+	) {
+		this._queryStatusEditors = {};
+	}
+
+	public render(container: HTMLElement): IDisposable {
+		this._element = append(container, $('.query-statusbar-group'));
+		this._queryElement = append(this._element, $('div.query-statusbar-item'));
+		hide(this._queryElement);
+
+		this._toDispose = [];
+		this._toDispose.push(
+			this._queryModelService.onRunQueryStart((uri:string) => this._onRunQueryStart(uri)),
+			this._queryModelService.onRunQueryComplete((uri:string) => this._onRunQueryComplete(uri)),
+			this._editorGroupService.onEditorsChanged(() => this._onEditorsChanged()),
+			this._editorGroupService.getStacksModel().onEditorClosed(event => this._onEditorClosed(event))
+		);
+
+		return combinedDisposable(this._toDispose);
+	}
+
+	private _onEditorClosed(event: IEditorCloseEvent): void{
+		let uri = WorkbenchUtils.getEditorUri(event.editor);
+		if (uri && uri in this._queryStatusEditors) {
+			// If active editor is being closed, hide the query status.
+			let activeEditor = this._editorService.getActiveEditor();
+			if (activeEditor) {
+				let currentUri = WorkbenchUtils.getEditorUri(activeEditor.input);
+				if (uri === currentUri) {
+					hide(this._queryElement);
+				}
+			}
+			delete this._queryStatusEditors[uri];
+		}
+	}
+
+	private _onEditorsChanged(): void{
+		let activeEditor = this._editorService.getActiveEditor();
+		if (activeEditor) {
+			let uri = WorkbenchUtils.getEditorUri(activeEditor.input);
+
+			// Show active editor's query status
+			if (uri && uri in this._queryStatusEditors){
+				this._showStatus(uri);
+			} else {
+				hide(this._queryElement);
+			}
+		} else {
+			hide(this._queryElement);
+		}
+	}
+
+	private _onRunQueryStart(uri: string): void {
+		this._updateStatus(uri, QueryExecutionStatus.Executing);
+	}
+
+	private _onRunQueryComplete(uri: string): void {
+		this._updateStatus(uri, QueryExecutionStatus.Completed);
+	}
+
+	// Update query status for the editor
+	private _updateStatus(uri: string, newStatus: QueryExecutionStatus){
+		if (uri) {
+			this._queryStatusEditors[uri] = newStatus;
+			this._showStatus(uri);
+		}
+	}
+
+	// Show/hide query status for active editor
+	private _showStatus(uri: string): void{
+		let activeEditor = this._editorService.getActiveEditor();
+		if (activeEditor) {
+			let currentUri = WorkbenchUtils.getEditorUri(activeEditor.input);
+			if (uri === currentUri) {
+				switch(this._queryStatusEditors[uri]){
+					case QueryExecutionStatus.Executing:
+						this._queryElement.textContent = LocalizedConstants.msgStatusRunQueryInProgress;
+						show(this._queryElement);
+						break;
+					default:
+						hide(this._queryElement);
+				}
+			}
+		}
+	}
+}