feat: Get-WebSocket -PSTypeName ( Fixes #34 )

PowerShellWeb · Dec 7, 2024 · 8c1c4de · 8c1c4de
1 parent b53081d
commit 8c1c4de
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/Commands/Get-WebSocket.ps1 b/Commands/Get-WebSocket.ps1
@@ -173,6 +173,12 @@ function Get-WebSocket {
     [TimeSpan]
     $TimeOut,
 
+    # If provided, will decorate the objects outputted from a websocket job.
+    # This will only decorate objects converted from JSON.
+    [Alias('PSTypeNames','Decorate','Decoration')]
+    [string[]]
+    $PSTypeName,
+
     # The maximum number of messages to receive before closing the WebSocket.
     [long]
     $Maximum,
@@ -256,6 +262,13 @@ function Get-WebSocket {
                             if ([string]::IsNullOrWhitespace($JS)) { continue }
                             ConvertFrom-Json $JS
                         }
+                    if ($PSTypeName) {
+                        $webSocketMessage.pstypenames.clear()
+                        [Array]::Reverse($PSTypeName)
+                        foreach ($psType in $psTypeName) {
+                            $webSocketMessage.pstypenames.add($psType)
+                        }
+                    }
                     if ($handler) {
                         $psCmd =  
                             if ($runspace.LanguageMode -eq 'NoLanguage' -or