M365Permissions

1.2.2

private/new-ExOQuery.ps1


function New-ExOQuery {
    <#
        Author = "Jos Lieben (jos@lieben.nu)"
        CompanyName = "Lieben Consultancy"
        Copyright = "https://www.lieben.nu/liebensraum/commercial-use/"
    #>        
    [CmdletBinding()]
    Param(
        [parameter(Mandatory = $True)]$cmdlet,
        $cmdParams,
        [Switch]$NoPagination,
        $retryCount = 3,
        [Array]$nonRetryErrors = @()       
    )
    $token = Get-AccessToken -Resource "https://$($global:octo.outlookUrl)"
    if ($cmdParams) {
        $Params = $cmdParams
    }else {
        $Params = @{}
    }
    
    $ExoBody = ConvertTo-Json -Depth 15 -InputObject @{
        CmdletInput = @{
            CmdletName = $cmdlet
            Parameters = $Params
        }
    } 

    $Headers = @{ 
        "Authorization" = "Bearer $token"
        "Accept-Charset" = "UTF-8"
        "X-ResponseFormat" = "json"
        "Accept" = "application/json"
        "X-ClientApplication" ="ExoManagementModule"
        "Prefer" = "odata.maxpagesize=1000"
        "X-CmdletName"= $cmdlet
        "X-SerializationLevel" = "Partial"
        'X-AnchorMailbox' = "UPN:SystemMailbox{bb558c35-97f1-4cb9-8ff7-d53741dc928c}@$($global:octo.OnMicrosoft)"
        "Content-Type" = "application/json"
    }

    $nextURL = "https://$($global:octo.outlookUrl)/adminapi/beta/$($global:octo.OnMicrosoft)/InvokeCommand"

    $ReturnedData = do {
        try {
            $attempts = 0
            while ($attempts -lt $retryCount) {
                try {
                    $Data = Invoke-RestMethod -Uri $nextURL -Method POST -Body $ExoBody -Headers $Headers -Verbose:$false         
                    $attempts = $retryCount
                }catch {
                    $attempts++
                    if($_.Exception.Message -like "*404 (Not Found)*"){
                        Write-LogMessage -level 6 -message "Not retrying: $($_)"
                        $nextUrl = $Null
                        throw $_
                    }                    
                    foreach($nonRetryError in $nonRetryErrors){
                        if($_.Exception.Message -like "*$nonRetryError*"){
                            Write-LogMessage -level 5 -message "EXO request failed, non-retryable error: $($_.Exception.Message)"
                            $nextUrl = $null
                            throw $_
                        }
                    }                    
                    if ($attempts -eq $retryCount) {
                        $nextUrl = $null
                        Throw $_
                    }
                    $sleepTime = $attempts * 2
                    Write-LogMessage -level 5 -message "EXO request failed, sleeping for $sleepTime seconds..."
                    Start-Sleep -Seconds $sleepTime
                }
            }
            if($NoPagination){
                $nextURL = $null
            }elseif($Data.'@odata.nextLink'){
                $nextURL = $Data.'@odata.nextLink'  
            }elseif($Data.'odata.nextLink'){
                $nextURL = $Data.'odata.nextLink'  
            }else{
                $nextURL = $null 
            } 
            if($Data.psobject.properties.name -icontains 'value' -or $Data.Keys -icontains 'value'){
                ($Data.value)
            }else{         
                ($Data)
            }         
        }catch {
            throw $_
        }
    }until($null -eq $nextURL)

    [System.GC]::GetTotalMemory($true) | out-null
    return $ReturnedData
}