private/core/Invoke-ZtGraphRequestCache.ps1

<#
.SYNOPSIS
    Enhanced version of Invoke-MgGraphRequest that supports caching.
#>

Function Invoke-ZtGraphRequestCache {
    [CmdletBinding()]
    param(
        [Parameter(Mandatory = $true)]
        [Uri] $Uri,

        [Parameter(Mandatory = $false)]
        [string] $Method = 'GET',

        [Parameter(Mandatory = $false)]
        [string] $OutputType,

        [Parameter(Mandatory = $false)]
        [System.Collections.IDictionary] $Headers,

        # Specify if this request should skip cache and go directly to Graph.
        [Parameter(Mandatory = $false)]
        [switch] $DisableCache,
        [string] $Body,

        # If specified, writes the raw results to disk
        [Parameter(Mandatory = $false)]
        [string] $OutputFilePath,

        # The index of Graph request page. If not specified, the first page is assumed (0).
        [Parameter(Mandatory = $false)]
        [string]$PageIndex
    )

    $results = $null
    $isBatch = $uri.AbsoluteUri.EndsWith('$batch')
    $isInCache = $__ZtSession.GraphCache.ContainsKey($Uri.AbsoluteUri)
    $cacheKey = $Uri.AbsoluteUri
    $isMethodGet = $Method -eq 'GET'

    if (!$DisableCache -and !$isBatch -and $isInCache -and $isMethodGet) {
        # Don't read from cache for batch requests.
        Write-PSFMessage "Using graph cache: $($cacheKey)" -Level Debug
        $results = $__ZtSession.GraphCache[$cacheKey]
    }

    if (!$results) {
        Write-PSFMessage "Invoking Graph: $($Uri.AbsoluteUri)" -Level Debug -Tag Graph
        Write-PSFMessage ([string]::IsNullOrEmpty($Body)) -Level Debug -Tag Graph


        if ($Method -eq 'GET') {
            if ($OutputFilePath) {
                $OutputType = 'Json' # Force JSON output if writing to file so we get the raw results
            }

            $results = Invoke-MgGraphRequest -Method $Method -Uri $Uri -Headers $Headers -OutputType $OutputType # -Body $Body # Cannot use Body with GET in PS 5.1
            if ($OutputFilePath) {
                $filePath = Get-ExportJsonFilePath -Path $OutputFilePath -PageIndex $PageIndex
                $results | Out-File (New-Item -Path $filePath -Force) # Write raw results to disk
                $results = $results | ConvertFrom-Json -depth 100 # Convert back to PSObject
            }
        }
        else {
            $results = Invoke-MgGraphRequest -Method $Method -Uri $Uri -Headers $Headers -OutputType $OutputType -Body $Body
        }

        if (!$isBatch -and $isMethodGet) {
            # Update cache
            if ($isInCache) {
                $__ZtSession.GraphCache[$cacheKey] = $results
            }
            else {
                $__ZtSession.GraphCache.Add($cacheKey, $results)
            }
        }
    }
    return $results
}