forked from sig_core/toolkit
69 lines
2.3 KiB
Bash
69 lines
2.3 KiB
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
# Source common variables
|
||
|
|
# shellcheck disable=SC2046,1091,1090
|
||
|
|
source "$(dirname "${BASH_SOURCE[0]}")/common"
|
||
|
|
|
||
|
|
usage() {
|
||
|
|
echo "usage: $0"
|
||
|
|
}
|
||
|
|
|
||
|
|
aws() {
|
||
|
|
# shellcheck disable=SC2068
|
||
|
|
command aws --profile resf-ami --output json $@
|
||
|
|
}
|
||
|
|
|
||
|
|
# Get the quota code for Public AMIs once
|
||
|
|
quota_code=$(aws service-quotas list-service-quotas --service-code ec2 --region us-east-1 --query "Quotas[*].{QuotaCode:QuotaCode,QuotaName:QuotaName}" | jq '.[] | select(.QuotaName == "Public AMIs") | .QuotaCode' | tr -d '"')
|
||
|
|
|
||
|
|
function get_current_quota() {
|
||
|
|
region=$1
|
||
|
|
# Get the current value of the quota
|
||
|
|
current_value=$(aws service-quotas get-service-quota --service-code ec2 --quota-code "$quota_code" --region "$region" 2>/dev/null | jq .Quota.Value 2>/dev/null)
|
||
|
|
# shellcheck disable=SC2181
|
||
|
|
if [[ $? -gt 0 ]]; then
|
||
|
|
echo "ERR"
|
||
|
|
fi
|
||
|
|
echo "$current_value"
|
||
|
|
}
|
||
|
|
|
||
|
|
function request_quota_increase() {
|
||
|
|
mapfile -t regions <<<"$@"
|
||
|
|
for region in "${regions[@]}"; do
|
||
|
|
# Get the current value of the quota
|
||
|
|
current_value=$(get_current_quota "$region")
|
||
|
|
if ((current_value >= QUOTA)); then
|
||
|
|
echo "Quota for Public AMIs in region $region is already set to $current_value, skipping request."
|
||
|
|
else
|
||
|
|
# Request the quota increase
|
||
|
|
request_output=$(aws service-quotas request-service-quota-increase --service-code ec2 --quota-code "$quota_code" --region "$region" --desired-value "$QUOTA")
|
||
|
|
request_id=$(echo "$request_output" | jq .RequestedQuota.Id | tr -d '"')
|
||
|
|
echo "Successfully submitted request with ID: $request_id"
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
function list_quotas() {
|
||
|
|
mapfile -t regions <<<"$@"
|
||
|
|
output="Region\tQuota\tDesired\tStatus"
|
||
|
|
for region in "${regions[@]}"; do
|
||
|
|
current_quota=$(get_current_quota "$region")
|
||
|
|
request_info=$(aws service-quotas list-requested-service-quota-change-history-by-quota --service-code ec2 --quota-code "$quota_code" --region "$region" --query "RequestedQuotas[-1:].{DesiredValue:DesiredValue,Status:Status}" 2>/dev/null)
|
||
|
|
requested_value=$(echo "$request_info" | jq .[].DesiredValue)
|
||
|
|
case_status=$(echo "$request_info" | jq .[].Status | tr -d '"')
|
||
|
|
output="$output\n$region $current_quota $requested_value $case_status"
|
||
|
|
done
|
||
|
|
echo -e "$output" | column -t
|
||
|
|
}
|
||
|
|
|
||
|
|
REGIONS=$(aws ec2 describe-regions \
|
||
|
|
--all-regions \
|
||
|
|
--query "Regions[].{Name:RegionName}" \
|
||
|
|
--output text)
|
||
|
|
|
||
|
|
QUOTA=50
|
||
|
|
|
||
|
|
list_quotas "$REGIONS"
|
||
|
|
|
||
|
|
# request_quota_increase "$REGIONS"
|