Add -NoColor switch parameter to control colored output

Co-authored-by: HeyItsGilbert <[email protected]>

Author: Copilot

ChocoLogParse/ChocoLog.format.ps1xml

@@ -36,12 +36,16 @@
               </TableColumnItem>
               <TableColumnItem>
                 <ScriptBlock>
-                $style = if ($_.exitCode -eq 0) {
-                  $PSStyle.Foreground.White
+                if ($script:ChocoLogNoColor) {
+                  [String]::Format("{0}", $_.exitCode)
                 } else {
-                  $PSStyle.Foreground.Yellow
+                  $style = if ($_.exitCode -eq 0) {
+                    $PSStyle.Foreground.White
+                  } else {
+                    $PSStyle.Foreground.Yellow
+                  }
+                  [String]::Format("{0}{1}{2}", $style, $_.exitCode, $PSStyle.Reset)
                 }
-                [String]::Format("{0}{1}{2}", $bg, $_.exitCode, $PSStyle.Reset)
                 </ScriptBlock>
               </TableColumnItem>
               <TableColumnItem>

ChocoLogParse/ChocoLogParse.psm1

@@ -10,6 +10,9 @@ foreach ($import in @($enums + $classes + $public )) {
   }
 }
 
+# Initialize module variable for color control (default to colored output)
+$script:ChocoLogNoColor = $false
+
 # Add our custom formatter that needed classes first
 $format = Join-Path -Path $PSScriptRoot -ChildPath 'ChocoLog.format.ps1xml'
 Update-FormatData -PrependPath $format

ChocoLogParse/Public/Get-ChocoLogEntry.ps1

@@ -9,6 +9,10 @@
   Get-ChocoLogEntry
 
   Grabs the laste entry from the latest log
+.EXAMPLE
+  Get-ChocoLogEntry -NoColor
+
+  Grabs the latest entry from the latest log without colored output
 .PARAMETER Report
   This changes the output to be more friendly for reporting
 .PARAMETER Path
@@ -20,6 +24,9 @@
   The log4net pattern layout used to parse the log. It is very unlikely that you
   need to supply this. The code expects pattern names: time, session, level, and
   message.
+.PARAMETER NoColor
+  Disables colored output in the formatter. When specified, the output will be
+  displayed without ANSI color codes.
 #>
 function Get-ChocoLogEntry {
   [CmdletBinding()]
@@ -36,15 +43,16 @@ function Get-ChocoLogEntry {
     $Filter = 'chocolatey*.log',
     [string]
     $PatternLayout = '%date %thread [%-5level] - %message',
-    [switch]$Report
+    [switch]$Report,
+    [switch]$NoColor
   )
   # ToDo:
   # - Support searching for a cli entry
   # - sub command type (e.g. search, list, upgrade)
   # - Exit code
   # - Filter to recent time to make sure we get the right one
 
-  $entry = Read-ChocoLog -FileLimit 1 -Path $Path -Filter $Filter -PatternLayout $PatternLayout | Select-Object -Last 1
+  $entry = Read-ChocoLog -FileLimit 1 -Path $Path -Filter $Filter -PatternLayout $PatternLayout -NoColor:$NoColor | Select-Object -Last 1
   if ($report) {
     # Print out in a format that's useful for Chef logging
     Write-Host ('Command: {0}' -F $entry.cli)

ChocoLogParse/Public/Read-ChocoLog.ps1

@@ -12,6 +12,10 @@
   Read-ChocoLog
 
   This will read the latest Chocolatey.log on the machine.
+.EXAMPLE
+  Read-ChocoLog -NoColor
+
+  This will read the latest Chocolatey.log on the machine without colored output.
 .PARAMETER Path
   The log path you want to parse. This will default to the latest local log.
   This can be a directory of logs.
@@ -23,6 +27,9 @@
   The log4net pattern layout used to parse the log. It is very unlikely that you
   need to supply this. The code expects pattern names: time, session, level, and
   message.
+.PARAMETER NoColor
+  Disables colored output in the formatter. When specified, the output will be
+  displayed without ANSI color codes.
 #>
 function Read-ChocoLog {
   # This makes PlatyPS sad.
@@ -41,8 +48,14 @@ function Read-ChocoLog {
     [String]
     $Filter = 'chocolatey*.log',
     [string]
-    $PatternLayout = '%date %thread [%-5level] - %message'
+    $PatternLayout = '%date %thread [%-5level] - %message',
+    [switch]
+    $NoColor
   )
+  
+  # Set module-level variable to control coloring in formatter
+  $script:ChocoLogNoColor = $NoColor.IsPresent
+  
   $files = Get-Item -Path $Path
   if ($files.PSIsContainer) {
     $files = Get-ChildItem -Path $Path -Filter $Filter |