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

import * as GridContentEvents from 'sql/workbench/services/query/common/gridContentEvents';
import QueryRunner from 'sql/workbench/services/query/common/queryRunner';
import { ICellValue, ResultSetSubset } from 'sql/workbench/services/query/common/query';
import { DataService } from 'sql/workbench/services/query/common/dataService';
import { IQueryModelService, IQueryEvent } from 'sql/workbench/services/query/common/queryModel';

import * as azdata from 'azdata';

import * as nls from 'vs/nls';
import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
import { ILogService } from 'vs/platform/log/common/log';
import { Event, Emitter } from 'vs/base/common/event';
import * as strings from 'vs/base/common/strings';
import * as types from 'vs/base/common/types';
import { INotificationService } from 'vs/platform/notification/common/notification';
import Severity from 'vs/base/common/severity';
import EditQueryRunner from 'sql/workbench/services/editData/common/editQueryRunner';
import { IRange } from 'vs/editor/common/core/range';

const selectionSnippetMaxLen = 100;

export interface QueryEvent {
	type: string;
	data: any;
}

/**
 * Holds information about the state of a query runner
 */
export class QueryInfo {
	public queryRunner?: EditQueryRunner;
	public dataService?: DataService;
	public queryEventQueue: QueryEvent[] = [];
	public batchRanges: Array<IRange> = [];
	public selectionSnippet?: string;

	// Notes if the angular components have obtained the DataService. If not, all messages sent
	// via the data service will be lost.
	public dataServiceReady: boolean = false;

	constructor() { }

	public set uri(newUri: string) {
		if (this.queryRunner) {
			this.queryRunner.uri = newUri;
		}
		if (this.dataService) {
			this.dataService.uri = newUri;
		}
	}
}

/**
 * Handles running queries and grid interactions for all URIs. Interacts with each URI's results grid via a DataService instance
 */
export class QueryModelService implements IQueryModelService {
	_serviceBrand: undefined;

	// MEMBER VARIABLES ////////////////////////////////////////////////////
	private _queryInfoMap: Map<string, QueryInfo>;
	private _onRunQueryStart: Emitter<string>;
	private _onRunQueryUpdate: Emitter<string>;
	private _onRunQueryComplete: Emitter<string>;
	private _onQueryEvent: Emitter<IQueryEvent>;
	private _onEditSessionReady: Emitter<azdata.EditSessionReadyParams>;
	private _onCellSelectionChangedEmitter = new Emitter<ICellValue[]>();

	// EVENTS /////////////////////////////////////////////////////////////
	public get onRunQueryStart(): Event<string> { return this._onRunQueryStart.event; }
	public get onRunQueryUpdate(): Event<string> { return this._onRunQueryUpdate.event; }
	public get onRunQueryComplete(): Event<string> { return this._onRunQueryComplete.event; }
	public get onQueryEvent(): Event<IQueryEvent> { return this._onQueryEvent.event; }
	public get onEditSessionReady(): Event<azdata.EditSessionReadyParams> { return this._onEditSessionReady.event; }
	public get onCellSelectionChanged(): Event<ICellValue[]> { return this._onCellSelectionChangedEmitter.event; }

	// CONSTRUCTOR /////////////////////////////////////////////////////////
	constructor(
		@IInstantiationService private _instantiationService: IInstantiationService,
		@INotificationService private _notificationService: INotificationService,
		@ILogService private _logService: ILogService
	) {
		this._queryInfoMap = new Map<string, QueryInfo>();
		this._onRunQueryStart = new Emitter<string>();
		this._onRunQueryUpdate = new Emitter<string>();
		this._onRunQueryComplete = new Emitter<string>();
		this._onQueryEvent = new Emitter<IQueryEvent>();
		this._onEditSessionReady = new Emitter<azdata.EditSessionReadyParams>();
	}

	// IQUERYMODEL /////////////////////////////////////////////////////////
	public getDataService(uri: string): DataService {
		let dataService: DataService | undefined;
		if (this._queryInfoMap.has(uri)) {
			dataService = this._getQueryInfo(uri)!.dataService;
		}
		if (!dataService) {
			throw new Error('Could not find data service for uri: ' + uri);
		}

		return dataService;
	}

	/**
	 * Notify the event subscribers about the new selected cell values
	 * @param selectedCells current selected cells
	 */
	public notifyCellSelectionChanged(selectedCells: ICellValue[]): void {
		this._onCellSelectionChangedEmitter.fire(selectedCells);
	}

	/**
	 * Force all grids to re-render. This is needed to re-render the grids when switching
	 * between different URIs.
	 */
	public refreshResultsets(uri: string): void {
		this._fireGridContentEvent(uri, GridContentEvents.RefreshContents);
	}

	/**
	 * Resize the grid UI to fit the current screen size.
	 */
	public resizeResultsets(uri: string): void {
		this._fireGridContentEvent(uri, GridContentEvents.ResizeContents);
	}

	public sendGridContentEvent(uri: string, eventName: string): void {
		this._fireGridContentEvent(uri, eventName);
	}

	/**
	 * To be called by a component's DataService when the component has finished loading.
	 * Sends all previously enqueued query events to the DataService and signals to stop enqueuing
	 * any further events.
	 */
	public onLoaded(uri: string) {
		if (this._queryInfoMap.has(uri)) {
			let info = this._getQueryInfo(uri)!;
			info.dataServiceReady = true;
			this._sendQueuedEvents(uri);
		}
	}

	/**
	 * Get more data rows from the current resultSets from the service layer
	 */
	public getQueryRows(uri: string, rowStart: number, numberOfRows: number, batchId: number, resultId: number): Promise<ResultSetSubset | undefined> {
		if (this._queryInfoMap.has(uri)) {
			return this._getQueryInfo(uri)!.queryRunner!.getQueryRows(rowStart, numberOfRows, batchId, resultId).then(results => {
				return results;
			});
		} else {
			return Promise.resolve(undefined);
		}
	}

	public getEditRows(uri: string, rowStart: number, numberOfRows: number): Promise<azdata.EditSubsetResult | undefined> {
		if (this._queryInfoMap.has(uri)) {
			return this._queryInfoMap.get(uri)!.queryRunner!.getEditRows(rowStart, numberOfRows).then(results => {
				return results;
			});
		} else {
			return Promise.resolve(undefined);
		}
	}

	public async copyResults(uri: string, selection: Slick.Range[], batchId: number, resultId: number, includeHeaders?: boolean): Promise<void> {
		if (this._queryInfoMap.has(uri)) {
			return this._queryInfoMap.get(uri)!.queryRunner!.copyResults(selection, batchId, resultId, includeHeaders);
		}
	}

	public showCommitError(error: string): void {
		this._notificationService.notify({
			severity: Severity.Error,
			message: nls.localize('commitEditFailed', "Commit row failed: ") + error
		});
	}

	public isRunningQuery(uri: string): boolean {
		return !this._queryInfoMap.has(uri)
			? false
			: this._getQueryInfo(uri)!.queryRunner!.isExecuting;
	}

	/**
	 * Run a query for the given URI with the given text selection
	 */
	public async runQuery(uri: string, range: IRange, runOptions?: azdata.ExecutionPlanOptions): Promise<void> {
		return this.doRunQuery(uri, range, false, runOptions);
	}

	/**
	 * Run the current SQL statement for the given URI
	 */
	public async runQueryStatement(uri: string, range: IRange): Promise<void> {
		return this.doRunQuery(uri, range, true);
	}

	/**
	 * Run the current SQL statement for the given URI
	 */
	public async runQueryString(uri: string, selection: string): Promise<void> {
		return this.doRunQuery(uri, selection, true);
	}

	/**
	 * Run Query implementation
	 */
	private async doRunQuery(uri: string, range: IRange | string,
		runCurrentStatement: boolean, runOptions?: azdata.ExecutionPlanOptions): Promise<void> {
		// Reuse existing query runner if it exists
		let queryRunner: QueryRunner | undefined;
		let info: QueryInfo;

		if (this._queryInfoMap.has(uri)) {
			info = this._getQueryInfo(uri)!;
			let existingRunner: QueryRunner = info.queryRunner!;

			// If the query is already in progress, don't attempt to send it
			if (existingRunner.isExecuting) {
				return;
			}

			// If the query is not in progress, we can reuse the query runner
			queryRunner = existingRunner!;
			info.batchRanges = [];
			info.selectionSnippet = undefined;
		} else {
			// We do not have a query runner for this editor, so create a new one
			// and map it to the results uri
			info = this.initQueryRunner(uri);
			queryRunner = info.queryRunner!;
		}

		if (types.isString(range)) {
			// Run the query string in this case
			if (range.length < selectionSnippetMaxLen) {
				info.selectionSnippet = range;
			} else {
				info.selectionSnippet = range.substring(0, selectionSnippetMaxLen - 3) + '...';
			}
			return queryRunner.runQuery(range, runOptions);
		} else if (runCurrentStatement) {
			return queryRunner.runQueryStatement(range);
		} else {
			return queryRunner.runQuery(range, runOptions);
		}
	}

	private initQueryRunner(uri: string): QueryInfo {
		let queryRunner = this._instantiationService.createInstance(EditQueryRunner, uri);
		let info = new QueryInfo();
		queryRunner.onResultSet(e => {
			this._fireQueryEvent(queryRunner.uri, 'resultSet', e);
		});
		queryRunner.onBatchStart(b => {
			let link = undefined;
			let messageText = nls.localize('runQueryBatchStartMessage', "Started executing query at ");
			if (b.range) {
				if (info.selectionSnippet) {
					// This indicates it's a query string. Do not include line information since it'll be inaccurate, but show some of the
					// executed query text
					messageText = nls.localize('runQueryStringBatchStartMessage', "Started executing query \"{0}\"", info.selectionSnippet);
				} else {
					link = {
						text: strings.format(nls.localize('runQueryBatchStartLine', "Line {0}"), b.range.startLineNumber)
					};
				}
				info.batchRanges.push(b.range);
			}
			let message = {
				message: messageText,
				batchId: b.id,
				isError: false,
				time: new Date().toLocaleTimeString(),
				link: link
			};
			this._fireQueryEvent(queryRunner.uri, 'message', message);
		});
		queryRunner.onMessage(m => {
			this._fireQueryEvent(queryRunner.uri, 'message', m);
		});
		queryRunner.onQueryEnd(totalMilliseconds => {
			this._onRunQueryComplete.fire(queryRunner.uri);
			// fire extensibility API event
			let event: IQueryEvent = {
				type: 'queryStop',
				uri: queryRunner.uri,
				queryInfo:
				{
					batchRanges: info.batchRanges,
					messages: info.queryRunner!.messages
				}
			};
			this._onQueryEvent.fire(event);

			// fire UI event
			this._fireQueryEvent(queryRunner.uri, 'complete', totalMilliseconds);
		});
		queryRunner.onQueryStart(() => {
			this._onRunQueryStart.fire(queryRunner.uri);

			// fire extensibility API event
			let event: IQueryEvent = {
				type: 'queryStart',
				uri: queryRunner.uri,
				queryInfo:
				{
					batchRanges: info.batchRanges,
					messages: info.queryRunner!.messages
				}
			};
			this._onQueryEvent.fire(event);

			this._fireQueryEvent(queryRunner.uri, 'start');
		});
		queryRunner.onResultSetUpdate(() => {
			this._onRunQueryUpdate.fire(queryRunner.uri);

			let event: IQueryEvent = {
				type: 'queryUpdate',
				uri: queryRunner.uri,
				queryInfo:
				{
					batchRanges: info.batchRanges,
					messages: info.queryRunner!.messages
				}
			};
			this._onQueryEvent.fire(event);

			this._fireQueryEvent(queryRunner.uri, 'update');
		});

		queryRunner.onQueryPlanAvailable(planInfo => {
			// fire extensibility API event
			let event: IQueryEvent = {
				type: 'executionPlan',
				uri: planInfo.fileUri,
				queryInfo:
				{
					batchRanges: info.batchRanges,
					messages: info.queryRunner!.messages
				},
				params: planInfo
			};
			this._onQueryEvent.fire(event);
		});

		queryRunner.onExecutionPlanAvailable(qp2Info => {
			// fire extensibility API event
			let event: IQueryEvent = {
				type: 'executionPlan',
				uri: qp2Info.fileUri,
				queryInfo:
				{
					batchRanges: info.batchRanges,
					messages: info.queryRunner!.messages
				},
				params: qp2Info.planGraphs
			};
			this._onQueryEvent.fire(event);
		});

		queryRunner.onVisualize(resultSetInfo => {
			let event: IQueryEvent = {
				type: 'visualize',
				uri: queryRunner.uri,
				queryInfo:
				{
					batchRanges: info.batchRanges,
					messages: info.queryRunner!.messages
				},
				params: resultSetInfo
			};
			this._onQueryEvent.fire(event);
		});

		info.queryRunner = queryRunner;
		info.dataService = this._instantiationService.createInstance(DataService, queryRunner.uri);
		this._queryInfoMap.set(queryRunner.uri, info);
		return info;
	}

	public cancelQuery(input: QueryRunner | string): void {
		let queryRunner: QueryRunner | undefined;

		if (typeof input === 'string') {
			if (this._queryInfoMap.has(input)) {
				queryRunner = this._getQueryInfo(input)!.queryRunner;
			}
		} else {
			queryRunner = input;
		}

		if (queryRunner === undefined || !queryRunner.isExecuting) {
			// TODO: Cannot cancel query as no query is running.
			return;
		}

		// Switch the spinner to canceling, which will be reset when the query execute sends back its completed event
		// TODO indicate on the status bar that the query is being canceled

		// Cancel the query
		queryRunner.cancelQuery().then(success => undefined, error => {
			// On error, show error message and notify that the query is complete so that buttons and other status indicators
			// can be correct
			this._notificationService.notify({
				severity: Severity.Error,
				message: strings.format(nls.localize('msgCancelQueryFailed', "Canceling the query failed: {0}"), error)
			});
			this._fireQueryEvent(queryRunner!.uri, 'complete', 0);
		});

	}

	public async disposeQuery(ownerUri: string): Promise<void> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(ownerUri);
		if (queryRunner) {
			await queryRunner.disposeQuery();
		}
		// remove our info map
		if (this._queryInfoMap.has(ownerUri)) {
			this._queryInfoMap.delete(ownerUri);
		}
	}

	public async changeConnectionUri(newUri: string, oldUri: string): Promise<void> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(oldUri);
		if (!queryRunner) {
			// Nothing to do if we don't have a query runner currently (no connection)
			return;
		}
		else if (this._queryInfoMap.has(newUri)) {
			this._logService.error(`New URI '${newUri}' already has query info associated with it.`);
			throw new Error(nls.localize('queryModelService.uriAlreadyHasQuery', '{0} already has an existing query', newUri));
		}

		await queryRunner.changeConnectionUri(newUri, oldUri);

		// remove the old key and set new key with same query info as old uri. (Info existence is checked in internalGetQueryRunner)
		let info = this._queryInfoMap.get(oldUri);
		info.uri = newUri;
		this._queryInfoMap.set(newUri, info);
		this._queryInfoMap.delete(oldUri);
	}

	// EDIT DATA METHODS /////////////////////////////////////////////////////
	async initializeEdit(ownerUri: string, schemaName: string, objectName: string, objectType: string, rowLimit: number, queryString: string): Promise<void> {
		// Reuse existing query runner if it exists
		let queryRunner: EditQueryRunner;
		let info: QueryInfo;

		if (this._queryInfoMap.has(ownerUri)) {
			info = this._getQueryInfo(ownerUri)!;
			let existingRunner = info.queryRunner!;

			// If the initialization is already in progress
			if (existingRunner.isExecuting) {
				return;
			}

			queryRunner = existingRunner;
		} else {
			info = new QueryInfo();

			// We do not have a query runner for this editor, so create a new one
			// and map it to the results uri
			queryRunner = this._instantiationService.createInstance(EditQueryRunner, ownerUri);
			const resultSetEventType = 'resultSet';
			queryRunner.onResultSet(resultSet => {
				this._fireQueryEvent(ownerUri, resultSetEventType, resultSet);
			});
			queryRunner.onResultSetUpdate(resultSetSummary => {
				this._fireQueryEvent(ownerUri, resultSetEventType, resultSetSummary);
			});
			queryRunner.onBatchStart(batch => {
				let link = undefined;
				let messageText = nls.localize('runQueryBatchStartMessage', "Started executing query at ");
				if (batch.range) {
					if (info.selectionSnippet) {
						// This indicates it's a query string. Do not include line information since it'll be inaccurate, but show some of the
						// executed query text
						messageText = nls.localize('runQueryStringBatchStartMessage', "Started executing query \"{0}\"", info.selectionSnippet);
					} else {
						link = {
							text: strings.format(nls.localize('runQueryBatchStartLine', "Line {0}"), batch.range.startLineNumber)
						};
					}
				}
				let message = {
					message: messageText,
					batchId: batch.id,
					isError: false,
					time: new Date().toLocaleTimeString(),
					link: link
				};
				this._fireQueryEvent(ownerUri, 'message', message);
			});
			queryRunner.onMessage(message => {
				this._fireQueryEvent(ownerUri, 'message', message);
			});
			queryRunner.onQueryEnd(totalMilliseconds => {
				this._onRunQueryComplete.fire(ownerUri);
				// fire extensibility API event
				let event: IQueryEvent = {
					type: 'queryStop',
					uri: ownerUri,
					queryInfo:
					{
						batchRanges: info.batchRanges,
						messages: info.queryRunner!.messages
					},
				};
				this._onQueryEvent.fire(event);

				// fire UI event
				this._fireQueryEvent(ownerUri, 'complete', totalMilliseconds);
			});
			queryRunner.onQueryStart(() => {
				this._onRunQueryStart.fire(ownerUri);
				// fire extensibility API event
				let event: IQueryEvent = {
					type: 'queryStart',
					uri: ownerUri,
					queryInfo:
					{
						batchRanges: info.batchRanges,
						messages: info.queryRunner!.messages
					},
				};
				this._onQueryEvent.fire(event);

				// fire UI event
				this._fireQueryEvent(ownerUri, 'start');
			});
			queryRunner.onEditSessionReady(e => {
				this._onEditSessionReady.fire(e);
				this._fireQueryEvent(e.ownerUri, 'editSessionReady');
			});

			info.queryRunner = queryRunner;
			info.dataService = this._instantiationService.createInstance(DataService, ownerUri);
			this._queryInfoMap.set(ownerUri, info);
		}

		if (queryString) {
			if (queryString.length < selectionSnippetMaxLen) {
				info.selectionSnippet = queryString;
			} else {
				info.selectionSnippet = queryString.substring(0, selectionSnippetMaxLen - 3) + '...';
			}
		}

		return queryRunner.initializeEdit(ownerUri, schemaName, objectName, objectType, rowLimit, queryString);
	}

	public cancelInitializeEdit(input: QueryRunner | string): void {
		// TODO: Implement query cancellation service
	}

	public disposeEdit(ownerUri: string): Promise<void> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(ownerUri);
		if (queryRunner) {
			return queryRunner.disposeEdit(ownerUri);
		}
		return Promise.resolve();
	}

	public updateCell(ownerUri: string, rowId: number, columnId: number, newValue: string): Promise<azdata.EditUpdateCellResult | undefined> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(ownerUri);
		if (queryRunner) {
			return queryRunner.updateCell(ownerUri, rowId, columnId, newValue).then((result) => result, error => {
				this._notificationService.notify({
					severity: Severity.Error,
					message: nls.localize('updateCellFailed', "Update cell failed: ") + error.message
				});
				return Promise.reject(error);
			});
		}
		return Promise.resolve(undefined);
	}

	public commitEdit(ownerUri: string): Promise<void> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(ownerUri);
		if (queryRunner) {
			return queryRunner.commitEdit(ownerUri).then(() => { }, error => {
				this._notificationService.notify({
					severity: Severity.Error,
					message: nls.localize('commitEditFailed', "Commit row failed: ") + error.message
				});
				return Promise.reject(error);
			});
		}
		return Promise.resolve();
	}

	public createRow(ownerUri: string): Promise<azdata.EditCreateRowResult | undefined> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(ownerUri);
		if (queryRunner) {
			return queryRunner.createRow(ownerUri);
		}
		return Promise.resolve(undefined);
	}

	public deleteRow(ownerUri: string, rowId: number): Promise<void> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(ownerUri);
		if (queryRunner) {
			return queryRunner.deleteRow(ownerUri, rowId);
		}
		return Promise.resolve();
	}

	public revertCell(ownerUri: string, rowId: number, columnId: number): Promise<azdata.EditRevertCellResult | undefined> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(ownerUri);
		if (queryRunner) {
			return queryRunner.revertCell(ownerUri, rowId, columnId);
		}
		return Promise.resolve(undefined);
	}

	public revertRow(ownerUri: string, rowId: number): Promise<void> {
		// Get existing query runner
		let queryRunner = this.internalGetQueryRunner(ownerUri);
		if (queryRunner) {
			return queryRunner.revertRow(ownerUri, rowId);
		}
		return Promise.resolve();
	}

	public getQueryRunner(ownerUri: string): QueryRunner | undefined {
		let queryRunner: QueryRunner | undefined = undefined;
		if (this._queryInfoMap.has(ownerUri)) {
			queryRunner = this._getQueryInfo(ownerUri)!.queryRunner;
		}
		// return undefined if not found or is already executing
		return queryRunner;
	}

	// PRIVATE METHODS //////////////////////////////////////////////////////

	private internalGetQueryRunner(ownerUri: string): EditQueryRunner | undefined {
		let queryRunner: EditQueryRunner | undefined;
		if (this._queryInfoMap.has(ownerUri)) {
			let existingRunner = this._getQueryInfo(ownerUri)!.queryRunner!;
			// If the query is not already executing then set it up
			if (!existingRunner.isExecuting) {
				queryRunner = this._getQueryInfo(ownerUri)!.queryRunner!;
			}
		}
		// return undefined if not found or is already executing
		return queryRunner;
	}

	private _fireGridContentEvent(uri: string, type: string): void {
		let info = this._getQueryInfo(uri);

		if (info && info.dataServiceReady) {
			let service: DataService = this.getDataService(uri);
			if (service) {
				// There is no need to queue up these events like there is for the query events because
				// if the DataService is not yet ready there will be no grid content to update
				service.fireGridContent(type);
			}
		}
	}

	private _fireQueryEvent(uri: string, type: string, data?: any) {
		let info = this._getQueryInfo(uri);

		if (info && info.dataServiceReady) {
			let service: DataService = this.getDataService(uri);
			service.fireQueryEvent({
				type: type,
				data: data
			});
		} else if (info) {
			let queueItem: QueryEvent = { type: type, data: data };
			info.queryEventQueue!.push(queueItem);
		}
	}

	private _sendQueuedEvents(uri: string): void {
		let info = this._getQueryInfo(uri);
		while (info && info.queryEventQueue!.length > 0) {
			let event = info.queryEventQueue!.shift()!;
			this._fireQueryEvent(uri, event.type, event.data);
		}
	}

	public _getQueryInfo(uri: string): QueryInfo | undefined {
		return this._queryInfoMap.get(uri);
	}
}