PSMultiLogPlus.psm1
|
<#
.DESCRIPTION A module that provides a single entry point to output a log to multiple sources. .EXAMPLE C:\PS> Import-Module PSMultiLog C:\PS> Start-FileLog -FilePath c:\ps\info.log C:\PS> Write-Log -EntryType Information -Message "Test Log Entry" #> # Determine if the Write-Information Cmdlet is available. $Script:WriteInformation = $PSVersionTable -ne $null -and $PSVersionTable.PSVersion.Major -ge 5 # Load strings file $CurrentPath = Split-Path -Path $MyInvocation.MyCommand.Definition -Parent Import-LocalizedData -BindingVariable r -FileName Strings.psd1 -BaseDirectory (Join-Path -Path $CurrentPath -ChildPath "Localized") $Script:r = $r # Array to hold log entries that will be e-mailed when using e-mail logging. $Script:LogEntries = @() # Settings for each of the log types. $Script:Settings = @{ File = New-Object -TypeName psobject -Property @{ Enabled = $false LogLevel = 0 FilePath = $null } Email = New-Object -TypeName psobject -Property @{ Enabled = $false } EventLog = New-Object -TypeName psobject -Property @{ Enabled = $false LogLevel = 0 LogName = $null Source = $null } Host = New-Object -TypeName psobject -Property @{ Enabled = $false LogLevel = 0 AnsiEscColor = $false TimeFormat= 'u' } PassThru = New-Object -TypeName psobject -Property @{ Enabled = $false LogLevel = 0 } Slack = New-Object -TypeName psobject -Property @{ Enabled = $false LogLevel = 0 Uri = "" Channel = "" Username = "" InformationIcon = "" WarningIcon = "" ErrorIcon = "" IncludeTimestamp = $false ChannelAlertLevel = -1 } } #------------------------------------------------------------------------------- # Public Cmdlets #------------------------------------------------------------------------------- #------------------------------------------------------------------------------- # File Logging Function Start-FileLog { <# .SYNOPSIS Begins to log output to file. .DESCRIPTION Begins to log output to file. Only entries with a severity at or above the specified level will be written. .PARAMETER FilePath Specifies the path and name of the log file that will be written. .PARAMETER LogLevel Specifies the minimum log entry severity to include in the file log. The default value is "Error". .PARAMETER Append Specifies that the file at <FilePath> should not be deleted if it already exists. New entries will be appended to the end of the file. .OUTPUTS None. .EXAMPLE C:\PS> Start-FileLog -FilePath c:\ps\data.log ----------- This command shows the minimum requirements for enabling file logging. It will replace any existing log file of the same name and will only log entries of type "Error". .EXAMPLE C:\PS> Start-FileLog -FilePath c:\ps\data.log -Append ----------- This command will leave an existing log file in place if it exists and append new log entries of type "Error" to the end of the file. .EXAMPLE C:\PS> Start-FileLog -FilePath c:\ps\data.log -LogLevel Warning -Append ----------- This command will leave an existing log file in place if it exists and append new log entries of type "Warning" or type "Error" to the end of the file. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [string]$FilePath, [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error", [Parameter()] [switch]$Append ) Process { $Script:Settings["File"].Enabled = $false # First attempt to remove existing file if necessary if (!$Append -and (Test-Path -LiteralPath $FilePath)) { try { Remove-Item -LiteralPath $FilePath -Force -ErrorAction Stop } catch { Write-Error -Exception $_.Exception -Message $Script:r.FileLogUnableToRemoveExistingFile return } } # Create file if necessary if (!(Test-Path -LiteralPath $FilePath)) { try { New-Item -Path $FilePath -ItemType File -Force -ErrorAction Stop | Out-Null } catch { Write-Error -Exception $_.Exception -Message $Script:r.FileLogUnableToCreateFile return } } $Script:Settings["File"].Enabled = $true $Script:Settings["File"].LogLevel = Get-LogLevel -EntryType $LogLevel $Script:Settings["File"].FilePath = $FilePath } } Function Stop-FileLog { <# .SYNOPSIS Stops writing log output to file. .DESCRIPTION Stops writing log output to file. Any log data that has already been written will remain. .OUTPUTS None. .EXAMPLE C:\PS> Stop-FileLog ----------- This command turns off file logging. #> [CmdletBinding(SupportsShouldProcess = $true)] Param () Process { if ($PSCmdlet.ShouldProcess($Script:r.CurrentSession)) { $Script:Settings["File"].Enabled = $false $Script:Settings["File"].FilePath = $null } } } #------------------------------------------------------------------------------- # E-mail Logging Function Start-EmailLog { <# .SYNOPSIS Starts recording log events so that they can be e-mailed. .DESCRIPTION Starts recording log events so that they can be e-mailed. A separate Cmdlet (Send-EmailLog) must be issued to actually send an e-mail. .PARAMETER ClearEntryCache Specifies whether any existing recorded log entries from the cache of entries to be e-mailed should be removed. .OUTPUTS None. .EXAMPLE C:\PS> Start-EmailLog ----------- This command begins the recording of log events so that they can be e-mailed. If any entries were already in the cache, they will remain there. .EXAMPLE C:\PS> Start-EmailLog -ClearEntryCache ----------- This command begins the recording of log events so that they can be e-mailed and clears the cache of recorded entries. #> [CmdletBinding()] Param ( [Parameter()] [switch]$ClearEntryCache = $false ) Process { $Script:Settings["Email"].Enabled = $true if ($ClearEntryCache) { $Script:LogEntries = @() } } } Function Stop-EmailLog { <# .SYNOPSIS Stops the recording of log events to the e-mail cache. .DESCRIPTION Stops the recording of log events to the e-mail cache. By default, the cache is also cleared. .PARAMETER RetainEntryCache Specifies whether any log entries that have already been recorded should be kept or discarded. .OUTPUTS None. .EXAMPLE C:\PS> Stop-EmailLog ----------- This command stops the recording of log entries and clears the e-mail cache. .EXAMPLE C:\PS> Stop-EmailLog -RetainEntryCache ----------- This command stops the recording of log entries but retains any that were already recorded. #> [CmdletBinding(SupportsShouldProcess = $true)] Param ( [Parameter()] [switch]$RetainEntryCache = $false ) Process { if ($PSCmdlet.ShouldProcess($Script:r.CurrentSession)) { $Script:Settings["Email"].Enabled = $false if (!$RetainEntryCache) { $Script:LogEntries = @() } } } } Function Send-EmailLog { <# .SYNOPSIS Sends an e-mail containing one or more of the log messages collected since e-mail logging was enabled. .DESCRIPTION Sends an e-mail containing one or more of the log messages collected since e-mail logging was enabled in the current session. Parameters can be used to control the severity of log message required to trigger sending an e-mail and also what levels are sent when an e-mail is triggered. .PARAMETER SmtpServer Specifies the SMTP server to use to send e-mail. .PARAMETER To Specifies one or more recipients for the e-mail. .PARAMETER From Specifies a from address to use when sending the e-mail. Note that some SMTP servers require this to be a valid mailbox. .PARAMETER Subject Specifies the subject of the e-mail message. .PARAMETER Message Specifies additional text to include in the e-mail message before the log data. .PARAMETER TriggerLogLevel Specifies the condition for sending an e-mail. A log entry at or above the specified level must have been recorded for an e-mail to be sent. .PARAMETER SendLogLevel Specifies what log events to include when sending an e-mail. This can be different than the TriggerLogLevel. .PARAMETER RetainEntryCache Specifies whether or not to keep the log entries that have been recorded. The default behavior is to clear them. .PARAMETER SendOnEmpty Specifies whether or not to send an e-mail if there are no log events that match the SendLogLevel parameter. .OUTPUTS None. .EXAMPLE C:\PS> Send-EmailLog -SmtpServer smtp.contoso.com -To ellen@contoso.com -From info@contoso.com ----------- This command shows the minimum requirements for e-mailing a log. A default subject will be used and no extra message will be added. Because the TriggerLogLevel parameter was not provided, the module will try to send an e-mail no matter what, using the default SendLogLevel of Error. .EXAMPLE C:\PS> Send-EmailLog -SmtpServer smtp.contoso.com -To ellen@contoso.com -From info@contoso.com -TriggerLogLevel Error -SendLogLevel Information ----------- This command will cause the module only to send an e-mail if any errors were encountered, but when sending an e-mail it will send the full log. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [string]$SmtpServer, [Parameter( Mandatory = $true, ValueFromPipeline = $true, ValueFromPipelineByPropertyName = $true )] [string[]]$To, [Parameter(Mandatory = $true)] [string]$From, [Parameter()] [string]$Subject = "", [Parameter()] [string]$Message = "", [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$TriggerLogLevel = "Error", [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$SendLogLevel = "Error", [Parameter()] [switch]$RetainEntryCache = $false, [Parameter()] [switch]$SendOnEmpty = $false, [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error" ) Begin { $Bypass = $false if ($PSBoundParameters.ContainsKey("LogLevel")) { # Deprecated functionality. Write-Warning -Message ([string]::Format($Script:r.Parameter_F0_Deprecated_F1, "LogLevel", "TriggerLogLevel and SendLogLevel")) $TriggerLogLevel = $LogLevel $SendLogLevel = $LogLevel } # Start by checking if anything was logged that fits our trigger level. $TriggerLogLevelNumber = Get-LogLevel -EntryType $TriggerLogLevel if ((!$PSBoundParameters.ContainsKey("TriggerLogLevel") -and !$PSBoundParameters.ContainsKey("LogLevel")) -or ($Script:LogEntries | Where-Object -FilterScript { $_.LogLevel -le $TriggerLogLevelNumber })) { if (!$Subject) { $Subject = $Script:r.EmailLogSubject } $EmailBody = "<style>.log-entries {font-family: `"Lucida Console`", Monaco, monospace;font-size: 10pt;}</style><body>" if ($Message) { $EmailBody += "<p>$Message</p>" } $SendLogLevelNumber = Get-LogLevel -EntryType $SendLogLevel $Entries = $Script:LogEntries | Where-Object -FilterScript { $_.LogLevel -le $SendLogLevelNumber } $Empty = $false if ($Entries) { $EmailBody += "<div class=`"log-entries`">" $EmailBody += $Entries | ConvertTo-HtmlUnorderedList -FormatScript { Param ($Entry) $Line = "[$($Entry.Timestamp.ToString("u"))] - " switch ($Entry.EntryType) { "Information" { $Line += "<span style=`"color: Teal`">$($Script:r.Info)</span>" } "Warning" { $Line += "<span style=`"color: GoldenRod`">$($Script:r.Warn)</span>" } "Error" { $Line += "<span style=`"color: Red`">$($Script:r.Errr)</span>" } } $Line += ": $($Entry.Message)" if ($Entry.Exception) { $Line += "<ul><li>$($Script:r.Message): $($Entry.Exception.Message)</li><li>$($Script:r.Source): $($Entry.Exception.Source)</li><li>$($Script:r.StackTrace):" if ($Entry.Exception.StackTrace -and $Entry.Exception.StackTrace.Count -gt 0) { $Line += "<ul>" foreach ($Stack in $Entry.Exception.StackTrace) { $Line += "<li>$Stack</li>" } $Line += "</ul>" } $Line += "</li><li>$($Script:r.TargetSite): $($Entry.Exception.TargetSite)</li></ul>" } $Line } $EmailBody += "</div>" } else { $Empty = $true $EmailBody += "<p>$($Script:r.NoEntriesToReport)</p>" } $EmailBody += "</body>" } else { # No events occurred that would trigger us to send an e-mail. $Bypass = $true } } Process { if (!$Bypass -and (!$Empty -or $SendOnEmpty)) { Send-MailMessage -From $From -To $To -Subject $Subject -Body $EmailBody -SmtpServer $SmtpServer -BodyAsHtml } } End { if (!$Bypass -and !$RetainEntryCache) { $Script:LogEntries = @() } } } #------------------------------------------------------------------------------- # Event Log Logging Function Start-EventLogLog { <# .SYNOPSIS Starts recording log events to the Windows Event Log. .DESCRIPTION Starts recording log events to the Windows Event Log. Which log is written to and what source is used are configurable. .PARAMETER Source Specifies the Event Log source to record events under. If the source does not exist, the module will attempt to create it, but this requires administrative rights. You might need to run the script as an administrator the first time to create the source, but once it exists you should not need to. .PARAMETER LogLevel Specifies the minimum log entry severity to write to the Event Log. The default value is "Error". .PARAMETER LogName Specifies the Windows Event Log to write events to. The default is "Application" .OUTPUTS None. .EXAMPLE C:\PS> Start-EventLogLog -Source "MyScript" ----------- This command shows the minimum required parameter set to enable Event Log logging. If the "MyScript" source does not exist in the Event Log, it will be created. Because the default LogLevel of "Error" is being used, only errors will be written to the Event Log. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [string]$Source, [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error", [Parameter()] [string]$LogName = "Application" ) Process { $Script:Settings["EventLog"].Enabled = $false # Check if the source exists. try { if (Test-EventLogSource -Source $Source) { # It does exist, make sure it points at the right log. if ((Get-LogName -Source $Source) -ne $LogName) { # Source exists but points to a different log. Not good! Write-Error -Message $Script:r.EventLogLogSourceInWrongLog return } } else { # Source does not exist, try to create it. try { New-EventLog -LogName $LogName -Source $Source } catch [System.Exception] { Write-Error -Exception $_.Exception -Message $Script:r.EventLogLogUnableToCreateLogOrSource return } } } catch [System.Exception] { Write-Error -Exception $_.Exception -Message $Script:r.EventLogLogUnableToReadLogSources return } $Script:Settings["EventLog"].Enabled = $true $Script:Settings["EventLog"].LogLevel = Get-LogLevel -EntryType $LogLevel $Script:Settings["EventLog"].LogName = $LogName $Script:Settings["EventLog"].Source = $Source } } Function Stop-EventLogLog { <# .SYNOPSIS Stops writing log output to the Windows Event Log. .DESCRIPTION Stops writing log output to the Windows Event Log. .OUTPUTS None. .EXAMPLE C:\PS> Stop-EventLogLog ----------- This command turns off Event Log logging. #> [CmdletBinding(SupportsShouldProcess = $true)] Param () Process { if ($PSCmdlet.ShouldProcess($Script:r.CurrentSession)) { $Script:Settings["EventLog"].Enabled = $false $Script:Settings["EventLog"].LogName = $null $Script:Settings["EventLog"].Source = $null } } } #------------------------------------------------------------------------------- # Host Logging Function Start-HostLog { <# .SYNOPSIS Turns on writing formatted log events to the host display. .DESCRIPTION Starts writing formatted log events to the host display. Includes timestamp, color-coded entry type, and message text. .PARAMETER LogLevel Specifies the minimum log entry severity to write to the host. The default value is "Error". .PARAMETER AnsiEscColor Enables ANSI escape color. .PARAMETER TimeFormat Specifies the time sample format. The default is 'u'. .OUTPUTS None. .EXAMPLE C:\PS> Start-HostLog ----------- This command turns on host logging. #> [CmdletBinding()] Param ( [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error", [Parameter()] [ValidateSet($true, $false)] [switch]$AnsiEscColor = $false, [Parameter()] [string]$TimeFormat = "u" ) Process { $Script:Settings["Host"].Enabled = $true $Script:Settings["Host"].LogLevel = Get-LogLevel -EntryType $LogLevel $Script:Settings["Host"].AnsiEscColor = $AnsiEscColor $Script:Settings["Host"].TimeFormat = $TimeFormat } } Function Stop-HostLog { <# .SYNOPSIS Turns off writing log messages to the host display. .DESCRIPTION Turns off writing log messages to the host display. .OUTPUTS None. .EXAMPLE C:\PS> Stop-HostLog ----------- This command turns off host logging. #> [CmdletBinding(SupportsShouldProcess = $true)] Param () Process { if ($PSCmdlet.ShouldProcess($Script:r.CurrentSession)) { $Script:Settings["Host"].Enabled = $false } } } #------------------------------------------------------------------------------- # "Pass Thru" Logging Function Start-PassThruLog { <# .SYNOPSIS Turns on "Pass Thru" display of log events by writing them to other streams. .DESCRIPTION Turns on "Pass Thru" display of log events by writing them to other streams. The streams used are: - Information - Verbose Stream - Warning - Warning Stream - Error - Error stream .PARAMETER LogLevel Specifies the minimum log entry severity to write to another stream. The default value is "Error". .OUTPUTS None. .EXAMPLE C:\PS> Start-PassThruLog ----------- This command turns on "Pass Thru" logging. #> [CmdletBinding()] Param ( [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error" ) Process { $Script:Settings["PassThru"].Enabled = $true $Script:Settings["PassThru"].LogLevel = Get-LogLevel -EntryType $LogLevel } } Function Stop-PassThruLog { <# .SYNOPSIS Turns off "Pass Thru" logging. .DESCRIPTION Turns off "Pass Thru" logging. .OUTPUTS None. .EXAMPLE C:\PS> Stop-PassThruLog ----------- This command turns off "Pass Thru" logging. #> [CmdletBinding(SupportsShouldProcess = $true)] Param () Process { if ($PSCmdlet.ShouldProcess($Script:r.CurrentSession)) { $Script:Settings["PassThru"].Enabled = $false } } } #------------------------------------------------------------------------------- # Slack Channel Logging Function Start-SlackLog { <# .SYNOPSIS Turns on Slack Channel logging. .DESCRIPTION Turns on Slack Channel logging using Slack custom webhook integrations. .PARAMETER LogLevel Specifies the minimum log entry severity to send to Slack. The default value is "Error". .PARAMETER Uri Specifies the webhook URI to use when sending events to Slack. .PARAMETER Channel Specifies the name of a Slack channel to post messages to. Do not include the hash (#) at the beginning of the channel name. .PARAMETER Username Specifies a custom username to display on messages sent from this process. .PARAMETER InformationIcon Specifies the name of an emoji icon to use for Information log messages. Do not surround the emoji name in colons (:). .PARAMETER WarningIcon Specifies the name of an emoji icon to use for Warning log messages. Do not surround the emoji name in colons (:). .PARAMETER ErrorIcon Specifies the name of an emoji icon to use for Error log messages. Do not surround the emoji name in colons (:). .PARAMETER IncludeTimestamp When included, this switch specifies that a timestamp should be written to the body of the message, instead of just relying on the time Slack received the entry. .PARAMETER ChannelAlertLevel Specifies the minimum log entry severity to include the @channel tag in. This tag will cause an alert to be sent to all members of the channel. The default value is for this to be disabled. .OUTPUTS None. .EXAMPLE C:\PS> Start-SlackLog -Uri "https://hooks.slack.com/services/ABC123" ---------- This command enables logging to Slack at the specified integration URL, using the default channel and username that were set up with the integration and the default icons for the different log entry levels. #> [CmdletBinding()] Param ( [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error", [Parameter(Mandatory = $true)] [string]$Uri, [Parameter()] [string]$Channel, [Parameter()] [string]$Username, [Parameter()] [string]$InformationIcon = "speech_balloon", [Parameter()] [string]$WarningIcon = "warning", [Parameter()] [string]$ErrorIcon = "rotating_light", [Parameter()] [switch]$IncludeTimestamp, [Parameter()] [ValidateSet("Information", "Warning", "Error", "None")] [string]$ChannelAlertLevel = "None" ) Process { $Script:Settings["Slack"].Enabled = $true $Script:Settings["Slack"].LogLevel = Get-LogLevel -EntryType $LogLevel $Script:Settings["Slack"].Uri = $Uri $Script:Settings["Slack"].InformationIcon = $InformationIcon $Script:Settings["Slack"].WarningIcon = $WarningIcon $Script:Settings["Slack"].ErrorIcon = $ErrorIcon $Script:Settings["Slack"].ChannelAlertLevel = Get-LogLevel -EntryType $ChannelAlertLevel if ($Channel) { $Script:Settings["Slack"].Channel = $Channel } else { $Script:Settings["Slack"].Channel = "" } if ($Username) { $Script:Settings["Slack"].Username = $Username } else { $Script:Settings["Slack"].Username = "" } if ($IncludeTimestamp) { $Script:Settings["Slack"].IncludeTimestamp = $true } else { $Script:Settings["Slack"].IncludeTimestamp = $false } } } Function Stop-SlackLog { <# .SYNOPSIS Turns off Slack Channel logging. .DESCRIPTION Turns off Slack Channel logging. .OUTPUTS None. .Example C:\PS> Stop-SlackLog ---------- This command turns off Slack Channel logging. #> [CmdletBinding(SupportsShouldProcess = $true)] Param () Process { if ($PSCmdlet.ShouldProcess($Script:r.CurrentSession)) { $Script:Settings["Slack"].Enabled = $false } } } #------------------------------------------------------------------------------- # Main logging method Function Write-Log { <# .SYNOPSIS Writes a log entry to whichever output formats are currently enabled. .DESCRIPTION Writes a log entry to whichever output formats are currently enabled. .PARAMETER EntryType Specifies what type of log entry to write. .PARAMETER Message Specifies a descriptive message for the log entry. This is separate from the message that is attached to any exception that might be included in the log event. .PARAMETER Exception For error type entries, includes information about an actual exception that occurred. .PARAMETER EventId For Event Log entries, specifies the Event Id to write in the Event Log. The default is 1000. .OUTPUTS None. .EXAMPLE C:\PS> Write-Log -EntryType Information -Message "This is a sample log message." ----------- This command writes a simple log message. .EXAMPLE C:\PS> Write-Log -EntryType Error -Message "An exception occurred." -Exception $_.Exception ----------- This command, which might be used in a try/catch block, logs an error, including data about the exception that was caught. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [ValidateSet("Information", "Warning", "Error")] [string]$EntryType, [Parameter(Mandatory = $true)] [string]$Message, [Parameter()] [System.Exception]$Exception = $null, [Parameter()] [int]$EventId = 1000 ) Process { $NewEntry = New-Object -TypeName psobject -Property @{ Timestamp = Get-Date EntryType = $EntryType LogLevel = Get-LogLevel -EntryType $EntryType Message = $Message Exception = $Exception EventId = $EventId } # Log to File if ($Script:Settings["File"].Enabled -and $NewEntry.LogLevel -le $Script:Settings["File"].LogLevel) { Write-FileLog -Entry $NewEntry -FilePath $Script:Settings["File"].FilePath } # Log to EventLog if ($Script:Settings["EventLog"].Enabled -and $NewEntry.LogLevel -le $Script:Settings["EventLog"].LogLevel) { Write-EventLogLog -Entry $NewEntry -LogName $Script:Settings["EventLog"].LogName -Source $Script:Settings["EventLog"].Source } # Record entry for e-mailing later if ($Script:Settings["Email"].Enabled) { Write-EmailLog -Entry $NewEntry } # Write to host if ($Script:Settings["Host"].Enabled -and $NewEntry.LogLevel -le $Script:Settings["Host"].LogLevel) { Write-HostLog -Entry $NewEntry } # Pass through to verbose/warning/error streams if ($Script:Settings["PassThru"].Enabled -and $NewEntry.LogLevel -le $Script:Settings["PassThru"].LogLevel) { Write-PassThruLog -Entry $NewEntry } # Slack channel logging. if ($Script:Settings["Slack"].Enabled -and $NewEntry.LogLevel -le $Script:Settings["Slack"].LogLevel) { Write-SlackLog -Entry $NewEntry } } } #------------------------------------------------------------------------------- # Internal Cmdlets #------------------------------------------------------------------------------- Function Get-LogLevel { <# .SYNOPSIS Gets the integer representation of the specified entry type. .DESCRIPTION Gets the integer representation of the specified entry type. Used for filtering log output. .PARAMETER EntryType Specifies the entry type to evaluate. .OUTPUTS Integer. #> [CmdletBinding()] [OutputType([int])] Param ( [Parameter(Mandatory = $true)] [ValidateSet("Information", "Warning", "Error", "None")] [string]$EntryType ) Process { switch ($EntryType) { "Information" { return 2 } "Warning" { return 1 } "Error" { return 0 } "None" { return -1 } } } } Function Format-LogMessage { <# .SYNOPSIS Formats a log entry for output and returns the formatted string. .DESCRIPTION Formats a log entry for output and returns the formatted string. Used by the File and PassThru logging methods. .PARAMETER Entry Specifies the log entry to format. .PARAMETER Type Specifies whether or not to include the log entry type in the formatted string. .PARAMETER Exception Specifies whether or not to include information about any exceptions included in the entry in the formatted string. .OUTPUTS String. #> [CmdletBinding()] [OutputType([string])] Param ( [Parameter(Mandatory = $true)] [psobject]$Entry, [Parameter()] [switch]$Type, [Parameter()] [switch]$Exception ) Process { $ReturnString = "[$($Entry.Timestamp.ToString("u"))]" if ($Type) { $TypeString = "" switch($Entry.EntryType) { "Information" { $TypeString = $Script:r.Info } "Warning" { $TypeString = $Script:r.Warn } "Error" { $TypeString = $Script:r.Errr } } $ReturnString += " - $TypeString" } $ReturnString += " - $($Entry.Message)" if ($Exception -and $Entry.Exception) { $ReturnString += " - $($Script:r.Exception): $($Entry.Exception.Message)" } return $ReturnString } } Function ConvertTo-HtmlUnorderedList { <# .SYNOPSIS Builds an HTML UnorderedList from the supplied input and returns its string. .DESCRIPTION Builds an HTML UnorderedList from the supplied input and returns its string. .PARAMETER FormatScript Specifies a script block to invoke for each object passed into the Cmdlet. .PARAMETER InputObject Specifies one or more objects to write to the unordered list. .OUTPUTS String. #> [CmdletBinding()] [OutputType([string])] Param ( [Parameter()] [scriptblock]$FormatScript = $null, [Parameter( Mandatory = $true, ValueFromPipeline = $true, ValueFromPipelineByPropertyName = $true )] $InputObject ) Begin { $OutputText = "<ul>`n" } Process { @($InputObject) | ForEach-Object -Process { $OutputText += "<li>" if ($FormatScript) { $OutputText += Invoke-Command -ScriptBlock $FormatScript } else { $OutputText += $_ } $OutputText += "</li>`n" } } End { $OutputText += "</ul>`n" $OutputText } } Function Write-FileLog { <# .SYNOPSIS Writes a log message to a file. .DESCRIPTION Writes a log message to a file. .PARAMETER Entry Specifies the log entry to write. .PARAMETER FilePath Specifies the file to write the log entry to. .OUTPUTS None. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [psobject]$Entry, [Parameter(Mandatory = $true)] [string]$FilePath ) Process { Format-LogMessage -Entry $Entry -Type -Exception | Out-File -FilePath $FilePath -Append -Encoding ascii } } Function Write-EventLogLog { <# .SYNOPSIS Creates a new Event Log object from a log message. .DESCRIPTION Creates a new Event Log object from a log message. .PARAMETER Entry Specifies the log entry which will be used to create the Event Log object. .PARAMETER LogName Specifies which log in the Windows Event Log to write to. .PARAMETER Source Specifies the source to use when creating the Event Log object. .OUTPUTS None. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [psobject]$Entry, [Parameter(Mandatory = $true)] [string]$LogName, [Parameter(Mandatory = $true)] [string]$Source ) Process { $EventLogMessage = $Entry.Message if ($Entry.Exception) { $EventLogMessage += "`n`n$($Script:r.ExceptionInformation)" + ` "`n$($Script:r.Message): $($Entry.Exception.Message)" + ` "`n$($Script:r.Source): $($Entry.Exception.Source)" + ` "`n$($Script:r.StackTrace): $($Entry.Exception.StackTrace)" + ` "`n$($Script:r.TargetSite): $($Entry.Exception.TargetSite)" } Write-EventLog -LogName $LogName -Source $Source -EntryType $Entry.EntryType -EventId $Entry.EventId -Message $EventLogMessage } } Function Write-EmailLog { <# .SYNOPSIS Stores a log entry in the cache used when e-mailing log data. .DESCRIPTION Stores a log entry in the cache used when e-mailing log data. .PARAMETER Entry Specifies the log entry to record. .OUTPUTS None. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [psobject]$Entry ) Process { $Script:LogEntries += $Entry } } Function Write-HostLog { <# .SYNOPSIS Writes a log entry to the host. .DESCRIPTION Writes a log entry to the host. .PARAMETER Entry Specifies the log entry to write to the host. .OUTPUTS None. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [psobject]$Entry ) Process { $SGRESC="$([char]27)" $CLEAR="$SGRESC[0m" Write-Host -Object "[$($Entry.Timestamp.ToString($Script:Settings["Host"].TimeFormat))] - " -NoNewline switch ($Entry.EntryType) { "Information" { if ($Script:Settings["Host"].AnsiEscColor) { $CYAN="$SGRESC[96m" Write-Host -Object ($CYAN + $Script:r.Info + $CLEAR) -NoNewline } else { Write-Host -Object $Script:r.Info -ForegroundColor Cyan -NoNewline } } "Warning" { if ($Script:Settings["Host"].AnsiEscColor) { $YELLOW="$SGRESC[33m" Write-Host -Object ($YELLOW + $Script:r.Warn + $CLEAR) -NoNewline } else { Write-Host -Object $Script:r.Warn -ForegroundColor Yellow -NoNewline } } "Error" { if ($Script:Settings["Host"].AnsiEscColor) { $RED="$SGRESC[31m" Write-Host -Object ($RED + $Script:r.Errr + $CLEAR) -NoNewline } else { Write-Host -Object $Script:r.Errr -ForegroundColor Red -NoNewline } } } $Message = $Entry.Message if ($Entry.Exception) { $Message += " - $($Script:r.Exception): $($Entry.Exception.Message)" } Write-Host -Object " - $Message" } } Function Write-PassThruLog { <# .SYNOPSIS Writes a log entry to one of the native PowerShell Streams. .DESCRIPTION Writes a log entry to one of the native PowerShell Streams. .PARAMETER Entry Specifies the log entry to write. #> [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [psobject]$Entry ) Process { switch ($Entry.EntryType) { "Information" { # Use the Write-Information Cmdlet introduced in PowerShell # version 5 if available. Otherwise Write-Verbose if ($Script:WriteInformation) { Write-Information -MessageData (Format-LogMessage -Entry $Entry) } else { Write-Verbose -Message (Format-LogMessage -Entry $Entry) } } "Warning" { Write-Warning -Message (Format-LogMessage -Entry $Entry) } "Error" { if ($Entry.Exception) { Write-Error -Message (Format-LogMessage -Entry $Entry) -Exception $Entry.Exception } else { Write-Error -Message (Format-LogMessage -Entry $Entry) } } } } } Function Write-SlackLog { [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [psobject]$Entry ) Process { $Message = "" if ($Entry.LogLevel -le $Script:Settings["Slack"].ChannelAlertLevel) { $Message += "<!channel>: " } if ($Script:Settings["Slack"].IncludeTimestamp) { $Message += "[$($Entry.Timestamp.ToString("u"))] - " | ConvertTo-SlackEncoding } $Message += $Entry.Message | ConvertTo-SlackEncoding if ($Entry.Exception) { $Message += "`n`n$($Script:r.ExceptionInformation)" + ` "`n$($Script:r.Message): $($Entry.Exception.Message)" + ` "`n$($Script:r.Source): $($Entry.Exception.Source)" + ` "`n$($Script:r.StackTrace): $($Entry.Exception.StackTrace)" + ` "`n$($Script:r.TargetSite): $($Entry.Exception.TargetSite)" | ConvertTo-SlackEncoding } switch ($Entry.EntryType) { "Information" { $Icon = $Script:Settings["Slack"].InformationIcon | ConvertTo-SlackEncoding } "Warning" { $Icon = $Script:Settings["Slack"].WarningIcon | ConvertTo-SlackEncoding } "Error" { $Icon = $Script:Settings["Slack"].ErrorIcon | ConvertTo-SlackEncoding } } $Elements = @() $Elements += '"text":"' + $Message + '"' $Elements += '"icon_emoji":":' + $Icon + ':"' if ($Script:Settings["Slack"].Channel) { $Elements += '"channel":"#' + ($Script:Settings["Slack"].Channel | ConvertTo-SlackEncoding) + '"' } if ($Script:Settings["Slack"].Username) { $Elements += '"username":"' + ($Script:Settings["Slack"].Username | ConvertTo-SlackEncoding) + '"' } $Body = "payload={$([string]::Join(',', $Elements))}" try { Invoke-WebRequest -Uri $Script:Settings["Slack"].Uri -Method Post -Body $Body -ErrorAction Stop | Out-Null } catch { Write-Error -Exception $_.Exception -Message "Exception while posting log to Slack. Message is '$($_.Exception.Message)'." } } } Function ConvertTo-SlackEncoding { <# .SYNOPSIS Encodes special characters for inclusion into a slack webhook request. .DESCRIPTION Encodes special characters for inclusion into a slack webhook request. .PARAMETER InputString Specifies the string to encode. .OUTPUTS String. The encoded value. #> [CmdletBinding()] Param( [Parameter( Mandatory = $true, ValueFromPipeline = $true )] [string]$InputString ) Process { return $InputString.Replace('"', '\"').Replace('&', '&').Replace('<', '<').Replace('>', '>') } } Function ConvertTo-HtmlUnorderedList { <# .SYNOPSIS Builds an HTML UnorderedList from the supplied input and returns its string. .DESCRIPTION Builds an HTML UnorderedList from the supplied input and returns its string. .PARAMETER FormatScript Specifies a script block to invoke for each object passed into the Cmdlet. .PARAMETER InputObject Specifies one or more objects to write to the unordered list. .OUTPUTS String. #> [CmdletBinding()] [OutputType([string])] Param ( [Parameter()] [scriptblock]$FormatScript = $null, [Parameter( Mandatory = $true, ValueFromPipeline = $true, ValueFromPipelineByPropertyName = $true )] $InputObject ) Begin { $OutputText = "<ul>`n" } Process { @($InputObject) | ForEach-Object -Process { $OutputText += "<li>" if ($FormatScript) { $OutputText += Invoke-Command -ScriptBlock $FormatScript -ArgumentList $_ } else { $OutputText += $_ } $OutputText += "</li>`n" } } End { $OutputText += "</ul>`n" $OutputText } } Function Test-EventLogSource { [CmdletBinding()] [OutputType([bool])] Param ( [Parameter(Mandatory = $true)] [string]$Source ) Process { return [System.Diagnostics.EventLog]::SourceExists($Source) } } Function Get-LogName { [CmdletBinding()] [OutputType([string])] Param ( [Parameter(Mandatory = $true)] [string]$Source ) Process { return [System.Diagnostics.EventLog]::LogNameFromSourceName($Source, ".") } } #------------------------------------------------------------------------------- # Deprecated Cmdlets #------------------------------------------------------------------------------- Function Enable-FileLog { [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [string]$FilePath, [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error", [Parameter()] [switch]$Append ) Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Start-FileLog")) Start-FileLog -FilePath $FilePath -LogLevel $LogLevel -Append:$Append } } Function Disable-FileLog { [CmdletBinding()] Param () Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Stop-FileLog")) Stop-FileLog } } Function Enable-EmailLog { [CmdletBinding()] Param ( [Parameter()] [switch]$ClearEntryCache ) Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Start-EmailLog")) Start-EmailLog -ClearEntryCache:$ClearEntryCache } } Function Disable-EmailLog { [CmdletBinding()] Param ( [Parameter()] [switch]$RetainEntryCache ) Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Stop-EmailLog")) Stop-EmailLog } } Function Enable-EventLogLog { [CmdletBinding()] Param ( [Parameter(Mandatory = $true)] [string]$Source, [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error", [Parameter()] [string]$LogName = "Application" ) Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Start-EventLogLog")) Start-EventLogLog -Source $Source -LogLevel $LogLevel -LogName $LogName } } Function Disable-EventLogLog { [CmdletBinding()] Param () Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Stop-EventLogLog")) Stop-EventLogLog } } Function Enable-HostLog { [CmdletBinding()] Param ( [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error" ) Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Start-HostLog")) Enable-HostlLog -LogLevel $LogLevel } } Function Disable-HostLog { [CmdletBinding()] Param () Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Stop-HostLog")) Stop-HostLog } } Function Enable-PassThruLog { [CmdletBinding()] Param ( [Parameter()] [ValidateSet("Information", "Warning", "Error")] [string]$LogLevel = "Error" ) Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Start-PassThruLog")) Start-PassThruLog -LogLevel $LogLevel } } Function Disable-PassThruLog { [CmdletBinding()] Param () Process { Write-Warning -Message ([string]::Format($Script:r.CmdletDeprecated_F0, "Stop-PassThruLog")) Stop-PassThruLog } } |