zsh utility function for selecting AWS profile to perform sso login on

aws_sso.zsh

# placed inside ~/.oh-my-zsh/custom/aliases.zsh
# requires fzf
sso() {
  local profile_file="$HOME/.aws/config"
  local profiles

  # Check if ~/.aws/config exists
  if [[ ! -f $profile_file ]]; then
    echo "AWS config file not found: $profile_file"
    return 1
  fi

  # Extract AWS profiles and strip the '[profile ]' prefix
  profiles=$(grep -E '^\[profile' "$profile_file" | sed -E 's/^\[profile (.*)\]/\1/')

  # Check if any profiles were found
  if [[ -z $profiles ]]; then
    echo "No profiles found in $profile_file"
    return 1
  fi

  # Use fzf to select a profile
  local profile
  profile=$(echo "$profiles" | fzf --prompt="Select an AWS Profile: " --height=10 --border)

  # If a profile is selected, run the aws sso login command
  if [[ -n $profile ]]; then
    echo "Logging in with profile: $profile"
    aws sso login --profile "$profile"
  else
    echo "No profile selected."
  fi
}