#!/usr/bin/env bash

# Retry a command with exponential backoff.

#

# Usage: retry.sh [--max-attempts N] -- <command...>

#

# Example:

# retry.sh --max-attempts 3 -- go install gotest.tools/gotestsum@latest

#

# This will retry the command up to 3 times with exponential backoff

# (2s, 4s, 8s delays between attempts).

set -euo pipefail

# shellcheck source=scripts/lib.sh

source "$(dirname "${BASH_SOURCE[0]}")/../../scripts/lib.sh"

max_attempts=3

args="$(getopt -o "" -l max-attempts: -- "$@")"

eval set -- "$args"

while true; do

case "$1" in

--max-attempts)

max_attempts="$2"

shift 2

;;

--)

shift

break

;;

*)

error "Unrecognized option: $1"

;;

esac

done

if [[ $# -lt 1 ]]; then

error "Usage: retry.sh [--max-attempts N] -- <command...>"

fi

attempt=1

until "$@"; do

if ((attempt >= max_attempts)); then

error "Command failed after $max_attempts attempts: $*"

fi

delay=$((2 ** attempt))

log "Attempt $attempt/$max_attempts failed, retrying in ${delay}s..."

sleep "$delay"

((attempt++))

done