TracePeek.psm1

$ErrorActionPreference = "Stop"
Set-StrictMode -Version Latest

Add-Type -Path (Join-Path $psscriptRoot 'TracePeek.dll') -ErrorAction Stop

function Start-TracePeek
{
    [CmdletBinding(PositionalBinding=$true)]
    param(
        [Parameter(Mandatory=$true)]
        [string[]]$Providers,
        [TracePeek.TracePeekProjectionStyle]$ProjectionStyle,
        [string]$SessionName
    )
    $tpc = New-Object -TypeName TracePeek.TracePeekController -ArgumentList $SessionName
    
    $providersList = New-Object -TypeName System.Collections.Generic.List`[string]
    foreach($provider in $providers){
        $providersList.Add($provider)
    }
    $tpc.EnableProviders($providersList)
    $eventSubscriberJob = Register-ObjectEvent -InputObject $tpc -EventName OnTracePeekEvent -SourceIdentifier $SessionName
    Write-Host "TracePeek starting up..."
    if($null -eq $ProjectionStyle){$null = $tpc.StartPeek()}
    else{$null = $tpc.StartPeek($ProjectionStyle)}
    Write-Host "TracePeek running."
    Write-Host '!!! Press ALT+s when done to stop session and clean up resources !!!'
    $consoleCancellationRequested = $false
    while($consoleCancellationRequested -eq $false)
    {
        Wait-Event -SourceIdentifier $SessionName -Timeout 1 -ErrorAction SilentlyContinue |
        ForEach-Object{ Write-Output $_.SourceArgs[0] ; Remove-Event -EventIdentifier $_.EventIdentifier}
        while([Console]::KeyAvailable)
        {
            $keyPress = [Console]::ReadKey($true)
            if ($keyPress.Modifiers.HasFlag([System.ConsoleModifiers]::Alt) -and $keyPress.KeyChar -eq 's')
            {
                Write-Host "TracePeek stopping and cleaning up."
                $consoleCancellationRequested = $true
                $tpc.StopPeek()
            }
        }
    }
    Unregister-Event -SourceIdentifier $SessionName
    if($null -ne $eventSubscriberJob){Remove-Job -Id $eventSubscriberJob.Id -Force}
}