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
     }
-}
+}

Tests/Engine/InvokeScriptAnalyzer.tests.ps1

@@ -561,9 +561,29 @@ Describe "Test -EnableExit Switch" {
 
         $pssaPath = (Get-Module PSScriptAnalyzer).Path
 
-        & $pwshExe -Command "Import-Module '$pssaPath'; Invoke-ScriptAnalyzer -ScriptDefinition gci -EnableExit"
+        & $pwshExe -NoProfile -Command "Import-Module '$pssaPath'; Invoke-ScriptAnalyzer -ScriptDefinition gci -EnableExit"
 
-        $LASTEXITCODE  | Should -Be 1
+        $LASTEXITCODE | Should -Be 1
+    }
+
+    It "Returns exit code equivalent to number of warnings for multiple piped files" {
+        if ($IsCoreCLR)
+        {
+            $pwshExe = (Get-Process -Id $PID).Path
+        }
+        else
+        {
+            $pwshExe = 'powershell'
+        }
+
+        $pssaPath = (Get-Module PSScriptAnalyzer).Path
+
+        & $pwshExe -NoProfile {
+            Import-Module $Args[0]
+            Get-ChildItem $Args[1] | Invoke-ScriptAnalyzer -EnableExit
+        } -Args $pssaPath, "$PSScriptRoot\RecursionDirectoryTest"
+
+        $LASTEXITCODE | Should -Be 2
     }
 
     Describe "-ReportSummary switch" {