From defcf56c82e981e2a761ebc1c8b8ed2b8b482abf Mon Sep 17 00:00:00 2001
From: DrIOS <58635327+DrIOSX@users.noreply.github.com>
Date: Sun, 23 Jun 2024 17:00:12 -0500
Subject: [PATCH] add: DomainName paramter to get-mggraph and test 1.3.1

---
 source/Private/Get-CISMgOutput.ps1 | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/source/Private/Get-CISMgOutput.ps1 b/source/Private/Get-CISMgOutput.ps1
index 9a02b9f..46f9ff7 100644
--- a/source/Private/Get-CISMgOutput.ps1
+++ b/source/Private/Get-CISMgOutput.ps1
@@ -14,11 +14,11 @@ function Get-CISMgOutput {
 
 #>
     [cmdletBinding()]
-    [OutputType([string])]
     param(
         [Parameter(Mandatory = $true)]
-        [String]
-        $Rec
+        [String]$Rec,
+        [Parameter(Mandatory = $false)]
+        [String]$DomainName
     )
 
     begin {
@@ -56,6 +56,15 @@ function Get-CISMgOutput {
                 $allGroups = Get-MgGroup -All | Where-Object { $_.Visibility -eq "Public" } | Select-Object DisplayName, Visibility
                 return $allGroups
             }
+            '1.3.1' {
+                # Test-PasswordNeverExpirePolicy.ps1
+                $domains = if ($DomainName) {
+                    Get-MgDomain -DomainId $DomainName
+                } else {
+                    Get-MgDomain
+                }
+                return $domains
+            }
             '5.1.2.3' {
                 # Retrieve the tenant creation policy
                 $tenantCreationPolicy = (Get-MgPolicyAuthorizationPolicy).DefaultUserRolePermissions | Select-Object AllowedToCreateTenants