Create PublishScriptDiagnostics wrapper method

Author: Kapil Borle

src/PowerShellEditorServices.Protocol/Server/LanguageServer.cs

@@ -187,11 +187,13 @@ private async Task HandleSetPSSARulesRequest(
             object param,
             RequestContext<object> requestContext)
         {
-            if (editorSession.AnalysisService != null)
+            var dynParams = param as dynamic;
+            if (editorSession.AnalysisService != null &&
+                    editorSession.AnalysisService.SettingsPath == null)
             {
                 var activeRules = new List<string>();
-                var dynParam = param as dynamic;
-                foreach (dynamic ruleInfo in dynParam)
+                var ruleInfos = dynParams.ruleInfos;
+                foreach (dynamic ruleInfo in ruleInfos)
                 {
                     if ((Boolean) ruleInfo.isEnabled)
                     {
@@ -201,17 +203,30 @@ private async Task HandleSetPSSARulesRequest(
                 editorSession.AnalysisService.ActiveRules = activeRules.ToArray();
             }
 
-            await requestContext.SendResult(null);
+            var sendresult = requestContext.SendResult(null);
+            // send the file uri from the client
+            // retrieve the file
+            // requestcontext is the eventcontext
+            var scripFile = editorSession.Workspace.GetFile((string)dynParams.filepath);
+            await RunScriptDiagnostics(
+                    new ScriptFile[] { scripFile },
+                        editorSession,
+                        requestContext.SendEvent);
+
+            await sendresult;
+
+            //await requestContext.SendResult(null);
         }
 
         private async Task HandleGetPSSARulesRequest(
             object param,
             RequestContext<object> requestContext)
         {
-            List<string> ruleList = new List<string>();
-            List<object> rules = new List<object>();
-            if (editorSession.AnalysisService != null)
+            List<object> rules = null;
+            if (editorSession.AnalysisService != null
+                    && editorSession.AnalysisService.SettingsPath == null)
             {
+                rules = new List<object>();
                 var ruleNames = editorSession.AnalysisService.GetPSScriptAnalyzerRules();
                 var activeRules = editorSession.AnalysisService.ActiveRules;
                 foreach (var ruleName in ruleNames)
@@ -1007,6 +1022,14 @@ private Task RunScriptDiagnostics(
             ScriptFile[] filesToAnalyze,
             EditorSession editorSession,
             EventContext eventContext)
+        {
+            return RunScriptDiagnostics(filesToAnalyze, editorSession, eventContext.SendEvent);
+        }
+
+        private Task RunScriptDiagnostics(
+            ScriptFile[] filesToAnalyze,
+            EditorSession editorSession,
+            Func<EventType<PublishDiagnosticsNotification>, PublishDiagnosticsNotification, Task> eventSender)
         {
             if (!this.currentSettings.ScriptAnalysis.Enable.Value)
             {
@@ -1054,7 +1077,7 @@ private Task RunScriptDiagnostics(
                         filesToAnalyze,
                         this.codeActionsPerFile,
                         editorSession,
-                        eventContext,
+                        eventSender,
                         existingRequestCancellation.Token),
                 CancellationToken.None,
                 TaskCreationOptions.None,
@@ -1063,13 +1086,32 @@ private Task RunScriptDiagnostics(
             return Task.FromResult(true);
         }
 
+
         private static async Task DelayThenInvokeDiagnostics(
             int delayMilliseconds,
             ScriptFile[] filesToAnalyze,
             Dictionary<string, Dictionary<string, MarkerCorrection>> correctionIndex,
             EditorSession editorSession,
             EventContext eventContext,
             CancellationToken cancellationToken)
+        {
+            await DelayThenInvokeDiagnostics(
+                delayMilliseconds,
+                filesToAnalyze,
+                correctionIndex,
+                editorSession,
+                eventContext.SendEvent,
+                cancellationToken);
+        }
+
+
+        private static async Task DelayThenInvokeDiagnostics(
+            int delayMilliseconds,
+            ScriptFile[] filesToAnalyze,
+            Dictionary<string, Dictionary<string, MarkerCorrection>> correctionIndex,
+            EditorSession editorSession,
+            Func<EventType<PublishDiagnosticsNotification>, PublishDiagnosticsNotification, Task> eventSender,
+            CancellationToken cancellationToken)
         {
             // First of all, wait for the desired delay period before
             // analyzing the provided list of files
@@ -1115,7 +1157,7 @@ await PublishScriptDiagnostics(
                     scriptFile,
                     semanticMarkers,
                     correctionIndex,
-                    eventContext);
+                    eventSender);
             }
         }
 
@@ -1124,6 +1166,19 @@ private static async Task PublishScriptDiagnostics(
             ScriptFileMarker[] semanticMarkers,
             Dictionary<string, Dictionary<string, MarkerCorrection>> correctionIndex,
             EventContext eventContext)
+        {
+            await PublishScriptDiagnostics(
+                scriptFile,
+                semanticMarkers,
+                correctionIndex,
+                eventContext.SendEvent);
+        }
+
+        private static async Task PublishScriptDiagnostics(
+            ScriptFile scriptFile,
+            ScriptFileMarker[] semanticMarkers,
+            Dictionary<string, Dictionary<string, MarkerCorrection>> correctionIndex,
+            Func<EventType<PublishDiagnosticsNotification>, PublishDiagnosticsNotification, Task> eventSender)
         {
             List<Diagnostic> diagnostics = new List<Diagnostic>();
 
@@ -1147,7 +1202,7 @@ private static async Task PublishScriptDiagnostics(
 
             // Always send syntax and semantic errors.  We want to
             // make sure no out-of-date markers are being displayed.
-            await eventContext.SendEvent(
+            await eventSender(
                 PublishDiagnosticsNotification.Type,
                 new PublishDiagnosticsNotification
                 {