BcContainerHelper

2.0.16-preview468

TelemetryHelper.ps1

function FormatValue {
    Param(
        $value
    )

    if ($value -is [switch]) {
        "$($value.IsPresent)"
    }
    elseif ($value -is [boolean]) {
        "$value"
    }
    elseif ($value -is [SecureString]) {
        "[SecureString]"
    }
    elseif ($value -is [PSCredential]) {
        "[PSCredential]"
    }
    elseif ($value -is [string]) {
        if (($value -like "https:*" -or $value -like "http:*") -and ($value.Contains('?'))) {
            """$($value.Split('?')[0])?[parameters]"""
        }
        else {
            """$value"""
        }
    }
    elseif ($value -is [System.Collections.IDictionary]) {
        $str = ""
        $value.GetEnumerator() | ForEach-Object {
            if ($str) { $str += ", " } else { $str = "{ " }
            $str += "$($_.Key): $(FormatValue -value $_.Value)"
        }
        "$str }"
    }
    elseif ($value -is [System.Collections.IEnumerable]
    ) {
        $str = ""
        $value.GetEnumerator() | ForEach-Object {
            if ($str) { $str += ", " } else { $str = "[ " }
            $str += "$(FormatValue -value $_)"
        }
        "$str ]"
    }
    else {
        "$value"
    }
}

function AddTelemetryProperty {
    Param(
        $telemetryScope,
        $key,
        $value
    )

    if ($telemetryScope) {
# Write-Host "Telemetry scope $($telemetryScope.Name), add property $key = $((FormatValue -value $value))"
        if ($telemetryScope.properties.ContainsKey($Key)) {
            $telemetryScope.properties."$key" += "`n$(FormatValue -value $value)"
        }
        else {
            $telemetryScope.properties.Add($key, (FormatValue -value $value))
        }
    }
}

function InitTelemetryScope {
    Param(
        [string] $name,
        [string[]] $includeParameters = @(),
        $parameterValues = $null
    )
    if ($telemetryClient) {
        if ($bcContainerHelperConfig.TelemetryConnectionString) {
            if ($telemetryClient.TelemetryConfiguration.DisableTelemetry -or $telemetryClient.TelemetryConfiguration.ConnectionString -ne $bcContainerHelperConfig.TelemetryConnectionString) {
                if ($bcContainerHelperConfig.TelemetryConnectionString) {
                    try {
                        $telemetryClient.TelemetryConfiguration.ConnectionString = $bcContainerHelperConfig.TelemetryConnectionString
                        $telemetryClient.TelemetryConfiguration.DisableTelemetry = $false
                        Write-Host "Telemetry client initialized"
                    }
                    catch {
                        $telemetryClient.TelemetryConfiguration.DisableTelemetry = $true
                    }
                }
            }
            if ($telemetryClient.IsEnabled()) {
                Write-Host "Init telemetry scope $name"
                $scope = @{
                    "Name" = $name
                    "StartTime" = [DateTime]::Now
                    "SeverityLevel" = 1
                    "Properties" = [Collections.Generic.Dictionary[string, string]]::new()
                    "Emitted" = $false
                }
                if ($includeParameters) {
                    $parameterValues.GetEnumerator() | % {
                        $includeParameter = $false
                        $key = $_.key
                        $includeParameters | ForEach-Object { if ($key -like $_) { $includeParameter = $true } }
                        if ($includeParameter) {
                            AddTelemetryProperty -telemetryScope $scope -key "Parameter[$Key]" -value $_.Value
                        }
                    }
                }
                AddTelemetryProperty -telemetryScope $scope -key "BcContainerHelperVersion" -value $BcContainerHelperVersion
                AddTelemetryProperty -telemetryScope $scope -key "IsAdministrator" -value $isAdministrator
                AddTelemetryProperty -telemetryScope $scope -key "StackTrace" -value (Get-PSCallStack | % { "$($_.Command) at $($_.Location)" }) -join "`n"
                $scope
            }
        }
        else {
            $telemetryClient.TelemetryConfiguration.DisableTelemetry = $true
        }
    }
}

function TrackTrace {
    Param(
        $telemetryScope
    )

    if ($telemetryScope -and !$telemetryScope.Emitted) {
        if ($telemetryClient.IsEnabled()) {
            Write-Host "Emit telemetry trace, scope $($telemetryScope.Name)"
            $telemetryScope.Properties.Add("Duration", [DateTime]::Now.Subtract($telemetryScope.StartTime).TotalSeconds)
            $telemetryClient.TrackTrace($telemetryScope.Name, $telemetryScope.SeverityLevel, $telemetryScope.Properties)
            $telemetryScope.Emitted = $true
        }
    }

}

function TrackException {
    Param(
        $telemetryScope,
        $errorRecord
    )

    if ($telemetryScope -and !$telemetryScope.Emitted) {
        if ($telemetryClient.IsEnabled()) {
            Write-Host "Emit telemetry exception, scope $($telemetryScope.Name)"
            $telemetryScope.Properties.Add("Duration", [DateTime]::Now.Subtract($telemetryScope.StartTime).TotalSeconds)
            $telemetryScope.Properties.Add("StackTrace", $errorRecord.ScriptStackTrace)
            $telemetryClient.TrackException($errorRecord.Exception, $telemetryScope.Properties)
            $telemetryScope.Emitted = $true
        }
    }
}