From 8b3d4f877bf2d75bd58507a4a4b9421cbb4e7ad7 Mon Sep 17 00:00:00 2001
From: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>
Date: Mon, 26 Aug 2024 21:50:04 +0300
Subject: [PATCH] Add option to return PSCustomObjects

This change adds a new option to return PSCustomObjects instead of
hashtables. The same option makes powershell-yaml output arrays
instead of generic lists.

Signed-off-by: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>
---
 powershell-yaml.psm1 | 59 +++++++++++++++++++++++++++++++++++---------
 1 file changed, 47 insertions(+), 12 deletions(-)

diff --git a/powershell-yaml.psm1 b/powershell-yaml.psm1
index 1b2f5d1..1e3d0a3 100644
--- a/powershell-yaml.psm1
+++ b/powershell-yaml.psm1
@@ -233,17 +233,36 @@ function Convert-ValueToProperType {
     }
 }
 
+function Convert-YamlMappingToPScustomObject {
+    [CmdletBinding()]
+    Param(
+        [Parameter(Mandatory=$true, ValueFromPipeline=$true)]
+        $Node,
+        [switch] $Ordered,
+        [switch] $AsHashtable
+    )
+    PROCESS {
+        $ret = New-Object -TypeName PSCustomObject
+        foreach($i in $Node.Children.Keys) {
+            $val = Convert-YamlDocumentToPSObject $Node.Children[$i] -Ordered:$Ordered -AsHashtable:$AsHashtable
+            $ret | Add-Member -MemberType NoteProperty -Name $i.Value -Value $val
+        }
+        return $ret
+    }
+}
+
 function Convert-YamlMappingToHashtable {
     [CmdletBinding()]
     Param(
         [Parameter(Mandatory=$true, ValueFromPipeline=$true)]
         $Node,
-        [switch] $Ordered
+        [switch] $Ordered,
+        [switch] $AsHashtable
     )
     PROCESS {
         if ($Ordered) { $ret = [ordered]@{} } else { $ret = @{} }
         foreach($i in $Node.Children.Keys) {
-            $ret[$i.Value] = Convert-YamlDocumentToPSObject $Node.Children[$i] -Ordered:$Ordered
+            $ret[$i.Value] = Convert-YamlDocumentToPSObject $Node.Children[$i] -Ordered:$Ordered -AsHashtable:$AsHashtable
         }
         return $ret
     }
@@ -254,12 +273,23 @@ function Convert-YamlSequenceToArray {
     Param(
         [Parameter(Mandatory=$true, ValueFromPipeline=$true)]
         $Node,
-        [switch]$Ordered
+        [switch]$Ordered,
+        [switch]$AsHashtable
     )
     PROCESS {
-        $ret = [System.Collections.Generic.List[object]](New-Object "System.Collections.Generic.List[object]")
+        if ($AsHashtable) {
+            $ret = [System.Collections.Generic.List[object]](New-Object "System.Collections.Generic.List[object]")
+        } else {
+            $ret = @()
+        }
         foreach($i in $Node.Children){
-            $ret.Add((Convert-YamlDocumentToPSObject $i -Ordered:$Ordered))
+            if ($AsHashtable) {
+                $val = Convert-YamlDocumentToPSObject $i -Ordered:$Ordered -AsHashtable:$AsHashtable
+                $ret.Add($val)
+            } else {
+                $val = Convert-YamlDocumentToPSObject $i -Ordered:$Ordered -AsHashtable:$AsHashtable
+                $ret += $val
+            }
         }
         return ,$ret
     }
@@ -270,15 +300,19 @@ function Convert-YamlDocumentToPSObject {
     Param(
         [Parameter(Mandatory=$true, ValueFromPipeline=$true)]
         [System.Object]$Node, 
-        [switch]$Ordered
+        [switch]$Ordered,
+        [switch]$AsHashtable=$false
     )
     PROCESS {
         switch($Node.GetType().FullName){
             "YamlDotNet.RepresentationModel.YamlMappingNode"{
-                return Convert-YamlMappingToHashtable $Node -Ordered:$Ordered
+                if ($AsHashtable) {
+                    return Convert-YamlMappingToHashtable $Node -Ordered:$Ordered -AsHashtable:$AsHashtable
+                }
+                return Convert-YamlMappingToPScustomObject $Node -Ordered:$Ordered -AsHashtable:$AsHashtable
             }
             "YamlDotNet.RepresentationModel.YamlSequenceNode" {
-                return Convert-YamlSequenceToArray $Node -Ordered:$Ordered
+                return Convert-YamlSequenceToArray $Node -Ordered:$Ordered -AsHashtable:$AsHashtable
             }
             "YamlDotNet.RepresentationModel.YamlScalarNode" {
                 return (Convert-ValueToProperType $Node)
@@ -349,7 +383,8 @@ function ConvertFrom-Yaml {
         [string]$Yaml,
         [switch]$AllDocuments=$false,
         [switch]$Ordered,
-        [switch]$UseMergingParser=$false
+        [switch]$UseMergingParser=$false,
+        [switch]$AsHashtable=$true
     )
 
     BEGIN {
@@ -370,14 +405,14 @@ function ConvertFrom-Yaml {
             return
         }
         if($documents.Count -eq 1){
-            return Convert-YamlDocumentToPSObject $documents[0].RootNode -Ordered:$Ordered
+            return Convert-YamlDocumentToPSObject $documents[0].RootNode -Ordered:$Ordered -AsHashtable:$AsHashtable
         }
         if(!$AllDocuments) {
-            return Convert-YamlDocumentToPSObject $documents[0].RootNode -Ordered:$Ordered
+            return Convert-YamlDocumentToPSObject $documents[0].RootNode -Ordered:$Ordered -AsHashtable:$AsHashtable
         }
         $ret = @()
         foreach($i in $documents) {
-            $ret += Convert-YamlDocumentToPSObject $i.RootNode -Ordered:$Ordered
+            $ret += Convert-YamlDocumentToPSObject $i.RootNode -Ordered:$Ordered -AsHashtable:$AsHashtable
         }
         return $ret
     }