1#!/usr/bin/env bash
2# check-quota.sh
3# Checks Azure quota limits and current usage for a resource provider.
4# Returns a table with limit, usage, and available capacity for every quota
5# (or a single quota when resource-name is provided).
6#
7# Usage:
8#   ./check-quota.sh <resource-provider> <region> [resource-name] [subscription-id]
9#
10# Examples:
11#   ./check-quota.sh Microsoft.Compute eastus                          # All compute quotas with usage
12#   ./check-quota.sh Microsoft.Compute eastus standardDSv3Family       # Single quota with usage
13 
14set -euo pipefail
15 
16RESOURCE_PROVIDER="${1:?Usage: $0 <resource-provider> <region> [resource-name] [subscription-id]}"
17REGION="${2:?Usage: $0 <resource-provider> <region> [resource-name] [subscription-id]}"
18RESOURCE_NAME="${3:-}"
19SUBSCRIPTION_ID="${4:-}"
20 
21# Ensure the quota extension is installed
22if ! az extension list --query "[?name=='quota'].name" -o tsv 2>/dev/null | grep -q quota; then
23    echo "Installing quota extension..."
24    az extension add --name quota --yes 2>/dev/null
25fi
26 
27# Resolve subscription
28if [ -z "$SUBSCRIPTION_ID" ]; then
29    SUBSCRIPTION_ID=$(az account show --query id -o tsv)
30fi
31 
32SCOPE="/subscriptions/$SUBSCRIPTION_ID/providers/$RESOURCE_PROVIDER/locations/$REGION"
33 
34echo "Checking quotas in scope $SCOPE"
35 
36if [ -n "$RESOURCE_NAME" ]; then
37    # Single-resource mode
38    echo "Quota for '$RESOURCE_NAME' ($RESOURCE_PROVIDER, $REGION):"
39    echo ""
40 
41    LIMIT=$(az quota show \
42        --resource-name "$RESOURCE_NAME" \
43        --scope "$SCOPE" \
44        --query "properties.limit.value" -o tsv)
45 
46    USAGE=$(az quota usage show \
47        --resource-name "$RESOURCE_NAME" \
48        --scope "$SCOPE" \
49        --query "properties.usages.value" -o tsv)
50 
51    AVAILABLE=$((LIMIT - USAGE))
52 
53    printf "%-30s %-10s %-10s %-10s %-10s\n" "Resource" "Region" "Limit" "Usage" "Available"
54    printf "%-30s %-10s %-10s %-10s %-10s\n" "--------" "------" "-----" "-----" "---------"
55    printf "%-30s %-10s %-10s %-10s %-10s\n" "$RESOURCE_NAME" "$REGION" "$LIMIT" "$USAGE" "$AVAILABLE"
56else
57    # All-quotas mode: fetch limits and usage, join by name
58    echo "Quotas for $RESOURCE_PROVIDER in $REGION:"
59    echo ""
60 
61    QUOTAS_JSON=$(az quota list --scope "$SCOPE" -o json 2>/dev/null)
62    USAGES_JSON=$(az quota usage list --scope "$SCOPE" -o json 2>/dev/null)
63 
64    printf "%-40s %-10s %-10s %-10s %-10s\n" "Resource" "Region" "Limit" "Usage" "Available"
65    printf "%-40s %-10s %-10s %-10s %-10s\n" "--------" "------" "-----" "-----" "---------"
66 
67    echo "$QUOTAS_JSON" | python3 -c "
68import json, sys
69 
70quotas = json.load(sys.stdin)
71usages = json.loads('''$USAGES_JSON''')
72 
73usage_lookup = {}
74for u in usages:
75    usage_lookup[u['name']] = u.get('properties', {}).get('usages', {}).get('value', 0)
76 
77for q in quotas:
78    name = q['name']
79    limit = q.get('properties', {}).get('limit', {}).get('value', 0)
80    used = usage_lookup.get(name, 0)
81    avail = limit - used
82    print(f'{name:<40} $REGION{\"\":<4} {limit:<10} {used:<10} {avail:<10}')
83"
84fi
85