Add back Notebook Completion List IntelliSense (#3520)

* Static notebook intellisense working

* Intellisense dynamic cells working

* Remove launch.json erroneous change

* PR comments #1

* PR feedback to change a minor condition

src/sql/workbench/api/node/mainThreadNotebookDocumentsAndEditors.ts

@@ -5,9 +5,11 @@
 'use strict';
 
 import * as sqlops from 'sqlops';
+import * as util from 'util';
 import { extHostNamedCustomer } from 'vs/workbench/api/electron-browser/extHostCustomers';
-import { Disposable } from 'vs/base/common/lifecycle';
+import { Disposable, IDisposable } from 'vs/base/common/lifecycle';
 import URI, { UriComponents } from 'vs/base/common/uri';
+import { Event, Emitter } from 'vs/base/common/event';
 import { IExtHostContext, IUndoStopOptions } from 'vs/workbench/api/node/extHost.protocol';
 import { IEditorService } from 'vs/workbench/services/editor/common/editorService';
 import { IEditorGroupsService } from 'vs/workbench/services/group/common/editorGroupsService';
@@ -18,7 +20,7 @@ import { Schemas } from 'vs/base/common/network';
 
 import {
 	SqlMainContext, MainThreadNotebookDocumentsAndEditorsShape, SqlExtHostContext, ExtHostNotebookDocumentsAndEditorsShape,
-	INotebookDocumentsAndEditorsDelta, INotebookEditorAddData, INotebookShowOptions, INotebookModelAddedData
+	INotebookDocumentsAndEditorsDelta, INotebookEditorAddData, INotebookShowOptions, INotebookModelAddedData, INotebookModelChangedData
 } from 'sql/workbench/api/node/sqlExtHost.protocol';
 import { NotebookInputModel, NotebookInput } from 'sql/parts/notebook/notebookInput';
 import { INotebookService, INotebookEditor } from 'sql/services/notebook/notebookService';
@@ -26,11 +28,17 @@ import { TPromise } from 'vs/base/common/winjs.base';
 import { getProviderForFileName } from 'sql/parts/notebook/notebookUtils';
 import { ISingleNotebookEditOperation } from 'sql/workbench/api/common/sqlExtHostTypes';
 import { disposed } from 'vs/base/common/errors';
+import { ICellModel, NotebookContentChange } from 'sql/parts/notebook/models/modelInterfaces';
 
 class MainThreadNotebookEditor extends Disposable {
+	private _contentChangedEmitter = new Emitter<NotebookContentChange>();
+	public readonly contentChanged: Event<NotebookContentChange> = this._contentChangedEmitter.event;
 
 	constructor(public readonly editor: INotebookEditor) {
 		super();
+		editor.modelReady.then(model => {
+			this._register(model.contentChanged((e) => this._contentChangedEmitter.fire(e)));
+		});
 	}
 
 	public get uri(): URI {
@@ -49,6 +57,10 @@ class MainThreadNotebookEditor extends Disposable {
 		return this.editor.notebookParams.providerId;
 	}
 
+	public get cells(): ICellModel[] {
+		return this.editor.cells;
+	}
+
 	public save(): Thenable<boolean> {
 		return this.editor.save();
 	}
@@ -253,7 +265,7 @@ export class MainThreadNotebookDocumentsAndEditors extends Disposable implements
 
 	private _proxy: ExtHostNotebookDocumentsAndEditorsShape;
 	private _notebookEditors = new Map<string, MainThreadNotebookEditor>();
-
+	private _modelToDisposeMap = new Map<string, IDisposable>();
 	constructor(
 		extHostContext: IExtHostContext,
 		@IInstantiationService private _instantiationService: IInstantiationService,
@@ -398,8 +410,34 @@ export class MainThreadNotebookDocumentsAndEditors extends Disposable implements
 
 		if (!empty) {
 			this._proxy.$acceptDocumentsAndEditorsDelta(extHostDelta);
+			this.processRemovedDocs(removedDocuments);
+			this.processAddedDocs(addedEditors);
 		}
 	}
+	processRemovedDocs(removedDocuments: URI[]): void {
+		if (!removedDocuments) {
+			return;
+		}
+		removedDocuments.forEach(removedDoc => {
+			let listener = this._modelToDisposeMap.get(removedDoc.toString());
+			if (listener) {
+				listener.dispose();
+				this._modelToDisposeMap.delete(removedDoc.toString());
+			}
+		});
+	}
+
+	processAddedDocs(addedEditors: MainThreadNotebookEditor[]): any {
+		if (!addedEditors) {
+			return;
+		}
+		addedEditors.forEach(editor => {
+			let modelUrl = editor.uri;
+			this._modelToDisposeMap.set(editor.uri.toString(), editor.contentChanged((e) => {
+				this._proxy.$acceptModelChanged(modelUrl, this._toNotebookChangeData(e, editor));
+			}));
+		});
+	}
 
 	private _toNotebookEditorAddData(editor: MainThreadNotebookEditor): INotebookEditorAddData {
 		let addData: INotebookEditorAddData = {
@@ -414,8 +452,54 @@ export class MainThreadNotebookDocumentsAndEditors extends Disposable implements
 		let addData: INotebookModelAddedData = {
 			uri: editor.uri,
 			isDirty: editor.isDirty,
-			providerId: editor.providerId
+			providerId: editor.providerId,
+			cells: this.convertCellModelToNotebookCell(editor.cells)
 		};
 		return addData;
 	}
+
+	private _toNotebookChangeData(e: NotebookContentChange, editor: MainThreadNotebookEditor): INotebookModelChangedData {
+		let changeData: INotebookModelChangedData = {
+			// Note: we just send all cells for now, not a diff
+			cells: this.convertCellModelToNotebookCell(editor.cells),
+			isDirty: e.isDirty,
+			providerId: editor.providerId,
+			uri: editor.uri
+		};
+		return changeData;
+	}
+
+	private convertCellModelToNotebookCell(cells: ICellModel | ICellModel[]): sqlops.nb.NotebookCell[] {
+		let notebookCells: sqlops.nb.NotebookCell[] = [];
+		if (Array.isArray(cells)) {
+			for (let cell of cells) {
+				notebookCells.push({
+					uri: cell.cellUri,
+					contents: {
+						cell_type: cell.cellType,
+						execution_count: undefined,
+						metadata: {
+							language: cell.language
+						},
+						source: undefined
+
+					}
+				});
+			}
+		}
+		else {
+			notebookCells.push({
+				uri: cells.cellUri,
+				contents: {
+					cell_type: cells.cellType,
+					execution_count: undefined,
+					metadata: {
+						language: cells.language
+					},
+					source: undefined
+				}
+			});
+		}
+		return notebookCells;
+	}
 }