XSockets.PerformanceCounters 0.1

Set-StrictMode -Version 2

function AddCounters {

    $counterCollection = new-object System.Diagnostics.CounterCreationDataCollection
    $counterCollection.AddRange($counters)

    [void] [System.Diagnostics.PerformanceCounterCategory]::Create($categoryName, "XSockets.NET statistics", [System.Diagnostics.PerformanceCounterCategoryType]::MultiInstance, $counterCollection)
    [System.Diagnostics.PerformanceCounter]::CloseSharedResources()
    Write-Host -ForegroundColor Green "Successfully Added Counters" | Out-Default
}

function DeleteCategory {
    [void] [System.Diagnostics.PerformanceCounterCategory]::Delete($categoryName)
}

function DoesCategoryExist {
    [System.Diagnostics.PerformanceCounterCategory]::Exists($categoryName)
}

$categoryName = "XSockets.NET"

$counters = @(

    (new-object System.Diagnostics.CounterCreationData( "TOTAL-ERRORS", 
                                                        "The number of errors registered", 
                                                        [System.Diagnostics.PerformanceCounterType]::NumberOfItems32)),
    (new-object System.Diagnostics.CounterCreationData( "CURRENT-CONNECTIONS", 
                                                        "The number of clients connected right now", 
                                                        [System.Diagnostics.PerformanceCounterType]::NumberOfItems32)),
    (new-object System.Diagnostics.CounterCreationData( "TOTAL-INVALID-HANDSHAKES", 
                                                        "The number of connectionattempts that had a bad handshake", 
                                                        [System.Diagnostics.PerformanceCounterType]::NumberOfItems32)),
    (new-object System.Diagnostics.CounterCreationData( "TOTAL-VALID-HANDSHAKES", 
                                                        "The number of connectionattempts that had a good handshake", 
                                                        [System.Diagnostics.PerformanceCounterType]::NumberOfItems32)),
                                                        
    (new-object System.Diagnostics.CounterCreationData( "OUT/SEC", 
                                                        "The current number of messages successfully sent per second.",
                                                        [System.Diagnostics.PerformanceCounterType]::RateOfCountsPerSecond32)),

    (new-object System.Diagnostics.CounterCreationData( "INVALID-HANDSHAKES/SEC", 
                                                        "The current number of invalid handshakes per second.", 
                                                        [System.Diagnostics.PerformanceCounterType]::RateOfCountsPerSecond32)),
    (new-object System.Diagnostics.CounterCreationData( "VALID-HANDSHAKES/SEC", 
                                                        "The current number of valid handshakes per second.", 
                                                        [System.Diagnostics.PerformanceCounterType]::RateOfCountsPerSecond32)),                                                    
    (new-object System.Diagnostics.CounterCreationData( "IN/SEC", 
                                                        "The current number of messages successfully received per second.", 
                                                        [System.Diagnostics.PerformanceCounterType]::RateOfCountsPerSecond32)),
    (new-object System.Diagnostics.CounterCreationData( "ERRORS/SEC", 
                                                        "The current number of errors registered per second.", 
                                                        [System.Diagnostics.PerformanceCounterType]::RateOfCountsPerSecond32))
)

$currentIdentity = [System.Security.Principal.WindowsIdentity]::GetCurrent()
$currentPrincipal = new-object System.Security.Principal.WindowsPrincipal($currentIdentity)
$adminRole = [System.Security.Principal.WindowsBuiltInRole]::Administrator

if (-not $currentPrincipal.IsInRole($adminRole)) {
	Write-Host "Elevated permissions are required to run this script"
	Write-Host "Use an elevated command prompt to complete these tasks"
	exit 1
}

$transcribe = ($args.Count -gt 0)

if ($transcribe) {
    Start-Transcript -Path $args[0] -Force
}
try
{
    Write-Host "Checking Performance Counters"
    if (DoesCategoryExist) {
        $missing = $counters | ? { [System.Diagnostics.PerformanceCounterCategory]::CounterExists($_.CounterName, $categoryName) -eq $false } 
        if ($missing)
        {
            DeleteCategory
            AddCounters    
        }
        else
        {
            Write-Host -ForegroundColor Green "Counter are already installed"
        }
    } else {
        AddCounters
    }
}
finally {
    if ($transcribe) {
        Stop-Transcript
    }
}