# Connect to Power BI Service
Connect-PowerBIServiceAccount

# Retrieve all workspaces
$Workspaces = Get-PowerBIWorkspace

# Specify the workspace to filter by name
$WorkspaceFilterNames = 'Data'

# Loop through the Workspaces filtering by the necessary workspace names
foreach($Workspace in $Workspaces) {
    if($Workspace.Name -eq $WorkspaceFilterNames) {
        # Define the directory where the dataflows will be backed up
        $Folder = 'C:\Users\CurrentUser\ Data Crafters Build\DFBackup'
        $FolderName = $Workspace.Name

        # Check if the folder already exists
        if (Test-Path -Path "$Folder\$FolderName") {
            # If the folder exists, append the current date and time to the folder name
            $timestamp = Get-Date -Format "yyyy-MM-dd_HH-mm-ss"
            $NewFolderName = "$FolderName-$timestamp"
            $FolderName = New-Item -Path $Folder -Name $NewFolderName -ItemType "directory"
        } else {
            # If the folder does not exist, create it normally
            $FolderName = New-Item -Path $Folder -Name $FolderName -ItemType "directory"
        }

        # Get dataflows for the current workspace
        $Dataflows = Get-PowerBIDataflow -WorkspaceId $Workspace.Id

        # Loop through each dataflow
        foreach($Dataflow in $Dataflows) {
            Write-Host "Exporting Dataflow: " $Dataflow.Name "from Workspace: " $Workspace.Name

            # Throttle requests to avoid hitting API limits
            Start-Sleep -Seconds 4

            # Construct the export filename and path
            $ExportFileName = Join-Path $FolderName -ChildPath ($Dataflow.Name + ".json")

            # Export the dataflow to a JSON file
            Export-PowerBIDataflow -WorkspaceId $Workspace.Id -Id $Dataflow.Id -Scope Individual -OutFile $ExportFileName

            # Load the exported JSON, modify it and save it
            $JSONFile = Get-Content -Path $ExportFileName | ConvertFrom-Json
            $JSONFile.'pbi:mashup'.allowNativeQueries = $false
            $JSONToWrite = $JSONFile | ConvertTo-Json -Depth 100
            Set-Content -Path $ExportFileName -Value $JSONToWrite

            # Prepare the multipart/form-data body for REST API submission
            $UID = [System.Guid]::NewGuid().ToString("N")
            $LF = "`r`n";
            $Body = (
                "--$UID",
                "Content-Disposition: form-data; name=`"model.json`"; filename=`"model.json`"",
                "Content-Type: application/json$LF",
                $JSONFile,
                "--$UID--$LF"
            ) -join $LF

            $Header = @{
                Authorization = "$AccessToken"
                'Content-Type'  = "multipart/form-data; boundary=--$UID"
            }

            # Uncomment the line below to execute the REST API call
            #Invoke-RestMethod -Uri $InvokeURL -ContentType 'multipart/form-data' -Method POST -Headers $Header -Body $Body
        }
    }
}