VirtualMetric

2.3.0

VirtualMetric.psm1

# The VirtualMetric module. The runtime is hand-written (Private, Public); the resource cmdlets are generated
# from the API catalog (Generated/Cmdlets.ps1, written by go run ./cmd/pwshgen) as thin stubs over
# Invoke-VMetricCatalogCmdlet, which reads VirtualMetric.Catalog.json.

$script:ModuleRoot = $PSScriptRoot
# While the root module loads, the module object does not carry the manifest's version yet: read it from the
# manifest itself.
$script:ModuleVersion = '0.0.0'
$manifestPath = Join-Path -Path $PSScriptRoot -ChildPath 'VirtualMetric.psd1'
if (Test-Path -LiteralPath $manifestPath) {
    $script:ModuleVersion = [string](Import-PowerShellDataFile -LiteralPath $manifestPath).ModuleVersion
}

# The current session (a synchronized hashtable, see Private/Session.ps1), every session this PowerShell
# session holds keyed by origin and tenant, and the catalog once read.
$script:VMetricSession = $null
$script:VMetricSessions = [ordered]@{}
$script:VMetricCatalog = $null
$script:VMetricCatalogPath = $null
$script:VMetricUserAgent = $null
# The transport's HttpClients, one per certificate policy (see Private/Http.ps1).
$script:VMetricHttpClients = @{}

foreach ($folder in @('Private', 'Public')) {
    $folderPath = Join-Path -Path $PSScriptRoot -ChildPath $folder
    if (-not (Test-Path -LiteralPath $folderPath)) {
        continue
    }
    foreach ($file in (Get-ChildItem -LiteralPath $folderPath -Filter '*.ps1' -File | Sort-Object -Property Name)) {
        . $file.FullName
    }
}

$generatedPath = Join-Path -Path $PSScriptRoot -ChildPath 'Generated/Cmdlets.ps1'
if (Test-Path -LiteralPath $generatedPath) {
    . $generatedPath
}

# Tab after -Field offers the fields of the resource the command names (Private/Transforms.ps1).
Register-VMetricFieldCompleter

# Removing the module forgets every session and zeroes the tokens it held, and closes the transport's
# connections. Nothing is revoked on the server: Disconnect-VMetric does that.
$ExecutionContext.SessionState.Module.OnRemove = {
    Clear-VMetricSessionState
    foreach ($client in @($script:VMetricHttpClients.Values)) {
        $client.Dispose()
    }
    $script:VMetricHttpClients.Clear()
}