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

src/sql/parts/queryPlan/queryPlanEditor.ts

@@ -0,0 +1,127 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import 'vs/css!sql/parts/query/editor/media/queryEditor';
+import * as DOM from 'vs/base/browser/dom';
+import { TPromise } from 'vs/base/common/winjs.base';
+import { Dimension, Builder } from 'vs/base/browser/builder';
+import { EditorOptions } from 'vs/workbench/common/editor';
+import { BaseEditor } from 'vs/workbench/browser/parts/editor/baseEditor';
+import { ITelemetryService } from 'vs/platform/telemetry/common/telemetry';
+import { IThemeService } from 'vs/platform/theme/common/themeService';
+import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
+import { QueryPlanInput } from './queryPlanInput';
+import { QueryPlanModule } from './queryPlan.module';
+import { IConnectionManagementService } from 'sql/parts/connection/common/connectionManagement';
+import { IMetadataService } from 'sql/services/metadata/metadataService';
+import { IScriptingService } from 'sql/services/scripting/scriptingService';
+import { IQueryEditorService } from 'sql/parts/query/common/queryEditorService';
+import { IBootstrapService } from 'sql/services/bootstrap/bootstrapService';
+import { QueryPlanParams } from 'sql/services/bootstrap/bootstrapParams';
+import { QUERYPLAN_SELECTOR } from 'sql/parts/queryPlan/queryPlan.component';
+
+declare let QP;
+
+export class QueryPlanEditor extends BaseEditor {
+
+	public static ID: string = 'workbench.editor.queryplan';
+
+	constructor(
+		@ITelemetryService telemetryService: ITelemetryService,
+		@IThemeService themeService: IThemeService,
+		@IInstantiationService private instantiationService: IInstantiationService,
+		@IConnectionManagementService private _connectionService: IConnectionManagementService,
+		@IMetadataService private _metadataService: IMetadataService,
+		@IScriptingService private _scriptingService: IScriptingService,
+		@IQueryEditorService private _queryEditorService: IQueryEditorService,
+		@IBootstrapService private _bootstrapService: IBootstrapService
+	) {
+		super(QueryPlanEditor.ID, telemetryService, themeService);
+	}
+
+	/**
+	 * Called to create the editor in the parent builder.
+	 */
+	public createEditor(parent: Builder): void {
+		//Enable scrollbars when drawing area is larger than viewport
+		parent.overflow('auto');
+		//Set background of parent to white (same as .qp-root from src\sql\parts\grid\load\css\qp.css)
+		//This is because the bottom-most tooltips can extend past the drawing area, which causes the
+		//scrolling area to have gaps on the bottom and left. So if the colors aren't matched then
+		//these gaps show up as different colors and look bad.
+		//Another option would be to check the tooltip positions and reposition them if necessary
+		//during the load - but changing the background color was the simplest and least error prone
+		//(plus it's probable that we won't be using this control in the future anyways if development)
+		//continues on the Query plan feature
+		parent.background('#fff');
+	}
+
+	/**
+	 * Sets focus on this editor. Specifically, it sets the focus on the hosted text editor.
+	 */
+	public focus(): void {
+	}
+
+	/**
+	 * Updates the internal variable keeping track of the editor's size, and re-calculates the sash position.
+	 * To be called when the container of this editor changes size.
+	 */
+	public layout(dimension: Dimension): void {
+	}
+
+	public setInput(input: QueryPlanInput, options: EditorOptions): TPromise<void> {
+		if (this.input instanceof QueryPlanInput && this.input.matches(input)) {
+			return TPromise.as(undefined);
+		}
+
+		if (!input.hasInitialized) {
+			this.bootstrapAngular(input);
+		}
+		this.revealElementWithTagName(input.uniqueSelector, this.getContainer().getHTMLElement());
+
+		return super.setInput(input, options);
+	}
+
+	/**
+	 * Reveal the child element with the given tagName and hide all other elements.
+	 */
+	private revealElementWithTagName(tagName: string, parent: HTMLElement): void {
+		let elementToReveal: HTMLElement;
+
+		for (let i = 0; i < parent.children.length; i++) {
+			let child: HTMLElement = <HTMLElement>parent.children[i];
+			if (child.tagName && child.tagName.toLowerCase() === tagName && !elementToReveal) {
+				elementToReveal = child;
+			} else {
+				child.style.display = 'none';
+			}
+		}
+
+		if (elementToReveal) {
+			elementToReveal.style.display = '';
+		}
+	}
+
+	/**
+	 * Load the angular components and record for this input that we have done so
+	 */
+	private bootstrapAngular(input: QueryPlanInput): void {
+		// Get the bootstrap params and perform the bootstrap
+		let params: QueryPlanParams = {
+			planXml: input.planXml
+		};
+
+		let uniqueSelector = this._bootstrapService.bootstrap(
+			QueryPlanModule,
+			this.getContainer().getHTMLElement(),
+			QUERYPLAN_SELECTOR,
+			params);
+		input.setUniqueSelector(uniqueSelector);
+	}
+
+	public dispose(): void {
+		super.dispose();
+	}
+}