manual pagination support for Invoke-ListMailQuarantine

Author: Zacgoose

Modules/CIPPHTTP/Public/Entrypoints/HTTP Functions/Email-Exchange/Spamfilter/Invoke-ListMailQuarantine.ps1

@@ -14,15 +14,27 @@ function Invoke-ListMailQuarantine {
 
     try {
         $GraphRequest = if ($TenantFilter -ne 'AllTenants') {
-            $Page = 1
             $PageSize = 1000
-            $AllMessages = [System.Collections.Generic.List[object]]::new()
-            do {
+            if ($Request.Query.manualPagination -and [System.Convert]::ToBoolean($Request.Query.manualPagination)) {
+                # Manual pagination: return one page per request. The frontend chains requests via
+                # Metadata.nextLink, which for this endpoint is the next Get-QuarantineMessage page number.
+                $Page = if ($Request.Query.nextLink -match '^\d+$') { [int]$Request.Query.nextLink } else { 1 }
                 $Results = New-ExoRequest -tenantid $TenantFilter -cmdlet 'Get-QuarantineMessage' -cmdParams @{ PageSize = $PageSize; Page = $Page } | Select-Object -ExcludeProperty *data.type*
-                if ($Results) { $AllMessages.AddRange(@($Results)) }
-                $Page++
-            } while (@($Results).Count -eq $PageSize)
-            $AllMessages
+                # Get-QuarantineMessage supports a maximum Page of 1000
+                if (@($Results).Count -eq $PageSize -and $Page -lt 1000) {
+                    $Metadata = [PSCustomObject]@{ nextLink = [string]($Page + 1) }
+                }
+                $Results
+            } else {
+                $Page = 1
+                $AllMessages = [System.Collections.Generic.List[object]]::new()
+                do {
+                    $Results = New-ExoRequest -tenantid $TenantFilter -cmdlet 'Get-QuarantineMessage' -cmdParams @{ PageSize = $PageSize; Page = $Page } | Select-Object -ExcludeProperty *data.type*
+                    if ($Results) { $AllMessages.AddRange(@($Results)) }
+                    $Page++
+                } while (@($Results).Count -eq $PageSize)
+                $AllMessages
+            }
         } else {
             $Table = Get-CIPPTable -TableName cacheQuarantineMessages
             $PartitionKey = 'QuarantineMessage'