Fix breaks introduced in the refactoring merge.

src/ServiceHost/Hosting/ServiceHost.cs

@@ -45,7 +45,13 @@ namespace Microsoft.SqlTools.ServiceLayer.Hosting
             // Initialize the shutdown activities
             shutdownCallbacks = new List<ShutdownCallback>();
             initializeCallbacks = new List<InitializeCallback>();
+        }
 
+        /// <summary>
+        /// Provide initialization that must occur after the service host is started
+        /// </summary>
+        public void Initialize()
+        {
             // Register the requests that this service host will handle
             this.SetRequestHandler(InitializeRequest.Type, this.HandleInitializeRequest);
             this.SetRequestHandler(ShutdownRequest.Type, this.HandleShutdownRequest);

src/ServiceHost/LanguageServices/LanguageService.cs

@@ -17,6 +17,7 @@ using Microsoft.SqlTools.ServiceLayer.WorkspaceServices.Contracts;
 using System.Linq;
 using Microsoft.SqlServer.Management.SqlParser.Parser;
 using Location = Microsoft.SqlTools.ServiceLayer.WorkspaceServices.Contracts.Location;
+using Microsoft.SqlTools.ServiceLayer.Connection;
 
 namespace Microsoft.SqlTools.ServiceLayer.LanguageServices
 {
@@ -97,6 +98,15 @@ namespace Microsoft.SqlTools.ServiceLayer.LanguageServices
             // Register the configuration update handler
             WorkspaceService<SqlToolsSettings>.Instance.RegisterConfigChangeCallback(HandleDidChangeConfigurationNotification);
 
+            // Register the file change update handler
+            WorkspaceService<SqlToolsSettings>.Instance.RegisterTextDocChangeCallback(HandleDidChangeTextDocumentNotification);
+
+            // Register the file open update handler
+            WorkspaceService<SqlToolsSettings>.Instance.RegisterTextDocOpenCallback(HandleDidOpenTextDocumentNotification);
+
+            // register an OnConnection callback 
+            ConnectionService.Instance.RegisterOnConnectionTask(OnConnection);      
+
             // Store the SqlToolsContext for future use
             Context = context;
         }
@@ -165,8 +175,11 @@ namespace Microsoft.SqlTools.ServiceLayer.LanguageServices
             TextDocumentPosition textDocumentPosition,
             RequestContext<CompletionItem[]> requestContext)
         {
-            Logger.Write(LogLevel.Verbose, "HandleCompletionRequest");
+            Logger.Write(LogLevel.Verbose, "HandleCompletionRequest"); 
-            await Task.FromResult(true);
+ 
+            // get the current list of completion items and return to client 
+            var completionItems = AutoCompleteService.Instance.GetCompletionItems(textDocumentPosition); 
+            await requestContext.SendResult(completionItems); 
         }
 
         private static async Task HandleCompletionResolveRequest(
@@ -221,6 +234,45 @@ namespace Microsoft.SqlTools.ServiceLayer.LanguageServices
 
         #region Handlers for Events from Other Services
 
+        /// <summary>
+        /// Handle the file open notification 
+        /// </summary>
+        /// <param name="scriptFile"></param>
+        /// <param name="eventContext"></param>
+        /// <returns></returns>
+        public async Task HandleDidOpenTextDocumentNotification(
+            ScriptFile scriptFile, 
+            EventContext eventContext)
+        {
+            await this.RunScriptDiagnostics( 
+                new ScriptFile[] { scriptFile },
+                eventContext); 
+
+            await Task.FromResult(true);             
+        }
+
+        
+        /// <summary> 
+        /// Handles text document change events 
+        /// </summary> 
+        /// <param name="textChangeParams"></param> 
+        /// <param name="eventContext"></param> 
+        /// <returns></returns> 
+        public async Task HandleDidChangeTextDocumentNotification(ScriptFile[] changedFiles, EventContext eventContext) 
+        { 
+            await this.RunScriptDiagnostics( 
+                changedFiles.ToArray(), 
+                eventContext); 
+
+            await Task.FromResult(true); 
+        }
+
+        /// <summary>
+        /// Handle the file configuration change notification
+        /// </summary>
+        /// <param name="newSettings"></param>
+        /// <param name="oldSettings"></param>
+        /// <param name="eventContext"></param>
         public async Task HandleDidChangeConfigurationNotification(
             SqlToolsSettings newSettings, 
             SqlToolsSettings oldSettings, 
@@ -251,7 +303,17 @@ namespace Microsoft.SqlTools.ServiceLayer.LanguageServices
             CurrentSettings.EnableProfileLoading = newSettings.EnableProfileLoading;
             CurrentSettings.ScriptAnalysis.Update(newSettings.ScriptAnalysis, CurrentWorkspace.WorkspacePath);
         }
-
+        
+        /// <summary> 
+        /// Callback for when a user connection is done processing 
+        /// </summary> 
+        /// <param name="sqlConnection"></param> 
+        public async Task OnConnection(ISqlConnection sqlConnection) 
+        { 
+            await AutoCompleteService.Instance.UpdateAutoCompleteCache(sqlConnection); 
+            await Task.FromResult(true); 
+        } 
+        
         #endregion
 
         #region Private Helpers

src/ServiceHost/Program.cs

@@ -7,6 +7,7 @@ using Microsoft.SqlTools.ServiceLayer.Hosting;
 using Microsoft.SqlTools.ServiceLayer.SqlContext;
 using Microsoft.SqlTools.ServiceLayer.WorkspaceServices;
 using Microsoft.SqlTools.ServiceLayer.LanguageServices;
+using Microsoft.SqlTools.ServiceLayer.Connection;
 
 namespace Microsoft.SqlTools.ServiceLayer
 {     
@@ -37,13 +38,16 @@ namespace Microsoft.SqlTools.ServiceLayer
             // Grab the instance of the service host
             ServiceHost serviceHost = ServiceHost.Instance;
 
+            // Start the service
+            serviceHost.Start().Wait();
+
             // Initialize the services that will be hosted here
             WorkspaceService<SqlToolsSettings>.Instance.InitializeService(serviceHost);
             AutoCompleteService.Instance.InitializeService(serviceHost);
             LanguageService.Instance.InitializeService(serviceHost, sqlToolsContext);
+            ConnectionService.Instance.Initialize(serviceHost);
 
-            // Start the service
+            serviceHost.Initialize();
-            serviceHost.Start().Wait();
             serviceHost.WaitForExit();
         }
     }

src/ServiceHost/WorkspaceServices/WorkspaceService.cs

@@ -43,6 +43,7 @@ namespace Microsoft.SqlTools.ServiceLayer.WorkspaceServices
         {
             ConfigChangeCallbacks = new List<ConfigChangeCallback>();
             TextDocChangeCallbacks = new List<TextDocChangeCallback>();
+            TextDocOpenCallbacks = new List<TextDocOpenCallback>();
         }
 
         #endregion
@@ -69,6 +70,13 @@ namespace Microsoft.SqlTools.ServiceLayer.WorkspaceServices
         /// <param name="eventContext">Context of the event raised for the changed files</param>
         public delegate Task TextDocChangeCallback(ScriptFile[] changedFiles, EventContext eventContext);
 
+        /// <summary>
+        /// Delegate for callbacks that occur when a text document is opened
+        /// </summary>
+        /// <param name="openFile">File that was opened</param>
+        /// <param name="eventContext">Context of the event raised for the changed files</param>
+        public delegate Task TextDocOpenCallback(ScriptFile openFile, EventContext eventContext);
+        
         /// <summary>
         /// List of callbacks to call when the configuration of the workspace changes
         /// </summary>
@@ -79,6 +87,12 @@ namespace Microsoft.SqlTools.ServiceLayer.WorkspaceServices
         /// </summary>
         private List<TextDocChangeCallback> TextDocChangeCallbacks { get; set; }
 
+        /// <summary>
+        /// List of callbacks to call when a text document is opened
+        /// </summary>
+        private List<TextDocOpenCallback> TextDocOpenCallbacks { get; set; }
+
+
         #endregion
 
         #region Public Methods
@@ -140,6 +154,15 @@ namespace Microsoft.SqlTools.ServiceLayer.WorkspaceServices
             TextDocChangeCallbacks.Add(task);
         }
 
+        /// <summary>
+        /// Adds a new task to be called when a file is opened
+        /// </summary>
+        /// <param name="task">Delegate to call when a document is opened</param>
+        public void RegisterTextDocOpenCallback(TextDocOpenCallback task)
+        {
+            TextDocOpenCallbacks.Add(task);
+        }        
+
         #endregion
 
         #region Event Handlers
@@ -158,7 +181,7 @@ namespace Microsoft.SqlTools.ServiceLayer.WorkspaceServices
             // A text change notification can batch multiple change requests
             foreach (var textChange in textChangeParams.ContentChanges)
             {
-                string fileUri = textChangeParams.TextDocument.Uri;
+                string fileUri = textChangeParams.Uri ?? textChangeParams.TextDocument.Uri; 
                 msg.AppendLine(String.Format("  File: {0}", fileUri));
 
                 ScriptFile changedFile = Workspace.GetFile(fileUri);
@@ -177,12 +200,20 @@ namespace Microsoft.SqlTools.ServiceLayer.WorkspaceServices
             return Task.WhenAll(handlers);
         }
 
-        protected Task HandleDidOpenTextDocumentNotification(
+        protected async Task HandleDidOpenTextDocumentNotification(
             DidOpenTextDocumentNotification openParams,
             EventContext eventContext)
         {
             Logger.Write(LogLevel.Verbose, "HandleDidOpenTextDocumentNotification");
-            return Task.FromResult(true);
+
+            // read the SQL file contents into the ScriptFile 
+            ScriptFile openedFile = Workspace.GetFileBuffer(openParams.Uri, openParams.Text); 
+
+             // Propagate the changes to the event handlers
+            var textDocOpenTasks = TextDocOpenCallbacks.Select(
+                t => t(openedFile, eventContext));
+
+            await Task.WhenAll(textDocOpenTasks);
         }
 
         protected Task HandleDidCloseTextDocumentNotification(