feat: enhance backup restoration process to filter by selected types and count restored rows

Author: JohnDuprey

Modules/CIPPCore/Public/Entrypoints/HTTP Functions/CIPP/Settings/Invoke-ExecRestoreBackup.ps1

@@ -29,34 +29,50 @@ function Invoke-ExecRestoreBackup {
                     throw "Failed to parse backup JSON: $($_.Exception.Message)"
                 }
 
+                $SelectedTypes = $Request.Body.SelectedTypes
+                if ($SelectedTypes -and $SelectedTypes.Count -gt 0) {
+                    $BackupData = $BackupData | Where-Object {
+                        $item = $_
+                        if ($item.table -eq 'templates') {
+                            $typeKey = "templates:$($item.PartitionKey)"
+                        } else {
+                            $typeKey = $item.table
+                        }
+                        $SelectedTypes -contains $typeKey
+                    }
+                }
+                $RestoredCount = 0
                 $BackupData | ForEach-Object {
                     $Table = Get-CippTable -tablename $_.table
                     $ht2 = @{}
                     $_.psobject.properties | ForEach-Object { $ht2[$_.Name] = [string]$_.Value }
                     $Table.Entity = $ht2
-                    Add-CIPPAzDataTableEntity @Table -Force
+                    Add-AzDataTableEntity @Table -Force
+                    $RestoredCount++
                 }
-                Write-LogMessage -headers $Request.Headers -API $APINAME -message "Restored backup $($Request.Body.BackupName)" -Sev 'Info'
+                Write-LogMessage -headers $Request.Headers -API $APINAME -message "Restored backup $($Request.Body.BackupName) - $RestoredCount rows restored" -Sev 'Info'
                 $body = [pscustomobject]@{
-                    'Results' = 'Successfully restored backup.'
+                    'Results' = "Successfully restored $RestoredCount rows from backup."
                 }
             } else {
                 $body = [pscustomobject]@{
                     'Results' = 'Backup not found.'
                 }
             }
         } else {
+            $RestoredCount = 0
             foreach ($line in ($Request.body | Select-Object * -ExcludeProperty ETag, Timestamp)) {
                 $Table = Get-CippTable -tablename $line.table
                 $ht2 = @{}
                 $line.psobject.properties | ForEach-Object { $ht2[$_.Name] = [string]$_.Value }
                 $Table.Entity = $ht2
                 Add-AzDataTableEntity @Table -Force
+                $RestoredCount++
             }
-            Write-LogMessage -headers $Request.Headers -API $APINAME -message "Restored backup $($Request.Body.BackupName)" -Sev 'Info'
+            Write-LogMessage -headers $Request.Headers -API $APINAME -message "Restored backup - $RestoredCount rows restored" -Sev 'Info'
 
             $body = [pscustomobject]@{
-                'Results' = 'Successfully restored backup.'
+                'Results' = "Successfully restored $RestoredCount rows from backup."
             }
         }
     } catch {