feat: add allTenants support for transport rules page

Author: kris6673

Modules/CIPPCore/Public/Entrypoints/Activity Triggers/Push-ListTransportRulesAllTenants.ps1

@@ -0,0 +1,42 @@
+function Push-ListTransportRulesAllTenants {
+    <#
+    .FUNCTIONALITY
+        Entrypoint
+    #>
+    param($Item)
+
+    $Tenant = Get-Tenants -TenantFilter $Item.customerId
+    $DomainName = $Tenant.defaultDomainName
+    $Table = Get-CIPPTable -TableName CacheTransportRules
+
+    try {
+        $TransportRules = New-ExoRequest -tenantid $DomainName -cmdlet 'Get-TransportRule'
+        $Results = foreach ($rule in $TransportRules) {
+            $GUID = (New-Guid).Guid
+            $Results = @{
+                TransportRule = [string]($rule | ConvertTo-Json -Depth 10)
+                RowKey        = [string]$GUID
+                PartitionKey  = 'TransportRule'
+                Tenant        = [string]$DomainName
+            }
+            Add-CIPPAzDataTableEntity @Table -Entity $Results -Force | Out-Null
+        }
+
+    } catch {
+        $GUID = (New-Guid).Guid
+        $ErrorText = ConvertTo-Json -InputObject @{
+            Tenant      = $DomainName
+            Name        = "Could not connect to Tenant: $($_.Exception.Message)"
+            State       = 'Error'
+            Priority    = 0
+            Description = "Error retrieving transport rules: $($_.Exception.Message)"
+        }
+        $Results = @{
+            TransportRule = [string]$ErrorText
+            RowKey        = [string]$GUID
+            PartitionKey  = 'TransportRule'
+            Tenant        = [string]$DomainName
+        }
+        Add-CIPPAzDataTableEntity @Table -Entity $Results -Force | Out-Null
+    }
+}

Modules/CIPPCore/Public/Entrypoints/HTTP Functions/Email-Exchange/Transport/Invoke-ListTransportRules.ps1

@@ -13,21 +13,76 @@ Function Invoke-ListTransportRules {
     $APIName = $Request.Params.CIPPEndpoint
     $Headers = $Request.Headers
     Write-LogMessage -headers $Headers -API $APIName -message 'Accessed this API' -Sev 'Debug'
-    $TenantFilter = $request.Query.tenantFilter
+
+    # Interact with query parameters or the body of the request.
+    $TenantFilter = $Request.Query.tenantFilter
 
     try {
-        $GraphRequest = New-ExoRequest -tenantid $TenantFilter -cmdlet 'Get-TransportRule'
+        $Results = if ($TenantFilter -ne 'AllTenants') {
+            # Single tenant functionality
+            New-ExoRequest -tenantid $TenantFilter -cmdlet 'Get-TransportRule'
+        } else {
+            # AllTenants functionality
+            $Table = Get-CIPPTable -TableName CacheTransportRules
+            $PartitionKey = 'TransportRule'
+            $Filter = "PartitionKey eq '$PartitionKey'"
+            $Rows = Get-CIPPAzDataTableEntity @Table -filter $Filter | Where-Object -Property Timestamp -GT (Get-Date).AddMinutes(-60)
+            $QueueReference = '{0}-{1}' -f $TenantFilter, $PartitionKey
+            $RunningQueue = Invoke-ListCippQueue | Where-Object { $_.Reference -eq $QueueReference -and $_.Status -notmatch 'Completed' -and $_.Status -notmatch 'Failed' }
+
+            # If a queue is running, we will not start a new one
+            if ($RunningQueue) {
+                $Metadata = [PSCustomObject]@{
+                    QueueMessage = 'Still loading transport rules for all tenants. Please check back in a few more minutes'
+                }
+            } elseif (!$Rows -and !$RunningQueue) {
+                # If no rows are found and no queue is running, we will start a new one
+                $TenantList = Get-Tenants -IncludeErrors
+                $Queue = New-CippQueueEntry -Name 'Transport Rules - All Tenants' -Link '/email/transport/list-rules?tenantFilter=AllTenants' -Reference $QueueReference -TotalTasks ($TenantList | Measure-Object).Count
+                $Metadata = [PSCustomObject]@{
+                    QueueMessage = 'Loading transport rules for all tenants. Please check back in a few minutes'
+                }
+                $InputObject = [PSCustomObject]@{
+                    OrchestratorName = 'TransportRuleOrchestrator'
+                    QueueFunction    = @{
+                        FunctionName = 'GetTenants'
+                        QueueId      = $Queue.RowKey
+                        TenantParams = @{
+                            IncludeErrors = $true
+                        }
+                        DurableName  = 'ListTransportRulesAllTenants'
+                    }
+                    SkipLog          = $true
+                }
+                Start-NewOrchestration -FunctionName 'CIPPOrchestrator' -InputObject ($InputObject | ConvertTo-Json -Depth 5 -Compress) | Out-Null
+            } else {
+                # Return cached data
+                $Rules = $Rows
+                foreach ($rule in $Rules) {
+                    $RuleObj = $rule.TransportRule | ConvertFrom-Json
+                    $RuleObj | Add-Member -MemberType NoteProperty -Name Tenant -Value $rule.Tenant -Force
+                    $RuleObj
+                }
+            }
+        }
         $StatusCode = [HttpStatusCode]::OK
     } catch {
         $ErrorMessage = Get-NormalizedError -Message $_.Exception.Message
         $StatusCode = [HttpStatusCode]::Forbidden
-        $GraphRequest = $ErrorMessage
+        $Body = $ErrorMessage
+    }
+
+    # If the body is not set by an error, we will set it here
+    if (!$Body) {
+        $Body = [PSCustomObject]@{
+            Results  = @($Results)
+            Metadata = $Metadata
+        }
     }
 
     # Associate values to output bindings by calling 'Push-OutputBinding'.
     Push-OutputBinding -Name Response -Value ([HttpResponseContext]@{
             StatusCode = $StatusCode
-            Body       = @($GraphRequest)
+            Body       = $Body
         })
-
 }