#!/bin/bash

printf "Fetching logs of instances in resource group %s\n" $1

# get list of all scale sets
scalesetsjson=$(az vmss list --resource-group $1 -o json)
scalesetslist=$(echo $scalesetsjson | jq -r '.[] | .name')
subscription=$(az account  show | jq -r .id)

printf "Checking scalesets %s\n" $scalesetslist

for scaleset in $scalesetslist; do
    instanceids=$(az vmss list-instances --resource-group $1 --name ${scaleset} -o json | jq -r '.[] | .instanceId')
    printf "Checking instance IDs %s\n" $instanceids
    for instanceid in $instanceids; do
        bloburi=$(az rest --method post --url https://management.azure.com/subscriptions/${subscription}/resourceGroups/${1}/providers/Microsoft.Compute/virtualMachineScaleSets/${scaleset}/virtualmachines/$instanceid/retrieveBootDiagnosticsData?api-version=2022-03-01 | jq '.serialConsoleLogBlobUri' -r)
        sleep 4
        curl -sL -o "./${scaleset}-${instanceid}.log" $bloburi
        echo $(realpath "./${scaleset}-${instanceid}.log")
    done
done