RabbitMqLogger.psm1

$scriptPath = Split-Path -parent $PSCommandPath

Add-Type -Path "$scriptPath\RabbitMQ.Client.dll"
Add-Type -Path "$scriptPath\EasyNetQ.dll"
Add-Type -Path "$scriptPath\RabbitMqLogger.dll"

# Create convenience access to the logger and enum values in a global object
$global:rmq = @{}
$global:rmq.Info = [RabbitMqLogger.MessageType]::Info
$global:rmq.JobStart = [RabbitMqLogger.MessageType]::JobStart
$global:rmq.JobEnd = [RabbitMqLogger.MessageType]::JobEnd
$global:rmq.StepStart = [RabbitMqLogger.MessageType]::StepStart
$global:rmq.StepEnd = [RabbitMqLogger.MessageType]::StepEnd

function HandleException($exception) {
	$inner = $exception.InnerException
	while ($inner) {
		$exception = $inner
		$inner = $exception.InnerException
	}
	Write-Error "Could not log to RMQ: $($exception.Message)"
}

function New-RmqLogger($host, $topic = "GlobalLogging") {
	$global:rmq.Logger = New-Object RabbitMqLogger.Logger -ArgumentList $host, $topic
}

function Assert-Logger() {
	if (-Not $global:rmq.Logger) {
		Write-Error "RMQ logger has not been initialized. Use New-RmqLogger."
	}
}

function Write-DebugRmqLog(
		[Parameter(Mandatory=$true)][string]$message, 
		[RabbitMqLogger.MessageType]$type = $global:rmq.Info, 
		[string]$category = "Uncategorized") {
	Assert-Logger
	try {
		$global:rmq.Logger.Debug($message, $type, $category)
	} catch {
		HandleException $_.Exception
	}
}

function Write-InfoRmqLog(
		[Parameter(Mandatory=$true)][string]$message, 
		[RabbitMqLogger.MessageType]$type = $global:rmq.Info, 
		[string]$category = "Uncategorized") {
	Assert-Logger
	try {
		$global:rmq.Logger.Info($message, $type, $category)
	} catch {
		HandleException $_.Exception
	}
}

function Write-WarnRmqLog(
		[Parameter(Mandatory=$true)][string]$message, 
		[RabbitMqLogger.MessageType]$type = $global:rmq.Info, 
		[string]$category = "Uncategorized") {
	Assert-Logger
	try {
		$global:rmq.Logger.Warn([string]$message, $type, $category)
	} catch {
		HandleException $_.Exception
	}
}

function Write-ErrorRmqLog(
		[Parameter(Mandatory=$true)][string]$message, 
		[RabbitMqLogger.MessageType]$type = $global:rmq.Info, 
		[string]$category = "Uncategorized", 
		$error = $null) {
	Assert-Logger
	try {
		$global:rmq.Logger.Error($message, $type, $category, $error)
	} catch {
		HandleException $_.Exception
	}
}

function Write-FatalRmqLog(
		[string]$message, 
		[RabbitMqLogger.MessageType]$type = $global:rmq.Info, 
		[string]$category = "Uncategorized", 
		$error = $null) {
	Assert-Logger
	try {
		$global:rmq.Logger.Fatal($message, $type, $category, $error)
	} catch {
		HandleException $_.Exception
	}
}

Export-ModuleMember -Function New-RmqLogger
Export-ModuleMember -Function Write-DebugRmqLog
Export-ModuleMember -Function Write-InfoRmqLog
Export-ModuleMember -Function Write-WarnRmqLog
Export-ModuleMember -Function Write-ErrorRmqLog
Export-ModuleMember -Function Write-FatalRmqLog