Files
TA-Deployment-Broker/update.sh
T
2026-07-25 15:57:12 -05:00

91 lines
3.0 KiB
Bash
Executable File

#!/usr/bin/env bash
set -Eeuo pipefail
SCRIPT_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")" && pwd -P)"
SCRIPT_PATH="${SCRIPT_DIR}/$(basename -- "${BASH_SOURCE[0]}")"
fail() {
printf 'ERROR: %s\n' "$*" >&2
exit 1
}
trap 'printf "ERROR: broker update failed at line %d.\n" "$LINENO" >&2' ERR
cd "$SCRIPT_DIR"
command -v git >/dev/null 2>&1 || fail "git is required"
if [[ "${1:-}" != "--post-pull" ]]; then
branch="$(git symbolic-ref --quiet --short HEAD)" ||
fail "the repository is in detached HEAD state"
upstream="$(git rev-parse --abbrev-ref --symbolic-full-name '@{upstream}' 2>/dev/null)" ||
fail "branch ${branch} does not have an upstream branch"
remote="$(git config --get "branch.${branch}.remote")" ||
fail "branch ${branch} does not have a configured remote"
git diff --quiet && git diff --cached --quiet ||
fail "tracked local changes must be committed or stashed before updating"
printf 'Checking %s against %s...\n' "$branch" "$upstream"
git fetch --prune "$remote"
local_commit="$(git rev-parse HEAD)"
upstream_commit="$(git rev-parse '@{upstream}')"
if [[ "$local_commit" == "$upstream_commit" ]]; then
printf 'TAPM Deployment Broker is already up to date; no changes were made.\n'
exit 0
fi
if git merge-base --is-ancestor "$local_commit" "$upstream_commit"; then
printf 'Updating TAPM Deployment Broker from %s...\n' "$upstream"
git merge --ff-only "$upstream_commit"
elif git merge-base --is-ancestor "$upstream_commit" "$local_commit"; then
fail "local branch ${branch} is ahead of ${upstream}; refusing to rebuild"
else
fail "local branch ${branch} has diverged from ${upstream}"
fi
# Restart from the newly pulled script so updates to this file take effect.
exec "$SCRIPT_PATH" --post-pull
fi
command -v docker >/dev/null 2>&1 || fail "docker is required"
docker compose version >/dev/null 2>&1 || fail "Docker Compose v2 is required"
[[ -f compose.yaml ]] || fail "compose.yaml was not found in ${SCRIPT_DIR}"
[[ -f .env ]] || fail ".env was not found in ${SCRIPT_DIR}"
printf 'Building broker image...\n'
docker compose build broker
printf 'Applying pending database migrations...\n'
docker compose run --rm migrate
printf 'Starting broker...\n'
docker compose up -d broker
container_id="$(docker compose ps -q broker)"
[[ -n "$container_id" ]] || fail "Docker Compose did not return a broker container"
printf 'Waiting for broker health check'
for ((attempt = 1; attempt <= 75; attempt++)); do
status="$(docker inspect --format '{{if .State.Health}}{{.State.Health.Status}}{{else}}{{.State.Status}}{{end}}' "$container_id")"
case "$status" in
healthy)
printf ' healthy.\n'
docker compose ps broker
printf 'TAPM Deployment Broker update completed successfully.\n'
exit 0
;;
exited | dead | unhealthy)
printf ' %s.\n' "$status"
docker compose logs --tail=100 broker
fail "broker failed its health check"
;;
esac
printf '.'
sleep 2
done
printf ' timed out.\n'
docker compose logs --tail=100 broker
fail "broker did not become healthy within 150 seconds"