update.sh

#!/usr/bin/env bash
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at https://mozilla.org/MPL/2.0/.
#
# SentryMail - Update-Routine / update routine
#
# DE: Aktualisiert eine bestehende Installation sicher: optionales DB-Backup,
#     Code per git aktualisieren, Stack neu bauen/starten (Migrationen laufen
#     dabei automatisch beim Backend-Start) und Gesundheit pruefen.
# EN: Safely updates an existing installation: optional DB backup, update code
#     via git, rebuild/restart the stack (migrations run automatically on
#     backend start) and verify health.
#
# Die .env wird NICHT veraendert. / The .env is NOT modified.

# DE: Unter bash laufen. Wird das Skript mit "sh update.sh" (dash) gestartet,
#     hier automatisch mit bash neu starten - sonst scheitern bash-Features.
# EN: Run under bash. If started via "sh update.sh" (dash), re-exec with bash.
if [ -z "${BASH_VERSION:-}" ]; then
  exec bash "$0" "$@"
fi

set -euo pipefail

# --- Pfade / paths -----------------------------------------------------------
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
: "${ENV_FILE:=${SCRIPT_DIR}/.env}"
: "${BACKUP_DIR:=${SCRIPT_DIR}/backups}"

# --- Nicht-interaktiver Modus / non-interactive mode -------------------------
# ASSUME_DEFAULTS=1 uebernimmt bei jeder Frage die Vorgabe (fuer Tests/CI).
: "${ASSUME_DEFAULTS:=0}"

# --- Farben / colours (nur bei TTY) -----------------------------------------
if [ -t 1 ]; then
  BOLD="$(printf '\033[1m')"; DIM="$(printf '\033[2m')"
  GREEN="$(printf '\033[32m')"; YELLOW="$(printf '\033[33m')"
  RED="$(printf '\033[31m')"; CYAN="$(printf '\033[36m')"; RESET="$(printf '\033[0m')"
else
  BOLD=""; DIM=""; GREEN=""; YELLOW=""; RED=""; CYAN=""; RESET=""
fi

# --- Sprache / language ------------------------------------------------------
UILANG="${UILANG:-}"

# msg <deutsch> <english>
msg() { if [ "$UILANG" = "en" ]; then printf '%s' "$2"; else printf '%s' "$1"; fi; }

die() { printf '%s%s%s\n' "$RED" "$1" "$RESET" >&2; exit 1; }

# get_env <KEY> -> aktueller Wert aus ENV_FILE (leer, falls nicht/auskommentiert)
get_env() {
  [ -f "$ENV_FILE" ] || return 0
  KEY="$1" awk -F= 'BEGIN{k=ENVIRON["KEY"]}
    $0 ~ "^"k"=" { sub("^"k"=",""); print; exit }' "$ENV_FILE"
}

# yesno <prompt_de> <prompt_en> <default:y|n> -> 0=ja / 1=nein
yesno() {
  local pd="$1" pe="$2" default="$3" ans
  local prompt; prompt="$(msg "$pd" "$pe")"
  local hint; if [ "$default" = "y" ]; then hint="[J/n]"; else hint="[j/N]"; fi
  [ "$UILANG" = "en" ] && { if [ "$default" = "y" ]; then hint="[Y/n]"; else hint="[y/N]"; fi; }
  if [ "$ASSUME_DEFAULTS" = "1" ]; then
    ans="$default"
  else
    printf '%s%s %s%s ' "$BOLD" "$prompt" "$hint" "$RESET" >&2
    IFS= read -r ans || ans=""
    [ -z "$ans" ] && ans="$default"
  fi
  case "$ans" in [yYjJ]*) return 0 ;; *) return 1 ;; esac
}

# --- Start / language --------------------------------------------------------
if [ -z "$UILANG" ] && [ "$ASSUME_DEFAULTS" != "1" ]; then
  printf '%sSprache / Language:%s [1] Deutsch  [2] English [1]: ' "$BOLD" "$RESET"
  IFS= read -r _l || _l=""
  case "$_l" in 2) UILANG=en ;; *) UILANG=de ;; esac
fi
UILANG="${UILANG:-de}"

printf '\n%s%s%s\n' "$CYAN$BOLD" "SentryMail $(msg 'Update-Routine' 'update routine')" "$RESET"
printf '%s\n\n' "$(msg 'Aktualisiert deine bestehende Installation.' 'Updates your existing installation.')"

# --- 1. Voraussetzungen / prerequisites -------------------------------------
printf '%s%s%s\n' "$BOLD" "$(msg '1) Voraussetzungen pruefen' '1) Checking prerequisites')" "$RESET"
command -v docker >/dev/null 2>&1 || die "$(msg 'docker nicht gefunden.' 'docker not found.')"
docker compose version >/dev/null 2>&1 || die "$(msg 'docker compose v2 nicht gefunden.' 'docker compose v2 not found.')"
printf '   %s✓%s docker / docker compose\n' "$GREEN" "$RESET"
[ -f "$ENV_FILE" ] || die "$(msg '.env fehlt - zuerst ./install.sh ausfuehren.' '.env missing - run ./install.sh first.')"
printf '   %s✓%s .env\n' "$GREEN" "$RESET"

HAVE_GIT=0
if command -v git >/dev/null 2>&1 && git -C "$SCRIPT_DIR" rev-parse --git-dir >/dev/null 2>&1; then
  HAVE_GIT=1
  printf '   %s✓%s git (%s)\n' "$GREEN" "$RESET" "$(git -C "$SCRIPT_DIR" describe --tags --always 2>/dev/null || echo '?')"
else
  printf '   %s!%s %s\n' "$YELLOW" "$RESET" "$(msg 'kein git-Repository - Code-Update wird uebersprungen (ZIP-Installation).' 'no git repository - code update skipped (ZIP install).')"
fi
printf '\n'

# --- 2. DB-Backup / DB backup -----------------------------------------------
printf '%s%s%s\n' "$BOLD" "$(msg '2) Datenbank-Backup' '2) Database backup')" "$RESET"
DB_NAME="$(get_env POSTGRES_DB || true)"; DB_USER="$(get_env POSTGRES_USER || true)"
if [ -z "$DB_NAME" ] || [ -z "$DB_USER" ]; then
  printf '   %s!%s %s\n' "$YELLOW" "$RESET" "$(msg 'POSTGRES_DB/USER nicht in .env - Backup uebersprungen.' 'POSTGRES_DB/USER not in .env - backup skipped.')"
elif ! ( cd "$SCRIPT_DIR" && docker compose ps postgres 2>/dev/null | grep -q "Up\|running" ); then
  printf '   %s!%s %s\n' "$YELLOW" "$RESET" "$(msg 'postgres laeuft nicht - Backup uebersprungen.' 'postgres is not running - backup skipped.')"
elif yesno 'Vor dem Update ein DB-Backup erstellen?' 'Create a DB backup before updating?' y; then
  mkdir -p "$BACKUP_DIR"
  STAMP="$(date +%Y%m%d-%H%M%S)"
  OUT="${BACKUP_DIR}/db-${DB_NAME}-${STAMP}.sql.gz"
  printf '   %s' "$(msg 'Sichere Datenbank ...' 'Dumping database ...')"
  if ( cd "$SCRIPT_DIR" && docker compose exec -T postgres pg_dump -U "$DB_USER" "$DB_NAME" ) | gzip > "$OUT"; then
    printf ' %s✓%s %s (%s)\n' "$GREEN" "$RESET" "$OUT" "$(du -h "$OUT" 2>/dev/null | cut -f1)"
  else
    rm -f "$OUT"
    die "$(msg 'Backup fehlgeschlagen - Update abgebrochen.' 'Backup failed - update aborted.')"
  fi
else
  printf '   %s%s%s\n' "$YELLOW" "$(msg 'Ohne Backup fortgefahren.' 'Continuing without backup.')" "$RESET"
fi
printf '\n'

# --- 3. Code aktualisieren / update code ------------------------------------
printf '%s%s%s\n' "$BOLD" "$(msg '3) Code aktualisieren' '3) Update code')" "$RESET"
if [ "$HAVE_GIT" = "1" ]; then
  git -C "$SCRIPT_DIR" fetch --tags --quiet || true
  if BRANCH="$(git -C "$SCRIPT_DIR" symbolic-ref --short -q HEAD)"; then
    UPSTREAM="$(git -C "$SCRIPT_DIR" rev-parse --abbrev-ref --symbolic-full-name '@{u}' 2>/dev/null || true)"
    if [ -z "$UPSTREAM" ]; then
      printf '   %s!%s %s\n' "$YELLOW" "$RESET" "$(msg "Branch '$BRANCH' hat kein Upstream - uebersprungen." "Branch '$BRANCH' has no upstream - skipped.")"
    else
      AHEAD="$(git -C "$SCRIPT_DIR" rev-list --count "HEAD..$UPSTREAM" 2>/dev/null || echo 0)"
      if [ "$AHEAD" = "0" ]; then
        printf '   %s✓%s %s\n' "$GREEN" "$RESET" "$(msg 'Bereits aktuell.' 'Already up to date.')"
      else
        printf '   %s\n' "$(msg "$AHEAD neue Commit(s) auf $UPSTREAM." "$AHEAD new commit(s) on $UPSTREAM.")"
        if git -C "$SCRIPT_DIR" pull --ff-only --quiet; then
          printf '   %s✓%s %s -> %s\n' "$GREEN" "$RESET" "$(msg 'aktualisiert auf' 'updated to')" "$(git -C "$SCRIPT_DIR" describe --tags --always)"
        else
          die "$(msg 'git pull nicht moeglich (lokale Aenderungen?). Bitte manuell aufloesen.' 'git pull failed (local changes?). Please resolve manually.')"
        fi
      fi
    fi
  else
    # Detached HEAD (auf einem Tag ausgecheckt): neueste Version anzeigen.
    LATEST="$(git -C "$SCRIPT_DIR" tag -l 'v*' --sort=-v:refname | head -n1)"
    printf '   %s!%s %s\n' "$YELLOW" "$RESET" "$(msg "Detached HEAD (fester Tag). Neueste Version: ${LATEST:-?}." "Detached HEAD (pinned tag). Latest version: ${LATEST:-?}.")"
    if [ -n "$LATEST" ] && yesno "Auf $LATEST wechseln?" "Switch to $LATEST?" n; then
      git -C "$SCRIPT_DIR" checkout --quiet "$LATEST" && printf '   %s✓%s %s\n' "$GREEN" "$RESET" "$LATEST"
    fi
  fi
else
  printf '   %s\n' "$(msg 'ZIP-Installation: neue Version manuell herunterladen und entpacken.' 'ZIP install: download and extract the new version manually.')"
fi
printf '\n'

# --- 4. Stack neu bauen/starten / rebuild & restart -------------------------
printf '%s%s%s\n' "$BOLD" "$(msg '4) Stack neu bauen und starten' '4) Rebuild and restart the stack')" "$RESET"
printf '   %s\n' "$(msg 'DB-Migrationen laufen automatisch beim Backend-Start.' 'DB migrations run automatically on backend start.')"
if yesno 'Jetzt "docker compose up -d --build" ausfuehren?' 'Run "docker compose up -d --build" now?' y; then
  # Containernamen vor dem Update merken. Sie tragen seit dem Rebranding das
  # Praefix "sentrymail-" statt "humanshield-"; compose ordnet Container ueber
  # Projekt- und Service-Label zu, erstellt sie also regulaer neu. Die Volumes
  # heissen unveraendert (postgres_data, backend_data, ...) und bleiben damit
  # samt Daten erhalten - allein das alte Netz bleibt ungenutzt zurueck.
  ALT_NETZ="$(docker network ls --format '{{.Name}}' 2>/dev/null | grep -E '_humanshield$' || true)"
  ( cd "$SCRIPT_DIR" && docker compose up -d --build )
  if [ -n "$ALT_NETZ" ]; then
    printf '   %s!%s %s\n' "$YELLOW" "$RESET" "$(msg "Die Container heissen jetzt sentrymail-*. Das alte Netz $ALT_NETZ wird nicht mehr benutzt und kann entfernt werden:" "Containers are now named sentrymail-*. The old network $ALT_NETZ is unused and can be removed:")"
    printf '     %sdocker network rm %s%s\n' "$BOLD" "$ALT_NETZ" "$RESET"
  fi
else
  printf '   %s\n' "$(msg 'Uebersprungen. Spaeter:' 'Skipped. Later:')"
  printf '   %sdocker compose up -d --build%s\n' "$BOLD" "$RESET"
  printf '\n%s\n' "$(msg 'Update beendet (ohne Neustart).' 'Update finished (without restart).')"
  exit 0
fi
printf '\n'

# --- 5. Gesundheit pruefen / health check -----------------------------------
printf '%s%s%s\n' "$BOLD" "$(msg '5) Gesundheit pruefen' '5) Health check')" "$RESET"
OK=0
for _ in $(seq 1 30); do
  if ( cd "$SCRIPT_DIR" && docker compose exec -T backend curl -sf http://localhost:8000/health >/dev/null 2>&1 ); then
    OK=1; break
  fi
  sleep 2
done
if [ "$OK" = "1" ]; then
  printf '   %s✓%s %s\n' "$GREEN" "$RESET" "$(msg 'Backend gesund (/health).' 'Backend healthy (/health).')"
else
  printf '   %s!%s %s\n' "$YELLOW" "$RESET" "$(msg 'Backend noch nicht bereit. Logs: docker compose logs -f backend' 'Backend not ready yet. Logs: docker compose logs -f backend')"
fi
printf '\n'

# --- 6. Veraltete .env-Werte / stale .env values -----------------------------
# DE: Werte aus der Zeit vor der Umbenennung bleiben beim Update stehen - die
#     Vorgaben im Code greifen nur, wenn der Schluessel in der .env fehlt. Eine
#     Altinstallation verliert dadurch stillschweigend die Update-Meldung bzw.
#     erreicht den Lizenzserver nicht. Es wird nichts rot, es faellt nur nie auf.
# EN: Values from before the rename survive an update - the defaults in code
#     only apply when the key is absent from .env. An old installation silently
#     loses update notifications or fails to reach the license server.
#
# Nur warnen, nicht aendern: Die .env wird von diesem Skript bewusst nicht
# angefasst (siehe Kopf). Sie enthaelt SECRET_KEY, DB-Passwort und
# Lizenzschluessel; ein fehlgeschlagenes sed darin waere teurer als der Fehler,
# den es beheben soll. Der Betreiber entscheidet und fuehrt selbst aus.
printf '%s%s%s\n' "$BOLD" "$(msg '6) Veraltete Werte in der .env' '6) Stale values in .env')" "$RESET"
STALE=0
# Format je Eintrag: SCHLUESSEL|Suchmuster|empfohlener Wert
for eintrag in \
  "UPDATE_CHECK_URL|HumanShield|https://api.github.com/repos/securebits-cyber/SentryMail/releases/latest" \
  "LICENSE_SERVER_URL|humanshield|https://license.sentrymail.de"
do
  KEY="${eintrag%%|*}"; REST="${eintrag#*|}"
  MUSTER="${REST%%|*}"; EMPFOHLEN="${REST#*|}"
  WERT="$(get_env "$KEY")"
  [ -n "$WERT" ] || continue
  case "$WERT" in
    *"$MUSTER"*)
      STALE=1
      printf '   %s!%s %s\n' "$YELLOW" "$RESET" \
        "$(msg "$KEY zeigt noch auf die alte Marke:" "$KEY still points at the old brand:")"
      printf '       %s\n' "$WERT"
      printf '     %s\n' "$(msg 'Empfohlen:' 'Recommended:')"
      printf '       %s=%s\n' "$KEY" "$EMPFOHLEN"
      ;;
  esac
done
if [ "$STALE" = "0" ]; then
  printf '   %s✓%s %s\n' "$GREEN" "$RESET" "$(msg 'Keine veralteten Werte gefunden.' 'No stale values found.')"
else
  printf '\n   %s\n' "$(msg 'Bitte von Hand in der .env berichtigen und danach:' 'Please correct these in .env by hand, then run:')"
  printf '       docker compose restart backend\n'
fi
printf '\n'

# --- 7. Hinweis Add-ons / add-on note ---------------------------------------
printf '%s%s%s\n' "$DIM" "$(msg 'Hinweis: Business-/Enterprise-Add-ons haben eigene Releases. In Produktion sind sie Teil des Backend-Images (Rebuild oben genuegt); im Dev-Mount aktualisierst du sie per git pull in ihren Repos + docker compose restart backend.' 'Note: Business/Enterprise add-ons have their own releases. In production they are part of the backend image (the rebuild above covers them); with the dev mount, update them via git pull in their repos + docker compose restart backend.')" "$RESET"
printf '\n%s%s%s\n' "$GREEN$BOLD" "$(msg '✓ Update abgeschlossen.' '✓ Update complete.')" "$RESET"