Set exit code of Invoke-ScriptAnalyzer -EnableExit to total number of diagnostics

Author: MatejKafka

Engine/Commands/InvokeScriptAnalyzerCommand.cs

@@ -34,6 +34,7 @@ public class InvokeScriptAnalyzerCommand : PSCmdlet, IOutputWriter
 
         #region Private variables
         List<string> processedPaths;
+        private int totalDiagnosticCount = 0;
         #endregion // Private variables
 
         #region Parameters
@@ -412,6 +413,10 @@ protected override void EndProcessing()
         {
             ScriptAnalyzer.Instance.CleanUp();
             base.EndProcessing();
+
+            if (EnableExit) {
+                this.Host.SetShouldExit(totalDiagnosticCount);
+            }
         }
 
         protected override void StopProcessing()
@@ -426,7 +431,10 @@ protected override void StopProcessing()
 
         private void ProcessInput()
         {
-            WriteToOutput(RunAnalysis());
+            // collect to array to avoid double enumeration
+            var diagnosticRecords = RunAnalysis().ToArray();
+            WriteToOutput(diagnosticRecords);
+            totalDiagnosticCount += diagnosticRecords.Length;
         }
 
         private IEnumerable<DiagnosticRecord> RunAnalysis()
@@ -454,7 +462,7 @@ private IEnumerable<DiagnosticRecord> RunAnalysis()
             return diagnostics;
         }
 
-        private void WriteToOutput(IEnumerable<DiagnosticRecord> diagnosticRecords)
+        private void WriteToOutput(DiagnosticRecord[] diagnosticRecords)
         {
             foreach (ILogger logger in ScriptAnalyzer.Instance.Loggers)
             {
@@ -507,11 +515,6 @@ private void WriteToOutput(IEnumerable<DiagnosticRecord> diagnosticRecords)
                     }
                 }
             }
-
-            if (EnableExit.IsPresent)
-            {
-                this.Host.SetShouldExit(diagnosticRecords.Count());
-            }
         }
 
         private void ProcessPath()
@@ -535,4 +538,4 @@ private bool OverrideSwitchParam(bool paramValue, string paramName)
 
         #endregion // Private Methods
     }
-}
+}