[CmdletBinding()]
param (
    [Parameter(Mandatory = $true, Position = 0)]
    [System.Net.WebSockets.ClientWebSocket]
    $WebSocket
)

if ($WebSocket.State -eq [System.Net.WebSockets.WebSocketState]::Closed) {
    return;
}
if ($WebSocket.State -ne [System.Net.WebSockets.WebSocketState]::Open) {
    Write-Error "Cannot disconnect when $($WebSocket.State)";
    return;
}

$WebSocket.CloseAsync([System.Net.WebSockets.WebSocketCloseStatus]::NormalClosure, "", [System.Threading.CancellationToken]::None).GetAwaiter().GetResult() | Out-Null;
$WebSocket.Dispose();