jbergner
8c67c7a7fa
All checks were successful
release-tag / release-image (push) Successful in 10m52s
43 lines
1.9 KiB
Bash
43 lines
1.9 KiB
Bash
#!/bin/sh
|
|
set -eu
|
|
ENV_FILE=${ENV_FILE:-.env}
|
|
SNAPSHOT=${1:-}
|
|
HELPER_IMAGE=${BACKUP_HELPER_IMAGE:-busybox:1.36}
|
|
[ -n "$SNAPSHOT" ] && [ -d "$SNAPSHOT" ] || { echo "usage: $0 SNAPSHOT_DIR" >&2; exit 2; }
|
|
[ -f "$ENV_FILE" ] || { echo "restore: missing $ENV_FILE" >&2; exit 1; }
|
|
( cd "$SNAPSHOT" && sha256sum -c SHA256SUMS )
|
|
./scripts/preflight.sh
|
|
getv() { sed -n "s/^$1=//p" "$ENV_FILE" | tail -n1 | tr -d '\r'; }
|
|
|
|
restart() { docker compose --env-file "$ENV_FILE" up -d knowledge neuroforge neuroforge-worker agent >/dev/null 2>&1 || true; }
|
|
trap restart EXIT INT TERM
|
|
|
|
docker compose --env-file "$ENV_FILE" stop neuroforge-worker agent neuroforge knowledge
|
|
docker compose --env-file "$ENV_FILE" create neuroforge agent >/dev/null
|
|
volume_for() {
|
|
service=$1; destination=$2
|
|
cid=$(docker compose --env-file "$ENV_FILE" ps -aq "$service")
|
|
[ -n "$cid" ] || return 1
|
|
docker inspect -f "{{range .Mounts}}{{if eq .Destination \"$destination\"}}{{.Name}}{{end}}{{end}}" "$cid"
|
|
}
|
|
restore_volume() {
|
|
service=$1; destination=$2; source=$3
|
|
volume=$(volume_for "$service" "$destination")
|
|
[ -n "$volume" ] || { echo "restore: volume for $service:$destination not found" >&2; exit 1; }
|
|
src=$(CDPATH= cd -- "$source" && pwd)
|
|
docker run --rm -v "$volume:/target" -v "$src:/source:ro" "$HELPER_IMAGE" sh -eu -c 'rm -rf /target/* /target/.[!.]* /target/..?* 2>/dev/null || true; cp -a /source/. /target/'
|
|
}
|
|
restore_volume neuroforge /app/data "$SNAPSHOT/neuroforge-data"
|
|
restore_volume agent /app/data "$SNAPSHOT/agent-data"
|
|
restore_host() {
|
|
src=$1; dst=$2
|
|
[ -d "$src" ] || return 0
|
|
mkdir -p "$dst"
|
|
find "$dst" -mindepth 1 -maxdepth 1 -exec rm -rf -- {} +
|
|
cp -a "$src"/. "$dst"/
|
|
}
|
|
restore_host "$SNAPSHOT/knowledge" "$(getv KB_DATA_PATH)"
|
|
restore_host "$SNAPSHOT/staging" "$(getv KB_STAGING_PATH)"
|
|
restore_host "$SNAPSHOT/knowledge-backups" "$(getv KB_BACKUP_PATH)"
|
|
echo "restore: snapshot restored; services will be restarted"
|